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)