add tests
This commit is contained in:
Timofey Turenko
committed by
Markus Mäkelä
Markus Mäkelä
parent
dbfd631fed
commit
8c6ca38a8a
258
maxscale-system-test/sql_t1.cpp
Normal file
258
maxscale-system-test/sql_t1.cpp
Normal file
@ -0,0 +1,258 @@
|
||||
#include "sql_t1.h"
|
||||
|
||||
#include <pthread.h>
|
||||
|
||||
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
|
||||
static char** sql = NULL;
|
||||
static size_t sql_size = 0;
|
||||
|
||||
int execute_select_query_and_check(MYSQL *conn, char *sql, unsigned long long int rows)
|
||||
{
|
||||
MYSQL_RES *res;
|
||||
MYSQL_ROW row;
|
||||
unsigned long long int i;
|
||||
unsigned long long int num_fields;
|
||||
unsigned long long int int_res;
|
||||
unsigned long long int row_i = 0;
|
||||
int test_result = 0;
|
||||
unsigned long long int rows_from_select = 0;
|
||||
int wait_i = 0;
|
||||
|
||||
printf("Trying SELECT, num_of_rows=%llu\n", rows);
|
||||
int res_alloc = 0;
|
||||
if (conn != NULL)
|
||||
{
|
||||
rows_from_select = 0;
|
||||
wait_i = 0;
|
||||
while ((rows_from_select != rows) && (wait_i < 100))
|
||||
{
|
||||
if (mysql_query(conn, sql) != 0)
|
||||
{
|
||||
printf("Error: can't execute SQL-query: %s\n", mysql_error(conn));
|
||||
}
|
||||
|
||||
res = mysql_store_result(conn);
|
||||
res_alloc = 1;
|
||||
if (res == NULL)
|
||||
{
|
||||
printf("Error: can't get the result description\n");
|
||||
test_result = 1;
|
||||
mysql_free_result(res);
|
||||
res_alloc = 0;
|
||||
wait_i++;
|
||||
sleep(1);
|
||||
}
|
||||
else
|
||||
{
|
||||
rows_from_select = mysql_num_rows(res);
|
||||
printf("rows=%llu\n", rows_from_select);
|
||||
wait_i++;
|
||||
if (rows_from_select != rows)
|
||||
{
|
||||
printf("Waiting 1 second and trying again...\n");
|
||||
mysql_free_result(res);
|
||||
res_alloc = 0;
|
||||
sleep(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (rows_from_select != rows)
|
||||
{
|
||||
printf("SELECT returned %llu rows instead of %llu!\n", rows_from_select, rows);
|
||||
test_result = 1;
|
||||
printf("sql was %s\n", sql);
|
||||
}
|
||||
else
|
||||
{
|
||||
num_fields = mysql_num_fields(res);
|
||||
if (num_fields != 2)
|
||||
{
|
||||
printf("SELECT returned %llu fileds instead of 2!\n", num_fields);
|
||||
test_result = 1;
|
||||
}
|
||||
if (mysql_num_rows(res) > 0)
|
||||
{
|
||||
while ((row = mysql_fetch_row(res)) != NULL)
|
||||
{
|
||||
for (i = 0; i < num_fields; i++)
|
||||
{
|
||||
sscanf(row[i], "%llu", &int_res);
|
||||
if ((i == 0 ) && (int_res != row_i))
|
||||
{
|
||||
printf("SELECT returned wrong result! %llu instead of expected %llu\n", int_res, row_i);
|
||||
test_result = 1;
|
||||
printf("sql was %s\n", sql);
|
||||
}
|
||||
}
|
||||
row_i++;
|
||||
}
|
||||
}
|
||||
}
|
||||
if (res_alloc != 0)
|
||||
{
|
||||
mysql_free_result(res);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("FAILED: broken connection\n");
|
||||
test_result = 1;
|
||||
}
|
||||
|
||||
return test_result;
|
||||
}
|
||||
int create_t1(MYSQL * conn)
|
||||
{
|
||||
int result = 0;
|
||||
result += execute_query(conn, "DROP TABLE IF EXISTS t1;");
|
||||
printf("Creating test table\n");
|
||||
result += execute_query(conn, "CREATE TABLE t1 (x1 int, fl int);");
|
||||
return result;
|
||||
}
|
||||
|
||||
int create_t2(MYSQL * conn)
|
||||
{
|
||||
int result = 0;
|
||||
result += execute_query(conn, "DROP TABLE IF EXISTS t2;");
|
||||
printf("Creating test table\n");
|
||||
result += execute_query(conn, "CREATE TABLE t2 (x1 int, fl int);");
|
||||
return result;
|
||||
}
|
||||
|
||||
static const char ins1[] = "INSERT INTO t1 (x1, fl) VALUES ";
|
||||
|
||||
int create_insert_string(char* sql, int N, int fl)
|
||||
{
|
||||
char *wptr = sql;
|
||||
|
||||
strcpy(wptr, ins1);
|
||||
for (int i = 0; i < N; i++)
|
||||
{
|
||||
wptr = strchr(wptr, '\0');
|
||||
sprintf(wptr, "(%d, %d),", i, fl);
|
||||
}
|
||||
wptr = strrchr(wptr, ',');
|
||||
sprintf(wptr, ";");
|
||||
return 0;
|
||||
}
|
||||
|
||||
char* allocate_insert_string(int fl, int N)
|
||||
{
|
||||
char* rval = NULL;
|
||||
|
||||
pthread_mutex_lock(&mutex);
|
||||
|
||||
if (sql == NULL)
|
||||
{
|
||||
sql = (char**)calloc(16, sizeof(char*));
|
||||
sql_size = 16;
|
||||
}
|
||||
|
||||
if (fl >= sql_size)
|
||||
{
|
||||
fprintf(stderr, "Insert index %d is too large, setting it to %lu", fl, sql_size - 1);
|
||||
fl = sql_size - 1;
|
||||
}
|
||||
|
||||
if (sql[fl] == NULL)
|
||||
{
|
||||
char tmpstr[256];
|
||||
sprintf(tmpstr, "(%d, %d),", N, fl);
|
||||
sql[fl] = (char*)malloc(sizeof(ins1) + N * strlen(tmpstr) + 60);
|
||||
create_insert_string(sql[fl], N, fl);
|
||||
}
|
||||
|
||||
rval = sql[fl];
|
||||
pthread_mutex_unlock(&mutex);
|
||||
|
||||
return rval;
|
||||
}
|
||||
|
||||
int insert_into_t1(MYSQL *conn, int N)
|
||||
{
|
||||
|
||||
int x = 16;
|
||||
int result = 0;
|
||||
|
||||
printf("Generating long INSERTs\n");
|
||||
for (int i = 0; i < N; i++)
|
||||
{
|
||||
printf("sql %d, rows=%d\n", i, x);
|
||||
char *sqlstr = allocate_insert_string(i, x);
|
||||
printf("INSERT: rwsplitter\n");
|
||||
printf("Trying INSERT, len=%d\n", x);
|
||||
fflush(stdout);
|
||||
result += execute_query(conn, sqlstr);
|
||||
fflush(stdout);
|
||||
x *= 16;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
int select_from_t1(MYSQL *conn, int N)
|
||||
{
|
||||
int x = 16;
|
||||
int result = 0;
|
||||
int i;
|
||||
char sq[100];
|
||||
|
||||
for (i = 0; i < N; i++)
|
||||
{
|
||||
sprintf(&sq[0], "select * from t1 where fl=%d;", i);
|
||||
result += execute_select_query_and_check(conn, sq, x);
|
||||
x = x * 16;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
// 0 - if it does not exist
|
||||
// -1 - in case of error
|
||||
int check_if_t1_exists(MYSQL *conn)
|
||||
{
|
||||
MYSQL_RES *res;
|
||||
MYSQL_ROW row;
|
||||
unsigned long long int num_fields;
|
||||
|
||||
int t1 = 0;
|
||||
if (conn != NULL)
|
||||
{
|
||||
if (mysql_query(conn, "show tables;") != 0)
|
||||
{
|
||||
printf("Error: can't execute SQL-query: %s\n", mysql_error(conn));
|
||||
t1 = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
res = mysql_store_result(conn);
|
||||
if (res == NULL)
|
||||
{
|
||||
printf("Error: can't get the result description\n");
|
||||
t1 = - 1;
|
||||
}
|
||||
else
|
||||
{
|
||||
num_fields = mysql_num_fields(res);
|
||||
if (mysql_num_rows(res) > 0)
|
||||
{
|
||||
while ((row = mysql_fetch_row(res)) != NULL)
|
||||
{
|
||||
if ( (row[0] != NULL ) && (strcmp(row[0], "t1") == 0 ) )
|
||||
{
|
||||
t1 = 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
mysql_free_result(res);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("FAILED: broken connection\n");
|
||||
t1 = -1;
|
||||
}
|
||||
return t1;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user