class WhenTestingWorkersStatus(unittest.TestCase):
def setUp(self):
self.req = MagicMock()
self.resp = MagicMock()
self.db_handler = MagicMock()
self.resource = WorkersStatusResource(self.db_handler)
self.registration = WorkerRegistration('worker').format()
self.worker = Worker(**self.registration)
self.worker_dict = Worker(**self.registration).format()
def test_returns_200_on_get(self):
self.db_handler.find.return_value = [self.worker_dict]
self.resource.on_get(self.req, self.resp)
self.assertEquals(self.resp.status, falcon.HTTP_200)
resp = jsonutils.loads(self.resp.body)
status = resp['status'][0]
for key in resp.keys():
self.assertTrue(key in self.worker.get_status().keys())
class WhenTestingWorkersStatus(unittest.TestCase):
def setUp(self):
self.req = MagicMock()
self.resp = MagicMock()
self.resource = WorkersStatusResource()
self.registration = WorkerRegistration('worker').format()
self.worker = Worker(**self.registration)
self.worker_dict = Worker(**self.registration).format()
def test_returns_200_on_get(self):
with patch('meniscus.data.model.worker_util.retrieve_all_workers',
MagicMock(return_value=[self.worker])):
self.resource.on_get(self.req, self.resp)
self.assertEquals(self.resp.status, falcon.HTTP_200)
resp = jsonutils.loads(self.resp.body)
status = resp['status'][0]
for key in resp.keys():
self.assertTrue(key in self.worker.get_status().keys())
class WhenTestingWorkersStatus(unittest.TestCase):
def setUp(self):
self.req = MagicMock()
self.resp = MagicMock()
self.db_handler = MagicMock()
self.resource = WorkersStatusResource(self.db_handler)
self.registration = WorkerRegistration('correlation').format()
self.worker = Worker(**self.registration)
self.worker_dict = Worker(**self.registration).format()
def test_returns_200_on_get(self):
self.db_handler.find.return_value = [self.worker_dict]
self.resource.on_get(self.req, self.resp)
self.assertEquals(self.resp.status, falcon.HTTP_200)
resp = jsonutils.loads(self.resp.body)
status = resp['status'][0]
for key in resp.keys():
self.assertTrue(key in self.worker.get_status().keys())
class WhenTestingWorkersStatus(unittest.TestCase):
def setUp(self):
self.req = MagicMock()
self.resp = MagicMock()
self.hostname = 'worker01'
self.resource = WorkersStatusResource()
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={})
def test_returns_200_on_get(self):
with patch('meniscus.data.model.worker_util.retrieve_all_workers',
MagicMock(return_value=[self.worker])):
self.resource.on_get(self.req, self.resp)
self.assertEquals(self.resp.status, falcon.HTTP_200)
resp = jsonutils.loads(self.resp.body)
status = resp['status'][0]
for key in resp.keys():
self.assertTrue(key in self.worker.get_status().keys())
