def test_block_error(self):
with client_ctx(Network, self.client):
self.client.block_node.return_value = False, 'error_msg'
network = Network()
result = network.block('node_id')
self.assertEqual(result, 'error_msg')
self.client.block_node.assert_called_once_with('node_id')
def test_dht(self):
with client_ctx(Network, self.client):
net = Network()
result_1 = net.dht(None, full=False)
result_2 = net.dht(None, full=True)
self.__assert_peer_result(result_1, result_2)
def test_connect(self):
with client_ctx(Network, self.client):
with self.assertRaises(CommandException):
Network().connect('266.266.0.1', '25000')
assert not self.client.connect.called
assert Network().connect('127.0.0.1', '25000') is None
assert self.client.connect.called
def test_status(self):
with client_ctx(Network, self.client):
self.client.connection_status.return_value = 'Status'
result = Network().status()
assert self.client.connection_status.called
assert isinstance(result, basestring)
assert result
assert result != 'unknown'
self.client.connection_status.return_value = None
result = Network().status()
assert isinstance(result, basestring)
assert result == 'unknown'
def test_status(self):
with client_ctx(Network, self.client):
# given
msg = "Some random message"
self.client.connection_status.return_value = {
'msg': msg,
}
# when
result = Network().status()
# then
assert self.client.connection_status.called
assert isinstance(result, str)
assert result == msg
def test_block_success(self):
with client_ctx(Network, self.client):
self.client.block_node.return_value = True, None
network = Network()
network.block('node_id')
self.client.block_node.assert_called_once_with('node_id')