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