def test_should_return_all_backends(self, now_mock):
now_mock.return_value = datetime(2011, 10, 0o1)
now = '2011-09-30T23:00:00Z'
state = ServerState()
state.update_backend('varnish1', 'web2', 'healthy', '200', 'OK')
state.update_backend('varnish1', 'web1', 'healthy', '200', 'OK')
backends = state.get_backends('varnish1')
expected = {
'backends': [{
'name': 'web2',
'state': 'healthy',
'status_code': '200',
'status_text': 'OK',
'timestamp': now
}, {
'name': 'web1',
'state': 'healthy',
'status_code': '200',
'status_text': 'OK',
'timestamp': now
}]
}
self.assertEquals(expected, backends)
def test_should_return_empty_dict_for_unknown_backend(self):
state = ServerState()
state.update_backend('varnish1', 'web2', 'healthy', '200', 'OK')
backend_state = state.get_backend('varnish1', 'web1')
self.assertEquals({}, backend_state)
def test_should_update_backend_state(self, now_mock):
now_mock.return_value = datetime(2011, 10, 0o1)
state = ServerState()
state.update_backend('varnish1', 'web2', 'healthy', '200', 'OK')
backend_state = state.get_backend('varnish1', 'web2')
self.assertEquals(
{
'name': 'web2',
'state': 'healthy',
'status_code': '200',
'status_text': 'OK',
'timestamp': '2011-09-30T23:00:00Z'
}, backend_state)
