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())
