reformat code with black
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user