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