reformat code with black

This commit is contained in:
Martin
2019-10-30 13:44:04 +01:00
parent 354e959cc5
commit 0dd8d619b4
28 changed files with 1236 additions and 689 deletions
+27 -21
View File
@@ -6,9 +6,8 @@ from unittest.mock import patch
class TestEndpoints(flask_testing.TestCase):
def create_app(self):
return create_app('tests.config')
return create_app("tests.config")
def setUp(self):
db.create_all()
@@ -17,7 +16,7 @@ class TestEndpoints(flask_testing.TestCase):
db.session.remove()
db.drop_all()
def _add_designer(self, name, username=None, territory='SE', currency='SEK'):
def _add_designer(self, name, username=None, territory="SE", currency="SEK"):
if not username:
username = name.lower()
designer = Designer(name, username, territory, currency)
@@ -26,38 +25,45 @@ class TestEndpoints(flask_testing.TestCase):
return designer
def test_list_designers(self):
self._add_designer('designer1')
self._add_designer('designer2')
self._add_designer("designer1")
self._add_designer("designer2")
response = self.client.get("/designers/")
self.assertEqual(2, len(response.json.get('data')))
self.assertEqual(2, len(response.json.get("data")))
def test_get_designer(self):
designer = self._add_designer('designer1')
response = self.client.get('/designers/1')
designer = self._add_designer("designer1")
response = self.client.get("/designers/1")
self.assert200(response)
def test_create_designer(self):
data = {'name': 'Spider-Man', 'username': 'spiderman',
'territory': 'US', 'currency': 'USD'}
data = {
"name": "Spider-Man",
"username": "spiderman",
"territory": "US",
"currency": "USD",
}
response = self.client.post(
'/designers/', data=json.dumps(data), content_type='application/json')
"/designers/", data=json.dumps(data), content_type="application/json"
)
self.assertEqual(201, response.status_code)
designer = response.json
self.assertEqual('Spider-Man', designer['name'])
self.assertEqual('spider_man', designer['path'])
self.assertEqual("Spider-Man", designer["name"])
self.assertEqual("spider_man", designer["path"])
def test_create_designer_missing_data(self):
data = {'name': 'Only name'}
data = {"name": "Only name"}
response = self.client.post(
'/designers/', data=json.dumps(data), content_type='application/json')
"/designers/", data=json.dumps(data), content_type="application/json"
)
self.assert400(response)
@patch('api.tasks.esales.update_designer')
@patch("api.tasks.esales.update_designer")
def test_update_designer(self, esales_update_designer):
designer = self._add_designer('designer1')
data = {'name': 'test'}
designer = self._add_designer("designer1")
data = {"name": "test"}
response = self.client.put(
'/designers/1', data=json.dumps(data), content_type='application/json')
"/designers/1", data=json.dumps(data), content_type="application/json"
)
self.assertTrue(esales_update_designer.called)
self.assertEqual('test', response.json['name'])
self.assertEqual('test', Designer.query.get(1).name)
self.assertEqual("test", response.json["name"])
self.assertEqual("test", Designer.query.get(1).name)