tests: fix PEP8ness
This commit is contained in:
parent
cecf66e523
commit
76af20c99e
|
@ -1,4 +1,5 @@
|
||||||
import os
|
import os
|
||||||
|
import tempfile
|
||||||
|
|
||||||
DEBUG = False
|
DEBUG = False
|
||||||
|
|
||||||
|
@ -15,5 +16,4 @@ DATABASES = {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
import tempfile
|
|
||||||
MEDIA_ROOT = tempfile.mkdtemp('fargo-test')
|
MEDIA_ROOT = tempfile.mkdtemp('fargo-test')
|
||||||
|
|
|
@ -74,7 +74,7 @@ def test_push_document(app, admin_user, john_doe):
|
||||||
assert models.Origin.objects.get().slug == 'wcs'
|
assert models.Origin.objects.get().slug == 'wcs'
|
||||||
assert models.UserDocument.objects.count() == 1
|
assert models.UserDocument.objects.count() == 1
|
||||||
assert models.UserDocument.objects.get().user == john_doe
|
assert models.UserDocument.objects.get().user == john_doe
|
||||||
assert models.UserDocument.objects.get().deletable_by_user == True
|
assert models.UserDocument.objects.get().deletable_by_user is True
|
||||||
assert models.Document.objects.count() == 1
|
assert models.Document.objects.count() == 1
|
||||||
assert models.Document.objects.get().content.read() == b'coin'
|
assert models.Document.objects.get().content.read() == b'coin'
|
||||||
assert (models.UserDocument.objects.get().document
|
assert (models.UserDocument.objects.get().document
|
||||||
|
@ -86,15 +86,15 @@ def test_push_document(app, admin_user, john_doe):
|
||||||
data['deletable_by_user'] = False
|
data['deletable_by_user'] = False
|
||||||
response = app.post_json(url, params=data, status=200)
|
response = app.post_json(url, params=data, status=200)
|
||||||
assert response.json['result'] == 1
|
assert response.json['result'] == 1
|
||||||
assert models.Origin.objects.count() == 1 # same origin
|
assert models.Origin.objects.count() == 1 # same origin
|
||||||
assert models.UserDocument.objects.count() == 2 # new document
|
assert models.UserDocument.objects.count() == 2 # new document
|
||||||
assert models.UserDocument.objects.filter(deletable_by_user=False).count() == 1
|
assert models.UserDocument.objects.filter(deletable_by_user=False).count() == 1
|
||||||
|
|
||||||
data['file_b64_content'] = base64.b64encode(b'coin3').decode()
|
data['file_b64_content'] = base64.b64encode(b'coin3').decode()
|
||||||
data['deletable_by_user'] = True
|
data['deletable_by_user'] = True
|
||||||
response = app.post_json(url, params=data, status=200)
|
response = app.post_json(url, params=data, status=200)
|
||||||
assert response.json['result'] == 1
|
assert response.json['result'] == 1
|
||||||
assert models.UserDocument.objects.count() == 3 # new document
|
assert models.UserDocument.objects.count() == 3 # new document
|
||||||
assert models.UserDocument.objects.filter(deletable_by_user=True).count() == 2
|
assert models.UserDocument.objects.filter(deletable_by_user=True).count() == 2
|
||||||
|
|
||||||
# same document
|
# same document
|
||||||
|
@ -103,7 +103,7 @@ def test_push_document(app, admin_user, john_doe):
|
||||||
response = app.post_json(url, params=data, status=400)
|
response = app.post_json(url, params=data, status=400)
|
||||||
assert response.json['result'] == 0
|
assert response.json['result'] == 0
|
||||||
assert response.json['errors'][0]['code'] == 'document-exists'
|
assert response.json['errors'][0]['code'] == 'document-exists'
|
||||||
assert models.UserDocument.objects.count() == 3 # no new document
|
assert models.UserDocument.objects.count() == 3 # no new document
|
||||||
|
|
||||||
|
|
||||||
def test_push_document_max_document_size(app, private_settings, admin_user, john_doe):
|
def test_push_document_max_document_size(app, private_settings, admin_user, john_doe):
|
||||||
|
|
|
@ -238,9 +238,9 @@ def test_command_create_client(db):
|
||||||
def test_command_put_document(db, capsys, app, john_doe):
|
def test_command_put_document(db, capsys, app, john_doe):
|
||||||
call_command('oauth2-create-client', 'test', 'https://example.com/')
|
call_command('oauth2-create-client', 'test', 'https://example.com/')
|
||||||
client = OAuth2Client.objects.get()
|
client = OAuth2Client.objects.get()
|
||||||
path = os.path.join(os.path.dirname(__file__), 'pdf-sample.pdf')
|
path = os.path.join(os.path.dirname(__file__), 'pdf-sample.pdf')
|
||||||
redirect_uri = 'https://example.com/'
|
redirect_uri = 'https://example.com/'
|
||||||
r = call_command('oauth2-put-document', '--client-id=%s' % client.pk, redirect_uri, path)
|
call_command('oauth2-put-document', '--client-id=%s' % client.pk, redirect_uri, path)
|
||||||
out, err = capsys.readouterr()
|
out, err = capsys.readouterr()
|
||||||
assert err == ''
|
assert err == ''
|
||||||
url = out.strip()
|
url = out.strip()
|
||||||
|
|
Loading…
Reference in New Issue