tests: fix PEP8ness
This commit is contained in:
parent
cecf66e523
commit
76af20c99e
|
@ -1,4 +1,5 @@
|
|||
import os
|
||||
import tempfile
|
||||
|
||||
DEBUG = False
|
||||
|
||||
|
@ -15,5 +16,4 @@ DATABASES = {
|
|||
}
|
||||
}
|
||||
|
||||
import tempfile
|
||||
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.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):
|
||||
|
|
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue