def test_is_in_maintenance(self, mock_pyhorn): controller = MatterhornController('http://mh.example.edu') controller._hosts = [ Mock(base_url='foo', maintenance=False), Mock(base_url='bar', maintenance=True) ] self.assertFalse(controller.is_in_maintenance(Mock(mh_host_url="foo"))) self.assertTrue(controller.is_in_maintenance(Mock(mh_host_url="bar")))
def test_get_host(self, mock_pyhorn): controller = MatterhornController('http://mh.example.edu') controller._hosts = [ Mock(id=1, base_url='foo'), Mock(id=2, base_url='bar') ] self.assertEqual(controller.get_host(Mock(mh_host_url="foo")).id, 1) self.assertEqual(controller.get_host(Mock(mh_host_url="bar")).id, 2)
def test_is_registered(self, mock_pyhorn): controller = MatterhornController('http://mh.example.edu') controller._hosts = [ Mock(base_url='foo'), Mock(base_url='bar') ] self.assertTrue(controller.is_registered(Mock(mh_host_url="bar"))) self.assertFalse(controller.is_registered(Mock(mh_host_url="blerg")))