def test_get_existing_lookup(self):
"""
Getting an existing key that has completed returns a 'finished' status
and associated value.
"""
connector = HttpConnector(self.event_loop)
self.assertEqual({}, connector.lookups)
handler = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector,
1908)
faux_lookup = asyncio.Future()
faux_lookup.set_result('foo')
test_key = hashlib.sha512().hexdigest()
connector.lookups[test_key] = {
'last_access': 123.45,
'lookup': faux_lookup
}
result = connector.get(test_key, handler)
self.assertIn(test_key, connector.lookups)
# Check the last_access has been updated
self.assertTrue(connector.lookups[test_key]['last_access'] > 123.45)
self.assertEqual(connector.lookups[test_key]['lookup'], faux_lookup)
self.assertEqual(result['key'], test_key)
self.assertEqual(result['status'], faux_lookup._state.lower())
self.assertEqual(result['value'], 'foo')
self.assertEqual(3, len(result))
def test_handle_GET_no_cache(self):
"""
Ensure that if the cache-control header in the request is set to
no-cache then the lookup is foreced (i.e. don't use the local cache).
"""
test_key = hashlib.sha512().hexdigest()
mockMessage = mock.MagicMock()
mockMessage.method = 'GET'
mockMessage.version = '1.1'
mockMessage.path = ''.join(['/', test_key])
mockMessage.headers = {'Cache-Control': 'no-cache', }
connector = HttpConnector(self.event_loop)
def faux_get(*args, **kwargs):
return {
'key': test_key,
'status': 'pending',
}
connector.get = mock.MagicMock(side_effect=faux_get)
node = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector, 1908)
hrh = HttpRequestHandler(connector, node, debug=True)
peer = '192.168.0.1'
hrh.transport = mock.MagicMock()
hrh.transport.get_extra_info = mock.MagicMock(side_effect=peer)
hrh.writer = mock.MagicMock()
with mock.patch.object(aiohttp, 'Response',
return_value=mock.MagicMock()) as response:
self.event_loop.run_until_complete(hrh.handle_request(mockMessage,
None))
response.assert_called_once_with(hrh.writer, 200,
http_version=mockMessage.version)
# The connector's get method was called with the forced flag set
# to True.
connector.get.assert_called_once_with(test_key, node, True)
def test_handle_GET_request(self):
"""
A valid GET request casues a 200 reponse.
"""
test_key = hashlib.sha512().hexdigest()
mockMessage = mock.MagicMock()
mockMessage.method = 'GET'
mockMessage.version = '1.1'
mockMessage.path = ''.join(['/', test_key])
connector = HttpConnector(self.event_loop)
def faux_get(*args, **kwargs):
return {
'key': test_key,
'status': 'pending',
}
connector.get = mock.MagicMock(side_effect=faux_get)
node = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector, 1908)
hrh = HttpRequestHandler(connector, node, debug=True)
peer = '192.168.0.1'
hrh.transport = mock.MagicMock()
hrh.transport.get_extra_info = mock.MagicMock(side_effect=peer)
hrh.writer = mock.MagicMock()
with mock.patch.object(aiohttp, 'Response',
return_value=mock.MagicMock()) as response:
self.event_loop.run_until_complete(hrh.handle_request(mockMessage,
None))
response.assert_called_once_with(hrh.writer, 200,
http_version=mockMessage.version)
def test_handle_GET_internal_server_error(self):
"""
A GET request that causes an exception simply returns a 500 (Internal
Server Error).
"""
test_key = hashlib.sha512().hexdigest()
mockMessage = mock.MagicMock()
mockMessage.method = 'GET'
mockMessage.version = '1.1'
mockMessage.path = ''.join(['/', test_key])
connector = HttpConnector(self.event_loop)
def faux_get(*args, **kwargs):
raise Exception('Bang!')
connector.get = mock.MagicMock(side_effect=faux_get)
node = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector, 1908)
hrh = HttpRequestHandler(connector, node, debug=True)
peer = '192.168.0.1'
hrh.transport = mock.MagicMock()
hrh.transport.get_extra_info = mock.MagicMock(side_effect=peer)
hrh.writer = mock.MagicMock()
with mock.patch.object(aiohttp, 'Response',
return_value=mock.MagicMock()) as response:
self.event_loop.run_until_complete(hrh.handle_request(mockMessage,
None))
response.assert_called_once_with(hrh.writer, 500,
http_version=mockMessage.version)
def test_receive(self):
"""
The good case. Should return whatever handler.message_received
returns.
"""
connector = HttpConnector(self.event_loop)
ok = {
'uuid': str(uuid.uuid4()),
'recipient': PUBLIC_KEY,
'sender': PUBLIC_KEY,
'reply_port': 1908,
'version': self.version,
}
seal = get_seal(ok, PRIVATE_KEY)
ok['seal'] = seal
ok['message'] = 'ok'
raw = json.dumps(ok).encode('utf-8')
sender = '192.168.0.1'
handler = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector,
1908)
handler.message_received = mock.MagicMock()
connector.receive(raw, sender, handler)
msg = from_dict(ok)
handler.message_received.assert_called_once_with(msg, 'http',
sender,
msg.reply_port)
def test_handle_request_causes_exception(self):
"""
A request that raises an exception causes a 500 response.
"""
mockMessage = mock.MagicMock()
mockMessage.method = 'POST'
mockMessage.version = '1.1'
mockPayload = mock.MagicMock()
@asyncio.coroutine
def faux_read(*args, **kwargs):
return 'raw_data'
mockPayload.read = mock.MagicMock(side_effect=faux_read)
connector = HttpConnector(self.event_loop)
def faux_receive(*args, **kwargs):
raise ValueError('Boom! Something went wrong.')
connector.receive = mock.MagicMock(side_effect=faux_receive)
node = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector, 1908)
hrh = HttpRequestHandler(connector, node, debug=True)
peer = '192.168.0.1'
hrh.transport = mock.MagicMock()
hrh.transport.get_extra_info = mock.MagicMock(side_effect=peer)
hrh.writer = mock.MagicMock()
with mock.patch.object(aiohttp, 'Response',
return_value=mock.MagicMock()) as response:
self.event_loop.run_until_complete(hrh.handle_request(mockMessage,
mockPayload))
response.assert_called_once_with(hrh.writer, 500,
http_version=mockMessage.version)
def test_get_existing_lookup_failed(self):
"""
Getting an existing key that has resulted in an error returns a
'finished' status and an 'error' flag.
"""
connector = HttpConnector(self.event_loop)
self.assertEqual({}, connector.lookups)
handler = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector,
1908)
faux_lookup = asyncio.Future()
ex = Exception('Bang!')
faux_lookup.set_exception(ex)
test_key = hashlib.sha512().hexdigest()
connector.lookups[test_key] = {
'last_access': 123.45,
'lookup': faux_lookup
}
result = connector.get(test_key, handler)
self.assertIn(test_key, connector.lookups)
self.assertTrue(connector.lookups[test_key]['last_access'] > 123.45)
self.assertEqual(connector.lookups[test_key]['lookup'], faux_lookup)
self.assertEqual(result['key'], test_key)
self.assertEqual(result['status'], faux_lookup._state.lower())
self.assertEqual(result['error'], True)
self.assertEqual(3, len(result))
def test_sweep_and_clean_cache_schedules_again(self):
"""
Sweep and clean should schedule a new sweep and clean in
clean_interval seconds.
"""
event_loop = mock.MagicMock()
connector = HttpConnector(event_loop)
event_loop.call_later = mock.MagicMock()
connector._sweep_and_clean_cache(event_loop, 300)
sweep = connector._sweep_and_clean_cache
event_loop.call_later.assert_called_once_with(300, sweep, event_loop,
300)
def start_node(logfile, port):
"""
Starts a local drogulus node using throw away keys, logging to the
referenced directory and listening on the referenced port.
Return the Process encapsulating this node.
"""
handler = logging.FileHandler(logfile)
f = ' '.join([
'%(asctime)s', '%(processName)-10s', '%(name)s', '%(levelname)-8s',
'%(message)s'
])
formatter = logging.Formatter(f)
handler.setFormatter(formatter)
root = logging.getLogger()
root.addHandler(handler)
root.setLevel(logging.DEBUG)
root.info('Starting node in new process')
event_loop = asyncio.get_event_loop()
connector = HttpConnector(event_loop)
instance = Drogulus(PRIVATE_KEY, PUBLIC_KEY, event_loop, connector, port)
app = make_http_handler(event_loop, connector, instance._node)
f = event_loop.create_server(app, '0.0.0.0', port)
event_loop.run_until_complete(f)
try:
event_loop.run_forever()
except KeyboardInterrupt:
pass
def test_sweep_and_clean_cache_fresh_items_only(self):
"""
If the lookup cache is only full of fresh items then the cache remains
the same (nothing to delete).
"""
event_loop = mock.MagicMock()
connector = HttpConnector(event_loop)
event_loop.call_later = mock.MagicMock()
now = time.time()
for i in range(10):
connector.lookups[str(i)] = {
'last_access': now,
'lookup': mock.MagicMock()
}
connector._sweep_and_clean_cache(event_loop, 300)
self.assertEqual(len(connector.lookups), 10)
def start_node(logfile, port):
"""
Starts a local drogulus node using throw away keys, logging to the
referenced directory and listening on the referenced port.
Return the Process encapsulating this node.
"""
handler = logging.FileHandler(logfile)
f = ' '.join(['%(asctime)s', '%(processName)-10s', '%(name)s',
'%(levelname)-8s', '%(message)s'])
formatter = logging.Formatter(f)
handler.setFormatter(formatter)
root = logging.getLogger()
root.addHandler(handler)
root.setLevel(logging.DEBUG)
root.info('Starting node in new process')
event_loop = asyncio.get_event_loop()
connector = HttpConnector(event_loop)
instance = Drogulus(PRIVATE_KEY, PUBLIC_KEY, event_loop, connector, port)
def protocol_factory(connector=connector, node=instance._node):
"""
Returns an appropriately configured NetstringProtocol object for
each connection.
"""
return HttpRequestHandler(connector, node)
setup_server = event_loop.create_server(protocol_factory, port=port)
event_loop.run_until_complete(setup_server)
event_loop.run_forever()
def test_sweep_and_clean_cache_stale_items_only(self):
"""
If the lookup cache is only full of stale items then the cache should
be emtied.
"""
event_loop = mock.MagicMock()
connector = HttpConnector(event_loop)
event_loop.call_later = mock.MagicMock()
now = time.time()
for i in range(10):
connector.lookups[str(i)] = {
'last_access': now - 500,
'lookup': mock.MagicMock()
}
connector._sweep_and_clean_cache(event_loop, 300)
self.assertEqual(len(connector.lookups), 0)
def test_init_(self):
"""
The connector and node instances should be set properly.
"""
connector = HttpConnector(self.event_loop)
node = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector, 1908)
hrh = HttpRequestHandler(connector, node)
self.assertEqual(connector, hrh.connector)
self.assertEqual(node, hrh.node)
def test_sweep_and_clean_cache_mixed_fresh_and_stale_items(self):
"""
Only the stale items should be removed from the cache.
"""
event_loop = mock.MagicMock()
connector = HttpConnector(event_loop)
event_loop.call_later = mock.MagicMock()
now = time.time()
for i in range(10):
if i % 2:
access = now - 500
else:
access = now
connector.lookups[str(i)] = {
'last_access': access,
'lookup': mock.MagicMock()
}
connector._sweep_and_clean_cache(event_loop, 300)
self.assertEqual(len(connector.lookups), 5)
def test_init_with_extra_kwargs(self):
"""
An further arguments passed in (above and beyond the connector and node
instances) are correctly handled.
"""
connector = HttpConnector(self.event_loop)
node = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector, 1908)
hrh = HttpRequestHandler(connector, node, debug=True)
self.assertEqual(connector, hrh.connector)
self.assertEqual(node, hrh.node)
self.assertEqual(True, hrh.debug)
def test_send(self):
"""
Test the good case. We should end up with a task wrapping an
appropriate call to aiohttp.request.
"""
contact = PeerNode(PUBLIC_KEY, self.version, 'http://192.168.0.1:80')
msg = OK('uuid', 'recipient', 'sender', 9999, 'version', 'seal')
msg_json = json.dumps(to_dict(msg))
headers = {'content-type': 'application/json'}
connector = HttpConnector(self.event_loop)
@asyncio.coroutine
def faux_request(*args, **kwargs):
return 'foo'
with mock.patch.object(aiohttp, 'request',
return_value=faux_request()) as request:
result = connector.send(contact, msg)
self.assertIsInstance(result, asyncio.Task)
request.assert_called_once_with('post', contact.uri, data=msg_json,
headers=headers)
def test_get_forced_refresh_no_existing_cached_value(self):
"""
Ensures that even if there's no cached value a new lookup is executed
if the 'forced' flag is True.
"""
connector = HttpConnector(self.event_loop)
self.assertEqual({}, connector.lookups)
handler = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector,
1908)
faux_lookup = asyncio.Future()
handler.retrieve = mock.MagicMock(return_value=faux_lookup)
test_key = hashlib.sha512().hexdigest()
result = connector.get(test_key, handler, forced=True)
handler.retrieve.assert_called_once_with(test_key)
self.assertIn(test_key, connector.lookups)
self.assertIsInstance(connector.lookups[test_key]['last_access'],
float)
self.assertEqual(connector.lookups[test_key]['lookup'], faux_lookup)
self.assertEqual(result['key'], test_key)
self.assertEqual(result['status'], faux_lookup._state.lower())
self.assertEqual(2, len(result))
def test_get_new_lookup(self):
"""
Getting an unknown key fires a new lookup that is initially produces a
'pending' status.
"""
connector = HttpConnector(self.event_loop)
self.assertEqual({}, connector.lookups)
handler = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector,
1908)
faux_lookup = asyncio.Future()
handler.retrieve = mock.MagicMock(return_value=faux_lookup)
test_key = hashlib.sha512().hexdigest()
result = connector.get(test_key, handler)
handler.retrieve.assert_called_once_with(test_key)
self.assertIn(test_key, connector.lookups)
self.assertIsInstance(connector.lookups[test_key]['last_access'],
float)
self.assertEqual(connector.lookups[test_key]['lookup'], faux_lookup)
self.assertEqual(result['key'], test_key)
self.assertEqual(result['status'], faux_lookup._state.lower())
self.assertEqual(2, len(result))
def test_init(self):
"""
Ensure the instance has an empty lookups dict assigned and a sweep and
clean of the lookups cache is scheduled after X seconds.
"""
event_loop = mock.MagicMock()
event_loop.call_later = mock.MagicMock()
connector = HttpConnector(event_loop)
self.assertEqual({}, connector.lookups)
self.assertEqual(event_loop, connector.event_loop)
sweep = connector._sweep_and_clean_cache
event_loop.call_later.assert_called_once_with(300, sweep, event_loop,
300)
def test_init_with_bespoke_clean_interval(self):
"""
Ensures that a custom clean_interval value is used when scheduling the
_sweep_and_clean_cache method.
"""
event_loop = mock.MagicMock()
event_loop.call_later = mock.MagicMock()
connector = HttpConnector(event_loop, 900)
self.assertEqual({}, connector.lookups)
self.assertEqual(event_loop, connector.event_loop)
sweep = connector._sweep_and_clean_cache
event_loop.call_later.assert_called_once_with(900, sweep, event_loop,
900)
def test_receive_not_json(self):
"""
Appropriately handle a message that doesn't contain JSON.
"""
patcher = mock.patch('drogulus.net.http.log.error')
connector = HttpConnector(self.event_loop)
raw = "junk from the network".encode('utf-8')
sender = '192.168.0.1'
handler = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector,
1908)
mock_log = patcher.start()
self.assertRaises(ValueError, connector.receive, raw, sender, handler)
self.assertEqual(4, mock_log.call_count)
patcher.stop()
def test_get_forced_refresh_existing_value(self):
"""
Ensures that an existing result is ignored and a new lookup is executed
if the 'forced' flag is True.
"""
connector = HttpConnector(self.event_loop)
self.assertEqual({}, connector.lookups)
handler = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector,
1908)
cached_lookup = asyncio.Future()
cached_lookup.set_result('foo')
test_key = hashlib.sha512().hexdigest()
connector.lookups[test_key] = cached_lookup
new_lookup = asyncio.Future()
handler.retrieve = mock.MagicMock(return_value=new_lookup)
result = connector.get(test_key, handler, forced=True)
self.assertIn(test_key, connector.lookups)
self.assertIsInstance(connector.lookups[test_key]['last_access'],
float)
self.assertEqual(connector.lookups[test_key]['lookup'], new_lookup)
self.assertEqual(result['key'], test_key)
self.assertEqual(result['status'], new_lookup._state.lower())
self.assertEqual(2, len(result))
def test_handle_POST_request(self):
"""
A valid POST request causes a 200 response.
* WARNING * Too much mocking going on here (in the vain attempt to
achieve 100% test coverage).
"""
mockMessage = mock.MagicMock()
mockMessage.method = 'POST'
mockMessage.version = '1.1'
mockPayload = mock.MagicMock()
@asyncio.coroutine
def faux_read(*args, **kwargs):
return 'raw_data'
mockPayload.read = mock.MagicMock(side_effect=faux_read)
connector = HttpConnector(self.event_loop)
def faux_receive(*args, **kwargs):
return OK('uuid', 'recipient', 'sender', 9999, 'version', 'seal')
connector.receive = mock.MagicMock(side_effect=faux_receive)
node = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector, 1908)
hrh = HttpRequestHandler(connector, node, debug=True)
peer = '192.168.0.1'
hrh.transport = mock.MagicMock()
hrh.transport.get_extra_info = mock.MagicMock(side_effect=peer)
hrh.writer = mock.MagicMock()
with mock.patch.object(aiohttp, 'Response',
return_value=mock.MagicMock()) as response:
self.event_loop.run_until_complete(hrh.handle_request(mockMessage,
mockPayload))
response.assert_called_once_with(hrh.writer, 200,
http_version=mockMessage.version)
def test_handle_request_not_POST_or_GET(self):
"""
A request that is not a POST causes a 405 response.
"""
mockMessage = mock.MagicMock()
mockMessage.method = 'PUT'
mockMessage.version = '1.1'
mockPayload = mock.MagicMock()
connector = HttpConnector(self.event_loop)
node = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector, 1908)
hrh = HttpRequestHandler(connector, node, debug=True)
hrh.writer = mock.MagicMock()
with mock.patch.object(aiohttp, 'Response',
return_value=mock.MagicMock()) as response:
self.event_loop.run_until_complete(hrh.handle_request(mockMessage,
mockPayload))
response.assert_called_once_with(hrh.writer, 405,
http_version=mockMessage.version)
def test_handle_GET_bad_request(self):
"""
A GET request without a valid sha512 hexdigest as its path causes a
400 (Bad Request) response.
"""
test_key = 'not_a_valid_sha512_hexdigest'
mockMessage = mock.MagicMock()
mockMessage.method = 'GET'
mockMessage.version = '1.1'
mockMessage.path = ''.join(['/', test_key])
connector = HttpConnector(self.event_loop)
node = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector, 1908)
hrh = HttpRequestHandler(connector, node, debug=True)
hrh.writer = mock.MagicMock()
with mock.patch.object(aiohttp, 'Response',
return_value=mock.MagicMock()) as response:
self.event_loop.run_until_complete(hrh.handle_request(mockMessage,
None))
response.assert_called_once_with(hrh.writer, 400,
http_version=mockMessage.version)
def test_receive_bad_message(self):
"""
Appropriately handle a message that is valid JSON but not a valid
message type understood as part of the drogulus protocol.
"""
patcher = mock.patch('drogulus.net.http.log.error')
connector = HttpConnector(self.event_loop)
ok = {
'uuid': str(uuid.uuid4()),
'recipient': PUBLIC_KEY,
'sender': PUBLIC_KEY,
'reply_port': 1908,
'version': self.version,
}
raw = json.dumps(ok).encode('utf-8')
sender = '192.168.0.1'
handler = Node(PUBLIC_KEY, PRIVATE_KEY, self.event_loop, connector,
1908)
mock_log = patcher.start()
self.assertRaises(KeyError, connector.receive, raw, sender, handler)
self.assertEqual(4, mock_log.call_count)
patcher.stop()
