def create_notes(self): notes = [ Note.create(user=self.admin, message='admin'), Note.create(user=self.normal, message='normal'), ] self.admin_note, self.normal_note = notes return notes
def setUp(self): Note.drop_table(True) Message.drop_table(True) User.drop_table(True) User.create_table() Message.create_table() Note.create_table() self.flask_app = test_app.app self.flask_app._template_context = {} self.app = test_app.app.test_client()
def test_panel_simple(self): users = self.create_users() with self.flask_app.test_client() as c: self.login(c) self.assertEqual(Note.select().count(), 0) resp = c.post('/admin/notes/create/', data={'message': 'testing'}) self.assertEqual(resp.status_code, 302) self.assertTrue(resp.headers['location'].endswith('/admin/')) self.assertEqual(Note.select().count(), 1) note = Note.get(user=self.admin) self.assertEqual(note.message, 'testing')
def test_model_admin_index_pagination(self): users = self.create_users() notes = {} for user in users: notes[user] = [Note.create(user=user, message='test-%d' % i) for i in range(20)] with self.flask_app.test_client() as c: self.login(c) # test a simple lookup resp = c.get('/admin/note/?ordering=id') self.assertEqual(resp.status_code, 200) query = self.get_context('query') self.assertEqual(list(query.get_list()), notes[users[0]]) resp = c.get('/admin/note/?ordering=id&page=2') self.assertEqual(resp.status_code, 200) query = self.get_context('query') self.assertEqual(list(query.get_list()), notes[users[1]]) resp = c.get('/admin/note/?ordering=id&page=1&user=%d&user=%d' % (users[1].id, users[2].id)) self.assertEqual(resp.status_code, 200) query = self.get_context('query') self.assertEqual(list(query.get_list()), notes[users[1]]) resp = c.get('/admin/note/?ordering=id&page=2&user=%d&user=%d' % (users[1].id, users[2].id)) self.assertEqual(resp.status_code, 200) query = self.get_context('query') self.assertEqual(list(query.get_list()), notes[users[2]])
def get_users_and_notes(self): users = self.create_users() notes = [] for i in range(10): for user in users: notes.append(Note.create(user=user, message='%s-%s' % (user.username, i))) return users, notes
def test_delete(self): self.create_notes() url = '/api/note/%s/' % self.admin_note.id # authorized as an admin resp = self.app.delete(url, headers=self.auth_headers('normal', 'normal')) self.assertEqual(resp.status_code, 200) self.assertEqual(Note.select().count(), 1) resp_json = self.response_json(resp) self.assertEqual(resp_json, {'deleted': 1})
def test_create(self): note_data = {'message': 'test', 'user': self.inactive.id} serialized = json.dumps(note_data) # authorized as an admin resp = self.app.post('/api/note/', data=serialized, headers=self.auth_headers('normal', 'normal')) self.assertEqual(resp.status_code, 200) new_note = Note.get(message='test') self.assertEqual(new_note.user, self.inactive) resp_json = self.response_json(resp) self.assertAPINote(resp_json, new_note)
def test_edit(self): self.create_notes() note_data = {'message': 'edited'} serialized = json.dumps(note_data) url = '/api/note/%s/' % self.admin_note.id # authorized as an admin resp = self.app.put(url, data=serialized, headers=self.auth_headers('normal', 'normal')) self.assertEqual(resp.status_code, 200) note = Note.get(id=self.admin_note.id) self.assertEqual(note.message, 'edited') resp_json = self.response_json(resp) self.assertAPINote(resp_json, note)
def test_filtering(self): users, notes = self.get_users_and_notes() # do a simple filter on a related model resp = self.app.get('/api/note/?user=%s&ordering=id' % self.normal.id) resp_json = self.response_json(resp) self.assertAPIMeta(resp_json, { 'model': 'note', 'previous': '', 'next': '', 'page': 1, }) self.assertAPINotes(resp_json, self.normal.note_set.order_by('id')) # do a filter following a join resp = self.app.get('/api/note/?user__username=admin&ordering=id') resp_json = self.response_json(resp) self.assertAPIMeta(resp_json, { 'model': 'note', 'previous': '', 'next': '', 'page': 1, }) self.assertAPINotes(resp_json, self.admin.note_set.order_by('id')) # filter multiple fields notes = list(self.admin.note_set.order_by('id')) third_id = notes[3].id resp = self.app.get('/api/note/?user__username=admin&id__lt=%s&ordering=id' % third_id) resp_json = self.response_json(resp) self.assertAPINotes(resp_json, notes[:3]) # do a filter using multiple values resp = self.app.get('/api/note/?user__username=admin&user__username=inactive&ordering=id') resp_json = self.response_json(resp) self.assertAPIMeta(resp_json, { 'model': 'note', 'previous': '', 'next': '', 'page': 1, }) self.assertAPINotes(resp_json, Note.filter(user__in=[self.admin, self.inactive]).order_by('id'))
def test_model_admin_index_filters(self): users = self.create_users() notes = {} for user in users: notes[user] = [Note.create(user=user, message='test-%d' % i) for i in range(3)] with self.flask_app.test_client() as c: self.login(c) # test a simple lookup resp = c.get('/admin/user/?username=admin') self.assertEqual(resp.status_code, 200) self.assertContext('user', self.admin) self.assertContext('model_admin', admin._registry['user']) self.assertContext('ordering', '') self.assertContext('filters', [('username', 'admin')]) query = self.get_context('query') self.assertEqual(list(query.get_list()), [ self.admin, ]) # test a lookup using multiple values resp = c.get('/admin/user/?username=admin&username=normal&ordering=-username') self.assertEqual(resp.status_code, 200) self.assertContext('filters', [('username__in', ['admin', 'normal'])]) query = self.get_context('query') self.assertEqual(list(query.get_list()), [ self.normal, self.admin, ]) # test a lookup spanning a relation resp = c.get('/admin/note/?user=%d' % self.normal.id) self.assertEqual(resp.status_code, 200) self.assertContext('model_admin', admin._registry['note']) self.assertContext('filters', [('user', str(self.normal.id))]) query = self.get_context('query') self.assertEqual(list(query.get_list()), notes[self.normal])