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