hcq/MaxScale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
5fccce5bb4a5059937f79fe92d7210fcd07bdabd
MaxScale/server/core/test/test_session_track.cc
Niclas Antti c447e5cf15 Uncrustify maxscale 
See script directory for method. The script to run in the top level
MaxScale directory is called maxscale-uncrustify.sh, which uses
another script, list-src, from the same directory (so you need to set
your PATH). The uncrustify version was 0.66.
2018-09-09 22:26:19 +03:00

250 lines
10 KiB
C++
Raw Blame History

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <maxscale/alloc.h>
#include <maxscale/buffer.h>
#include <maxscale/protocol/mysql.h>
MySQLProtocol proto;
static const uint8_t resultset1[] =
{
/* BEGIN;*/
0x29, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x03,
0x40, 0x00, 0x00, 0x00, 0x20, 0x05, 0x09, 0x08,
0x54, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F,
0x04, 0x13, 0x12, 0x53, 0x54, 0x41, 0x52, 0x54,
0x20, 0x54, 0x52, 0x41, 0x4E, 0x53, 0x41, 0x43,
0x54, 0x49, 0x4F, 0x4E, 0x3B,
/* COMMIT;*/
0x17, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02,
0x40, 0x00, 0x00, 0x00, 0x0E, 0x05, 0x09, 0x08,
0x5F, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F,
0x04, 0x01, 0x00,
/* START TRANSACTION;*/
0x29, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x03,
0x40, 0x00, 0x00, 0x00, 0x20, 0x05, 0x09, 0x08,
0x54, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F,
0x04, 0x13, 0x12, 0x53, 0x54, 0x41, 0x52, 0x54,
0x20, 0x54, 0x52, 0x41, 0x4E, 0x53, 0x41, 0x43,
0x54, 0x49, 0x4F, 0x4E, 0x3B,
/* START TRANSACTION READ ONLY;*/
0x28, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x03,
0x60, 0x00, 0x00, 0x00, 0x1F, 0x04, 0x1D, 0x1C,
0x53, 0x54, 0x41, 0x52, 0x54, 0x20, 0x54, 0x52,
0x41, 0x4E, 0x53, 0x41, 0x43, 0x54, 0x49, 0x4F,
0x4E, 0x20, 0x52, 0x45, 0x41, 0x44, 0x20, 0x4F,
0x4E, 0x4C, 0x59, 0x3B,
/* COMMIT;*/
0x07, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02,
0x00, 0x00, 0x00,
/* SET AUTOCOMMIT=0;*/
0x1D, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00,
0x40, 0x00, 0x00, 0x00, 0x14, 0x00, 0x0F, 0x0A,
0x61, 0x75, 0x74, 0x6F, 0x63, 0x6F, 0x6D, 0x6D,
0x69, 0x74, 0x03, 0x4F, 0x46, 0x46, 0x02, 0x01,
0x31,
/* INSERT INTO t1 VALUES(1);*/
0x14, 0x00, 0x00, 0x01, 0x00, 0x01, 0x00, 0x01,
0x40, 0x00, 0x00, 0x00, 0x0B, 0x05, 0x09, 0x08,
0x49, 0x5F, 0x5F, 0x5F, 0x57, 0x5F, 0x5F, 0x5F,
/* COMMIT;*/
0x14, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00,
0x40, 0x00, 0x00, 0x00, 0x0B, 0x05, 0x09, 0x08,
0x5F, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F, 0x5F
};
#define PACKET_HDR_LEN 4
#define PACKET_1_IDX 0
#define PACKET_1_LEN (PACKET_HDR_LEN + 0x29)
#define PACKET_2_IDX (PACKET_1_IDX + PACKET_1_LEN)
#define PACKET_2_LEN (PACKET_HDR_LEN + 0x17)
#define PACKET_3_IDX (PACKET_2_IDX + PACKET_2_LEN)
#define PACKET_3_LEN (PACKET_HDR_LEN + 0x29)
#define PACKET_4_IDX (PACKET_3_IDX + PACKET_3_LEN)
#define PACKET_4_LEN (PACKET_HDR_LEN + 0x28)
#define PACKET_5_IDX (PACKET_4_IDX + PACKET_4_LEN)
#define PACKET_5_LEN (PACKET_HDR_LEN + 0x07)
#define PACKET_6_IDX (PACKET_5_IDX + PACKET_5_LEN)
#define PACKET_6_LEN (PACKET_HDR_LEN + 0x1D)
#define PACKET_7_IDX (PACKET_6_IDX + PACKET_6_LEN)
#define PACKET_7_LEN (PACKET_HDR_LEN + 0x14)
#define PACKET_8_IDX (PACKET_7_IDX + PACKET_7_LEN)
#define PACKET_8_LEN (PACKET_HDR_LEN + 0x14)
/* multi statments;*/
static const uint8_t resultset2[] =
{
/**
* set autocommit=0;
* create table t1(id int);
* insert into t1 select seq from seq_0_to_20;
* select '' from t1;
**/
0x1D, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00,
0x40, 0x00, 0x00, 0x00, 0x14, 0x00, 0x0F, 0x0A,
0x61, 0x75, 0x74, 0x6F, 0x63, 0x6F, 0x6D, 0x6D,
0x69, 0x74, 0x03, 0x4F, 0x46, 0x46, 0x02, 0x01,
0x31, 0x07, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x3B, 0x00, 0x00, 0x01,
0x00, 0x15, 0x00, 0x21, 0x40, 0x00, 0x00, 0x27,
0x52, 0x65, 0x63, 0x6F, 0x72, 0x64, 0x73, 0x3A,
0x20, 0x32, 0x31, 0x20, 0x20, 0x44, 0x75, 0x70,
0x6C, 0x69, 0x63, 0x61, 0x74, 0x65, 0x73, 0x3A,
0x20, 0x30, 0x20, 0x20, 0x57, 0x61, 0x72, 0x6E,
0x69, 0x6E, 0x67, 0x73, 0x3A, 0x20, 0x30, 0x0B,
0x05, 0x09, 0x08, 0x49, 0x5F, 0x52, 0x5F, 0x57,
0x5F, 0x5F, 0x5F, 0x01, 0x00, 0x00, 0x01, 0x01,
0x16, 0x00, 0x00, 0x02, 0x03, 0x64, 0x65, 0x66,
0x00, 0x00, 0x00, 0x00, 0x00, 0x0C, 0x21, 0x00,
0x00, 0x00, 0x00, 0x00, 0xFD, 0x01, 0x00, 0x27,
0x00, 0x00, 0x05, 0x00, 0x00, 0x03, 0xFE, 0x00,
0x00, 0x21, 0x00, 0x01, 0x00, 0x00, 0x04, 0x00,
0x01, 0x00, 0x00, 0x05, 0x00, 0x01, 0x00, 0x00,
0x06, 0x00, 0x01, 0x00, 0x00, 0x07, 0x00, 0x01,
0x00, 0x00, 0x08, 0x00, 0x01, 0x00, 0x00, 0x09,
0x00, 0x01, 0x00, 0x00, 0x0A, 0x00, 0x01, 0x00,
0x00, 0x0B, 0x00, 0x01, 0x00, 0x00, 0x0C, 0x00,
0x01, 0x00, 0x00, 0x0D, 0x00, 0x01, 0x00, 0x00,
0x0E, 0x00, 0x01, 0x00, 0x00, 0x0F, 0x00, 0x01,
0x00, 0x00, 0x10, 0x00, 0x01, 0x00, 0x00, 0x11,
0x00, 0x01, 0x00, 0x00, 0x12, 0x00, 0x01, 0x00,
0x00, 0x13, 0x00, 0x01, 0x00, 0x00, 0x14, 0x00,
0x01, 0x00, 0x00, 0x15, 0x00, 0x01, 0x00, 0x00,
0x16, 0x00, 0x01, 0x00, 0x00, 0x17, 0x00, 0x01,
0x00, 0x00, 0x18, 0x00, 0x05, 0x00, 0x00, 0x19,
0xFE, 0x00, 0x00, 0x21, 0x40
};
/**
* SET AUTOCOMMIT=0;
* CREATE TABLE t1(a VARCHAR(20), b INT, c INT, d INT);
* INSERT INTO t1 VALUES ('', 100, 200, 300);
* SELECT * FROM t1;
**/
static const uint8_t resultset3[] =
{
0x28, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01,
0x40, 0x00, 0x00, 0x00, 0x1F, 0x00, 0x0F, 0x0A,
0x61, 0x75, 0x74, 0x6F, 0x63, 0x6F, 0x6D, 0x6D,
0x69, 0x74, 0x03, 0x4F, 0x46, 0x46, 0x02, 0x01,
0x31, 0x05, 0x09, 0x08, 0x49, 0x5F, 0x52, 0x5F,
0x5F, 0x5F, 0x53, 0x5F, 0x1D, 0x00, 0x00, 0x01,
0x00, 0x00, 0x00, 0x00, 0x40, 0x00, 0x00, 0x00,
0x14, 0x00, 0x0F, 0x0A, 0x61, 0x75, 0x74, 0x6F,
0x63, 0x6F, 0x6D, 0x6D, 0x69, 0x74, 0x03, 0x4F,
0x46, 0x46, 0x02, 0x01, 0x31, 0x07, 0x00, 0x00,
0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x14, 0x00, 0x00, 0x01, 0x00, 0x01, 0x00, 0x01,
0x40, 0x00, 0x00, 0x00, 0x0B, 0x05, 0x09, 0x08,
0x49, 0x5F, 0x5F, 0x5F, 0x57, 0x5F, 0x5F, 0x5F,
0x01, 0x00, 0x00, 0x01, 0x04, 0x20, 0x00, 0x00,
0x02, 0x03, 0x64, 0x65, 0x66, 0x04, 0x74, 0x65,
0x73, 0x74, 0x02, 0x74, 0x31, 0x02, 0x74, 0x31,
0x01, 0x61, 0x01, 0x61, 0x0C, 0x21, 0x00, 0x3C,
0x00, 0x00, 0x00, 0xFD, 0x00, 0x00, 0x00, 0x00,
0x00, 0x20, 0x00, 0x00, 0x03, 0x03, 0x64, 0x65,
0x66, 0x04, 0x74, 0x65, 0x73, 0x74, 0x02, 0x74,
0x31, 0x02, 0x74, 0x31, 0x01, 0x62, 0x01, 0x62,
0x0C, 0x3F, 0x00, 0x0B, 0x00, 0x00, 0x00, 0x03,
0x00, 0x00, 0x00, 0x00, 0x00, 0x20, 0x00, 0x00,
0x04, 0x03, 0x64, 0x65, 0x66, 0x04, 0x74, 0x65,
0x73, 0x74, 0x02, 0x74, 0x31, 0x02, 0x74, 0x31,
0x01, 0x63, 0x01, 0x63, 0x0C, 0x3F, 0x00, 0x0B,
0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00, 0x00,
0x00, 0x20, 0x00, 0x00, 0x05, 0x03, 0x64, 0x65,
0x66, 0x04, 0x74, 0x65, 0x73, 0x74, 0x02, 0x74,
0x31, 0x02, 0x74, 0x31, 0x01, 0x64, 0x01, 0x64,
0x0C, 0x3F, 0x00, 0x0B, 0x00, 0x00, 0x00, 0x03,
0x00, 0x00, 0x00, 0x00, 0x00, 0x05, 0x00, 0x00,
0x06, 0xFE, 0x00, 0x00, 0x21, 0x00, 0x05, 0x00,
0x00, 0x07, 0xFE, 0x00, 0x00, 0x21, 0x00
};
/* functional test , test packet by packet */
void test1()
{
GWBUF* buffer;
proto.server_capabilities = GW_MYSQL_CAPABILITIES_SESSION_TRACK;
proto.num_eof_packets = 0;
fprintf(stderr, "test_session_track : Functional tests.\n");
// BEGIN
buffer = gwbuf_alloc_and_load(PACKET_1_LEN, resultset1 + PACKET_1_IDX);
mxs_mysql_get_session_track_info(buffer, &proto);
mxb_assert(strncmp(gwbuf_get_property(buffer, (char*)"trx_state"), "T_______", 8) == 0);
gwbuf_free(buffer);
// COMMIT
buffer = gwbuf_alloc_and_load(PACKET_2_LEN, resultset1 + PACKET_2_IDX);
mxs_mysql_get_session_track_info(buffer, &proto);
mxb_assert(strncmp(gwbuf_get_property(buffer, (char*)"trx_state"), "________", 8) == 0);
gwbuf_free(buffer);
// START TRANSACTION
buffer = gwbuf_alloc_and_load(PACKET_3_LEN, resultset1 + PACKET_3_IDX);
mxs_mysql_get_session_track_info(buffer, &proto);
mxb_assert(strncmp(gwbuf_get_property(buffer, (char*)"trx_state"), "T_______", 8) == 0);
gwbuf_free(buffer);
// START TRANSACTION READ ONLY
buffer = gwbuf_alloc_and_load(PACKET_4_LEN, resultset1 + PACKET_4_IDX);
mxs_mysql_get_session_track_info(buffer, &proto);
mxb_assert(strncmp(gwbuf_get_property(buffer, (char*)"trx_characteristics"),
"START TRANSACTION READ ONLY;",
28) == 0);
gwbuf_free(buffer);
// COMMIT
buffer = gwbuf_alloc_and_load(PACKET_5_LEN, resultset1 + PACKET_5_IDX);
mxs_mysql_get_session_track_info(buffer, &proto);
mxb_assert(gwbuf_get_property(buffer, (char*)"trx_characteristics") == NULL);
mxb_assert(gwbuf_get_property(buffer, (char*)"trx_state") == NULL);
gwbuf_free(buffer);
// SET AUTOCOMMIT=0;
buffer = gwbuf_alloc_and_load(PACKET_6_LEN, resultset1 + PACKET_6_IDX);
mxs_mysql_get_session_track_info(buffer, &proto);
mxb_assert(strncmp(gwbuf_get_property(buffer, (char*)"autocommit"), "OFF", 3) == 0);
gwbuf_free(buffer);
// INSERT INTO t1 VALUES(1);
buffer = gwbuf_alloc_and_load(PACKET_7_LEN, resultset1 + PACKET_7_IDX);
mxs_mysql_get_session_track_info(buffer, &proto);
mxb_assert(strncmp(gwbuf_get_property(buffer, (char*)"trx_state"), "I___W___", 8) == 0);
gwbuf_free(buffer);
// COMMIT
buffer = gwbuf_alloc_and_load(PACKET_8_LEN, resultset1 + PACKET_8_IDX);
mxs_mysql_get_session_track_info(buffer, &proto);
mxb_assert(strncmp(gwbuf_get_property(buffer, (char*)"trx_state"), "________", 8) == 0);
gwbuf_free(buffer);
}
/* multi results combine in one buffer, test for check boundary handle properly */
void test2()
{
GWBUF* buffer;
fprintf(stderr, "test_session_track: multi results test\n");
proto.server_capabilities = GW_MYSQL_CAPABILITIES_SESSION_TRACK;
proto.num_eof_packets = 0;
buffer = gwbuf_alloc_and_load(sizeof(resultset2), resultset2);
mxs_mysql_get_session_track_info(buffer, &proto);
mxb_assert(strncmp(gwbuf_get_property(buffer, (char*)"trx_state"), "I_R_W___", 8) == 0);
gwbuf_free(buffer);
}
void test3()
{
GWBUF* buffer;
proto.server_capabilities = GW_MYSQL_CAPABILITIES_SESSION_TRACK;
proto.num_eof_packets = 0;
fprintf(stderr, "test_session_track: protocol state test\n");
buffer = gwbuf_alloc_and_load(sizeof(resultset2), resultset2);
mxs_mysql_get_session_track_info(buffer, &proto);
mxb_assert(strncmp(gwbuf_get_property(buffer, (char*)"trx_state"), "I_R_W___", 8) == 0);
gwbuf_free(buffer);
}
int main(int argc, char** argv)
{
test1();
test2();
test3();
return 0;
}
Reference in New Issue View Git Blame Copy Permalink