def setUp(self):
super(RestApiKeyAuthTestCase, self).setUp()
self.tm1 = TestModel.create(data='test1')
self.tm2 = TestModel.create(data='test2')
self.k1 = APIKey.create(key='k', secret='s')
self.k2 = APIKey.create(key='k2', secret='s2')
def test_create(self):
with self.flask_app.test_client() as c:
test_data = {'data': 't3'}
serialized = json.dumps(test_data)
resp = c.post('/api/testmodel/', data=serialized)
self.assertEqual(resp.status_code, 401)
self.assertEqual(g.api_key, None)
resp = c.post('/api/testmodel/?key=k&secret=s2', data=serialized)
self.assertEqual(resp.status_code, 401)
self.assertEqual(g.api_key, None)
# test passing in via get args
resp = c.post('/api/testmodel/?key=k&secret=s', data=serialized)
self.assertEqual(g.api_key, self.k1)
resp_json = self.response_json(resp)
self.assertEqual(TestModel.select().count(), 3)
self.assertEqual(resp_json['data'], 't3')
def setUp(self):
super(RestApiTestCase, self).setUp()
TestModel.drop_table(True)
APIKey.drop_table(True)
APIKey.create_table()
TestModel.create_table()