Anton Shestakov <av6@dwimlabs.net>, Tue, 19 Jul 2016 18:16:23 +0800
ui: use dot operator to access known object properties
test.py
Permissions: -rwxr-xr-x
from itertools import chain from unittest import TestCase, main class BaseTestCase(TestCase): self.c = app.test_client() class IndexTestCase(BaseTestCase): response = self.c.get('/') self.assertIn('Fruit Bar', response.data) self.assertRegexpMatches(response.data, 'bundles/fruitbar\.[0-9a-f]+\.js') class ResourceTestCase(BaseTestCase): def create_resource(self, new_resource): url = self.list_url % self.workspace response = self.c.post(url, content_type='application/json', data=json.dumps(new_resource)) data = json.loads(response.data) self.assertIn('_t', data) self.assertEqual(data.pop('_t'), self.resource_type) self.assertIn('workspace', data) self.assertEqual(data.pop('workspace'), self.workspace) resource_id = data.pop('_id') self.assertDictEqual(new_resource, data) def read_resource(self, resource_id): url = self.item_url % (self.workspace, resource_id) response = self.c.get(url) data = json.loads(response.data) self.assertEqual(data['_id'], resource_id) url = self.list_url % self.workspace response = self.c.get(url) data = json.loads(response.data) self.assertIn(resource_id, [resource['_id'] for resource in data]) url = self.all_url % self.workspace response = self.c.get(url) data = json.loads(response.data) combined = chain(*data.values()) self.assertIn(resource_id, [resource['_id'] for resource in combined]) def update_resource(self, resource_id, resource_update): url = self.item_url % (self.workspace, resource_id) response = self.c.put(url, content_type='application/json', data=json.dumps(resource_update)) data = json.loads(response.data) self.assertEqual(data['_id'], resource_id) for key, value in resource_update.items(): self.assertEqual(data[key], value) def delete_resource(self, resource_id): url = self.item_url % (self.workspace, resource_id) response = self.c.delete(url) data = json.loads(response.data) self.assertEqual(data, '') response = self.c.get(url) self.assertEqual(response.status_code, 404) class ProjectsTestCase(ResourceTestCase): resource_type = 'project' list_url = '/%s/projects/' item_url = '/%s/projects/%s/' project_id = self.create_resource({ self.read_resource(project_id) self.update_resource(project_id, { 'name': 'Renamed Project' self.delete_resource(project_id) class TasksTestCase(ResourceTestCase): item_url = '/%s/tasks/%s/' task_id = self.create_resource({ self.read_resource(task_id) self.update_resource(task_id, { self.delete_resource(task_id) if __name__ == '__main__': app.config['DEBUG'] = False app.config['ASSETS_DEBUG'] = False app.config['TESTING'] = True app.config['DATABASE'] = 'test_db'