def test_exceptions(self):
# Check that we gracefully handle failed servers
server = MriServerDispatch({}, 'http://thiswontworkatall', 'test', 'tester')
result = server._send_request('/', 'GET', None)
self.assertEqual(result, None)
# Check that we fail gracefully on invalid urls
server = MriServerDispatch({}, 'http://744.255.255.1', 'test', 'tester')
result = server._send_request('/get', 'GET', None)
self.assertEqual(result, None)
# Check that we can detect 404s
server = MriServerDispatch({}, HTTP_BIN, 'test', 'tester')
result = server._send_request('/thisisntarealpath', 'GET', None)
self.assertEqual(result.status_code, 404)
def test_basic_requests(self):
# Check basic functionality
server = MriServerDispatch({}, HTTP_BIN, 'test', 'tester')
result = server._send_request('/get', 'GET', None)
self.assertEqual(result.status_code, 200)
# Check auth
server = MriServerDispatch({}, HTTP_BIN, 'test', 'tester')
result = server._send_request('/basic-auth/test/tester', 'GET', None)
self.assertEqual(result.status_code, 200)
# Check post
server = MriServerDispatch({}, HTTP_BIN, 'test', 'tester')
data = json.dumps({'introducing': 'kitton mittons'})
result = server._send_request('/post', 'POST', data)
self.assertEqual(result.status_code, 200)
def test_train_event(self):
# Check post
server = MriServerDispatch({}, HTTP_BIN, 'test', 'tester')
data = json.dumps(
{"type": "train.abcde", "properties": {"iteration": 100, "loss": 200, "accuracy": 300}}
)
result = server._send_request('/post', 'POST', data)
self.assertEqual(result.status_code, 200)
self.assertTrue('train.abcde' in result.text)
