tests: fix PEP8ness

This commit is contained in:
Benjamin Dauvergne 2019-05-15 18:19:14 +02:00
parent cecf66e523
commit 76af20c99e
3 changed files with 8 additions and 8 deletions

View File

@ -1,4 +1,5 @@
import os
import tempfile
DEBUG = False
@ -15,5 +16,4 @@ DATABASES = {
}
}
import tempfile
MEDIA_ROOT = tempfile.mkdtemp('fargo-test')

View File

@ -74,7 +74,7 @@ def test_push_document(app, admin_user, john_doe):
assert models.Origin.objects.get().slug == 'wcs'
assert models.UserDocument.objects.count() == 1
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.get().content.read() == b'coin'
assert (models.UserDocument.objects.get().document
@ -86,15 +86,15 @@ def test_push_document(app, admin_user, john_doe):
data['deletable_by_user'] = False
response = app.post_json(url, params=data, status=200)
assert response.json['result'] == 1
assert models.Origin.objects.count() == 1 # same origin
assert models.UserDocument.objects.count() == 2 # new document
assert models.Origin.objects.count() == 1 # same origin
assert models.UserDocument.objects.count() == 2 # new document
assert models.UserDocument.objects.filter(deletable_by_user=False).count() == 1
data['file_b64_content'] = base64.b64encode(b'coin3').decode()
data['deletable_by_user'] = True
response = app.post_json(url, params=data, status=200)
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
# same document
@ -103,7 +103,7 @@ def test_push_document(app, admin_user, john_doe):
response = app.post_json(url, params=data, status=400)
assert response.json['result'] == 0
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):

View File

@ -238,9 +238,9 @@ def test_command_create_client(db):
def test_command_put_document(db, capsys, app, john_doe):
call_command('oauth2-create-client', 'test', 'https://example.com/')
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/'
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()
assert err == ''
url = out.strip()