MaxScale/maxscale-system-test/sharding_load_data.cpp
2017-12-08 12:23:04 +02:00

54 lines
1.4 KiB
C++

/**
* MXS-1160: LOAD DATA LOCAL INFILE with schemarouter
*/
#include "testconnections.h"
#include <fstream>
#include <sstream>
#include <string>
int main(int argc, char** argv)
{
// Create a CSV file
unlink("data.csv");
std::ofstream output("data.csv");
std::stringstream ss;
for (int i = 0; i < 100; i++)
{
ss << i << "\n";
}
output << ss.str();
output.close();
TestConnections test(argc, argv);
test.repl->execute_query_all_nodes("DROP DATABASE db1");
test.repl->connect();
execute_query(test.repl->nodes[0], "CREATE DATABASE db1");
execute_query(test.repl->nodes[0], "CREATE TABLE db1.t1(id INT)");
test.maxscales->connect_maxscale(0);
test.tprintf("Loading local data file");
test.try_query(test.maxscales->conn_rwsplit[0], "LOAD DATA LOCAL INFILE 'data.csv' INTO TABLE db1.t1");
test.tprintf("Verifying that data was loaded");
long total = execute_query_count_rows(test.maxscales->conn_rwsplit[0], "SELECT * FROM db1.t1");
test.add_result(total != 100, "Expected 100 rows, got %ld", total);
test.tprintf("Dropping tables and databases");
test.try_query(test.maxscales->conn_rwsplit[0], "DROP TABLE db1.t1");
test.try_query(test.maxscales->conn_rwsplit[0], "DROP DATABASE db1");
test.maxscales->close_maxscale_connections(0);
// Remove the test data
unlink("data.csv");
return test.global_result;
}