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