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_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_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)