class StagesAPITest(APITestCase):
def setUp(self):
self.stage = Stage(id = 1, name="new stage")
self.stage.save()
def tearDown(self):
self.stage.delete()
def testStageGet(self):
response = self.client.get("/api/stages/1/", format="application/json")
self.assertEqual(status.HTTP_200_OK, response.status_code)
def testStageGet2(self):
response = self.client.get("/api/stages/2/", format="application/json")
self.assertEqual(status.HTTP_404_NOT_FOUND, response.status_code)
def testPost(self):
response = self.client.post("/api/stages/1/", {'stage':1})
self.assertEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)
