def test_write_nodes(self, mock_open): args = mock.Mock() mock_open.return_value = mock.MagicMock() args.nodes_json = 'test.json' build_nodes_json._write_nodes(TEST_NODES, args) data = json.dumps({'nodes': TEST_NODES}, indent=2) f = mock_open.return_value.__enter__.return_value f.write.assert_called_once_with(data)
def test_write_nodes(self, mock_open): args = mock.Mock() mock_open.return_value = mock.MagicMock() args.nodes_json = 'test.json' build_nodes_json._write_nodes(TEST_NODES, args) data = json.dumps({'nodes': TEST_NODES}, indent=2) f = mock_open.return_value.__enter__.return_value f.write.assert_called_once_with(data)
def test_write_nodes_extra_node(self, mock_open): args = mock.Mock() mock_open.return_value = mock.MagicMock() args.nodes_json = 'test.json' args.network_details = True extra_nodes = [{'foo': 'bar'}] network_details = {'bar': 'baz'} build_nodes_json._write_nodes(TEST_NODES, extra_nodes, network_details, args) data = json.dumps( { 'nodes': TEST_NODES, 'extra_nodes': extra_nodes, 'network_details': network_details }, indent=2) f = mock_open.return_value.__enter__.return_value f.write.assert_called_once_with(data)