def test_do_list_usage_deletes_with_instance(self): fake_request = self.mox.CreateMockAnything() fake_request.GET = {'instance': INSTANCE_ID_1} results = self.mox.CreateMockAnything() models.InstanceDeletes.objects.filter(instance=INSTANCE_ID_1)\ .AndReturn(results) usage = self.mox.CreateMockAnything() usage.instance = INSTANCE_ID_1 usage.launched_at = utils.decimal_utc() usage.deleted_at = usage.launched_at + 10 results.__iter__().AndReturn([usage].__iter__()) self.mox.ReplayAll() resp = stacky_server.do_list_usage_deletes(fake_request) self.assertEqual(resp.status_code, 200) resp_json = json.loads(resp.content) self.assertEqual(len(resp_json), 2) self.assertEqual(resp_json[0], ["UUID", "Launched At", "Deleted At"]) self.assertEqual(resp_json[1][0], INSTANCE_ID_1) launch_time_str = dt.dt_from_decimal(usage.launched_at) self.assertEqual(resp_json[1][1], str(launch_time_str)) delete_time_str = dt.dt_from_decimal(usage.deleted_at) self.assertEqual(resp_json[1][2], str(delete_time_str)) self.mox.VerifyAll()
def test_do_list_usage_deletes_bad_instance(self): fake_request = self.mox.CreateMockAnything() fake_request.GET = {"instance": "obviouslybaduuid"} self.mox.ReplayAll() resp = stacky_server.do_list_usage_deletes(fake_request) self.assertEqual(resp.status_code, 400) resp_json = json.loads(resp.content) self.assertEqual(len(resp_json), 2) self.assertEqual(resp_json[0], ["Error", "Message"]) msg = "obviouslybaduuid is not uuid-like" self.assertEqual(resp_json[1], ["Bad Request", msg]) self.mox.VerifyAll()
def test_do_list_usage_deletes_bad_instance(self): fake_request = self.mox.CreateMockAnything() fake_request.GET = {'instance': "obviouslybaduuid"} self.mox.ReplayAll() resp = stacky_server.do_list_usage_deletes(fake_request) self.assertEqual(resp.status_code, 400) resp_json = json.loads(resp.content) self.assertEqual(len(resp_json), 2) self.assertEqual(resp_json[0], ['Error', 'Message']) msg = 'obviouslybaduuid is not uuid-like' self.assertEqual(resp_json[1], ['Bad Request', msg]) self.mox.VerifyAll()
def test_do_list_usage_deletes(self): fake_request = self.mox.CreateMockAnything() fake_request.GET = {} results = self.mox.CreateMockAnything() models.InstanceDeletes.objects.all().AndReturn(results) usage = self.mox.CreateMockAnything() usage.instance = INSTANCE_ID_1 usage.launched_at = utils.decimal_utc() usage.deleted_at = usage.launched_at + 10 results.__iter__().AndReturn([usage].__iter__()) self.mox.ReplayAll() resp = stacky_server.do_list_usage_deletes(fake_request) self.assertEqual(resp.status_code, 200) resp_json = json.loads(resp.content) self.assertEqual(len(resp_json), 2) self.assertEqual(resp_json[0], ["UUID", "Launched At", "Deleted At"]) self.assertEqual(resp_json[1][0], INSTANCE_ID_1) launch_time_str = dt.dt_from_decimal(usage.launched_at) self.assertEqual(resp_json[1][1], str(launch_time_str)) delete_time_str = dt.dt_from_decimal(usage.deleted_at) self.assertEqual(resp_json[1][2], str(delete_time_str)) self.mox.VerifyAll()