def test_get_host_status(self): """ Verify that KuberContainerManager().test_get_host_status() works as expected. """ kube_container_mgr = KubeContainerManager(CONFIG) status_struct = {'status': {'use': 'kube'}} for test_data in ( (200, status_struct, False), (200, status_struct['status'], True)): kube_container_mgr.con.get = MagicMock(return_value=MagicMock( status_code=test_data[0], json=MagicMock(return_value=test_data[1]))) status_code, data = kube_container_mgr.get_host_status('10.2.0.2') self.assertEquals(test_data[0], status_code) self.assertEquals(test_data[1], data)