class WhenTestingWorkerStatus(unittest.TestCase): def setUp(self): self.req = MagicMock() self.resp = MagicMock() self.resource = WorkerStatusResource() self.registration = WorkerRegistration('worker').format() self.worker = Worker(**self.registration) self.worker_dict = Worker(**self.registration).format() self.worker_id = '51375fc4eea50d53066292b6' self.worker_not_found = None def test_raises_worker_not_found(self): with patch('meniscus.data.model.worker_util.find_worker', MagicMock(return_value=None)): with self.assertRaises(falcon.HTTPError): self.resource.on_get(self.req, self.resp, self.worker_id) def test_returns_200_on_get(self): with patch('meniscus.data.model.worker_util.find_worker', MagicMock(return_value=self.worker)): self.resource.on_get(self.req, self.resp, self.worker_id) self.assertEquals(self.resp.status, falcon.HTTP_200) resp = jsonutils.loads(self.resp.body) status = resp['status'] for key in resp.keys(): self.assertTrue(key in self.worker.get_status().keys())
class WhenTestingWorkerStatus(unittest.TestCase): def setUp(self): self.req = MagicMock() self.resp = MagicMock() self.hostname = 'worker01' self.resource = WorkerStatusResource() self.worker = Worker(_id='010101', hostname=self.hostname, ip_address_v4='172.23.1.100', ip_address_v6='::1', personality='worker01', status='online', system_info={}) self.hostname = 'worker01' self.worker_not_found = None def test_raises_worker_not_found(self): with patch('meniscus.data.model.worker_util.find_worker', MagicMock(return_value=None)): with self.assertRaises(falcon.HTTPError): self.resource.on_get(self.req, self.resp, self.hostname) def test_returns_200_on_get(self): with patch('meniscus.data.model.worker_util.find_worker', MagicMock(return_value=self.worker)): self.resource.on_get(self.req, self.resp, self.hostname) self.assertEquals(self.resp.status, falcon.HTTP_200) resp = jsonutils.loads(self.resp.body) status = resp['status'] for key in resp.keys(): self.assertTrue(key in self.worker.get_status().keys())
class WhenTestingWorkerStatus(unittest.TestCase): def setUp(self): self.req = MagicMock() self.resp = MagicMock() self.db_handler = MagicMock() self.resource = WorkerStatusResource(self.db_handler) self.registration = WorkerRegistration('worker').format() self.worker = Worker(**self.registration) self.worker_dict = Worker(**self.registration).format() self.worker_id = '51375fc4eea50d53066292b6' self.worker_not_found = None def test_raises_worker_not_found(self): self.db_handler.find_one.return_value = self.worker_not_found with self.assertRaises(falcon.HTTPError): self.resource.on_get(self.req, self.resp, self.worker_id) def test_returns_200_on_get(self): self.db_handler.find_one.return_value = self.worker_dict self.resource.on_get(self.req, self.resp, self.worker_id) self.assertEquals(self.resp.status, falcon.HTTP_200) resp = jsonutils.loads(self.resp.body) status = resp['status'] for key in resp.keys(): self.assertTrue(key in self.worker.get_status().keys())