« back to all changes in this revision
Viewing changes to test/client_server.c
- browse files at revision 2
- compare with another revision
- compare with revision 2
- download diff
- download tarball
- view history from revision 2
- Committer: Andrew Hutchings
- Date: 2012-09-03 18:38:34 UTC
- Revision ID: git-v1:e936672f2f2e4af2b54de08ebf06d53ca1dc6872
The "it compiles!" version
- files added:
- .gitignore
- libdrizzle
- test
added
removed
test/client_server.c
1
/* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2
*
3
* Drizzle Client & Protocol Library
4
*
5
* Copyright (C) 2008 Eric Day (eday@oddments.org)
6
* All rights reserved.
7
*
8
* Redistribution and use in source and binary forms, with or without
9
* modification, are permitted provided that the following conditions are
10
* met:
11
*
12
* * Redistributions of source code must retain the above copyright
13
* notice, this list of conditions and the following disclaimer.
14
*
15
* * Redistributions in binary form must reproduce the above
16
* copyright notice, this list of conditions and the following disclaimer
17
* in the documentation and/or other materials provided with the
18
* distribution.
19
*
20
* * The names of its contributors may not be used to endorse or
21
* promote products derived from this software without specific prior
22
* written permission.
23
*
24
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
25
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
26
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
27
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
28
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
29
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
30
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
31
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
32
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
33
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
34
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
35
*/
36
37
/**
38
* @file
39
* @brief Tests for drizzle_st Structures
40
*/
41
42
#include <libdrizzle-1.0/t/common.h>
43
44
#include <assert.h>
45
#include <unistd.h>
46
47
#define DRIZZLE_TEST_HOST "localhost"
48
#define DRIZZLE_TEST_PORT 12399
49
50
typedef enum
51
{
52
SERVER_STATE_START,
53
SERVER_STATE_HANDSHAKE_WRITE,
54
SERVER_STATE_HANDSHAKE_READ,
55
SERVER_STATE_HANDSHAKE_RESULT,
56
SERVER_STATE_COMMAND_BUFFER,
57
SERVER_STATE_RESULT_HEADER,
58
SERVER_STATE_RESULT_COLUMN_1,
59
SERVER_STATE_RESULT_COLUMN_2,
60
SERVER_STATE_RESULT_COLUMN_EOF,
61
SERVER_STATE_RESULT_ROW_1,
62
SERVER_STATE_RESULT_ROW_1_FIELD_1,
63
SERVER_STATE_RESULT_ROW_1_FIELD_2,
64
SERVER_STATE_RESULT_ROW_EOF,
65
SERVER_STATE_DONE
66
} server_state_t;
67
68
typedef struct
69
{
70
server_state_t state;
71
drizzle_result_st result;
72
drizzle_column_st column;
73
drizzle_command_t command;
74
char *data;
75
size_t total;
76
} server_state_st;
77
78
typedef enum
79
{
80
CLIENT_STATE_START,
81
CLIENT_STATE_RESULT,
82
CLIENT_STATE_DONE
83
} client_state_t;
84
85
typedef struct
86
{
87
client_state_t state;
88
drizzle_result_st result;
89
} client_state_st;
90
91
static void _server(drizzle_con_st *con, server_state_st *state);
92
static void _client(drizzle_con_st *con, client_state_st *state);
93
94
int main(void)
95
{
96
close(STDOUT_FILENO);
97
98
drizzle_st drizzle;
99
drizzle_con_st listen_con;
100
drizzle_con_st client;
101
drizzle_con_st server;
102
drizzle_return_t ret;
103
bool server_accepted = false;
104
server_state_st server_state;
105
client_state_st client_state;
106
107
drizzle_test("drizzle_create");
108
if (drizzle_create(&drizzle) == NULL)
109
{
110
drizzle_test_error("returned NULL");
111
}
112
113
drizzle_test("drizzle_con_add_tcp_listen");
114
if (drizzle_con_add_tcp_listen(&drizzle, &listen_con, DRIZZLE_TEST_HOST,
115
DRIZZLE_TEST_PORT, 1,
116
DRIZZLE_CON_NONE) == NULL)
117
{
118
drizzle_test_error("returned NULL");
119
}
120
121
drizzle_test("drizzle_con_listen");
122
ret= drizzle_con_listen(&listen_con);
123
if (ret != DRIZZLE_RETURN_OK)
124
{
125
drizzle_test_error("returned %s (%d)", drizzle_error(&drizzle), ret);
126
}
127
128
drizzle_test("drizzle_con_add_tcp");
129
if (drizzle_con_add_tcp(&drizzle, &client, DRIZZLE_TEST_HOST,
130
DRIZZLE_TEST_PORT, NULL, NULL, NULL,
131
DRIZZLE_CON_NONE) == NULL)
132
{
133
drizzle_test_error("returned NULL");
134
}
135
136
drizzle_test("drizzle_add_options");
137
drizzle_add_options(&drizzle, DRIZZLE_NON_BLOCKING);
138
139
server_state.state= SERVER_STATE_START;
140
client_state.state= CLIENT_STATE_START;
141
142
while (true)
143
{
144
if (server_accepted == false)
145
{
146
drizzle_test("drizzle_con_accept");
147
(void)drizzle_con_accept(&drizzle, &server, &ret);
148
if (ret == DRIZZLE_RETURN_OK)
149
{
150
server_accepted = true;
151
}
152
else if (ret == DRIZZLE_RETURN_COULD_NOT_CONNECT || ret == DRIZZLE_RETURN_NO_ACTIVE_CONNECTIONS)
153
{
154
break;
155
}
156
else if (ret != DRIZZLE_RETURN_IO_WAIT)
157
{
158
drizzle_test_error("returned %s (%s)", drizzle_error(&drizzle), drizzle_strerror(ret));
159
}
160
}
161
162
if (server_accepted)
163
{
164
_server(&server, &server_state);
165
}
166
167
_client(&client, &client_state);
168
169
if (server_state.state == SERVER_STATE_DONE &&
170
client_state.state == CLIENT_STATE_DONE)
171
{
172
break;
173
}
174
175
drizzle_test("drizzle_con_wait");
176
ret= drizzle_con_wait(&drizzle);
177
if (ret == DRIZZLE_RETURN_COULD_NOT_CONNECT || ret == DRIZZLE_RETURN_NO_ACTIVE_CONNECTIONS)
178
{
179
break;
180
}
181
else if (ret != DRIZZLE_RETURN_OK)
182
{
183
drizzle_test_error("returned %s (%d)", drizzle_error(&drizzle), ret);
184
}
185
}
186
187
if (server_accepted)
188
{
189
drizzle_test("drizzle_con_free");
190
drizzle_con_free(&server);
191
}
192
193
drizzle_test("drizzle_con_free");
194
drizzle_con_free(&client);
195
196
drizzle_test("drizzle_con_free");
197
drizzle_con_free(&listen_con);
198
199
drizzle_test("drizzle_free");
200
drizzle_free(&drizzle);
201
202
return 0;
203
}
204
205
static void _server(drizzle_con_st *con, server_state_st *state)
206
{
207
drizzle_return_t ret;
208
const drizzle_field_t fields[2]=
209
{
210
(drizzle_field_t)"test_field_1",
211
(drizzle_field_t)"test_field_2"
212
};
213
const size_t field_sizes[2]= { 12, 12 };
214
215
switch(state->state)
216
{
217
case SERVER_STATE_START:
218
drizzle_con_set_protocol_version(con, 10);
219
drizzle_con_set_server_version(con, "test_version");
220
drizzle_con_set_thread_id(con, 1);
221
drizzle_con_set_scramble(con, (const uint8_t *)"ABCDEFGHIJKLMNOPQRST");
222
drizzle_con_set_capabilities(con, DRIZZLE_CAPABILITIES_NONE);
223
drizzle_con_set_charset(con, 8);
224
drizzle_con_set_status(con, DRIZZLE_CON_STATUS_NONE);
225
drizzle_con_set_max_packet_size(con, DRIZZLE_MAX_PACKET_SIZE);
226
227
case SERVER_STATE_HANDSHAKE_WRITE:
228
drizzle_test("drizzle_handshake_server_write");
229
ret= drizzle_handshake_server_write(con);
230
if (ret == DRIZZLE_RETURN_IO_WAIT)
231
{
232
state->state = SERVER_STATE_HANDSHAKE_WRITE;
233
return;
234
}
235
else if (ret != DRIZZLE_RETURN_OK)
236
{
237
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
238
}
239
240
case SERVER_STATE_HANDSHAKE_READ:
241
drizzle_test("drizzle_handshake_client_read");
242
ret= drizzle_handshake_client_read(con);
243
if (ret == DRIZZLE_RETURN_IO_WAIT)
244
{
245
state->state = SERVER_STATE_HANDSHAKE_READ;
246
return;
247
}
248
else if (ret == DRIZZLE_RETURN_LOST_CONNECTION)
249
{
250
return;
251
}
252
else if (ret == DRIZZLE_RETURN_COULD_NOT_CONNECT)
253
{
254
return;
255
}
256
else if (ret != DRIZZLE_RETURN_OK)
257
{
258
drizzle_test_error("returned %s (%s)", drizzle_con_error(con), drizzle_strerror(ret));
259
}
260
261
drizzle_test("drizzle_result_create");
262
if (drizzle_result_create(con, &state->result) == NULL)
263
drizzle_test_error("returned %s", drizzle_con_error(con));
264
265
case SERVER_STATE_HANDSHAKE_RESULT:
266
drizzle_test("drizzle_handshake_result_write");
267
ret= drizzle_result_write(con, &state->result, true);
268
if (ret == DRIZZLE_RETURN_IO_WAIT)
269
{
270
state->state = SERVER_STATE_HANDSHAKE_RESULT;
271
return;
272
}
273
else if (ret != DRIZZLE_RETURN_OK)
274
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
275
276
drizzle_result_free(&state->result);
277
278
case SERVER_STATE_COMMAND_BUFFER:
279
drizzle_test("drizzle_con_command_buffer");
280
state->data= drizzle_con_command_buffer(con, &state->command, &state->total,
281
&ret);
282
if (ret == DRIZZLE_RETURN_IO_WAIT)
283
{
284
state->state = SERVER_STATE_COMMAND_BUFFER;
285
return;
286
}
287
else if (ret != DRIZZLE_RETURN_OK)
288
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
289
290
if (state->command != DRIZZLE_COMMAND_QUERY)
291
drizzle_test_error("command was not a query");
292
293
if (state->total != 6 && !memcmp(state->data, "SELECT", 6))
294
drizzle_test_error("query doesn't match");
295
296
if (state->data != NULL)
297
{
298
free(state->data);
299
state->data= NULL;
300
}
301
302
drizzle_test("drizzle_result_create");
303
if (drizzle_result_create(con, &state->result) == NULL)
304
drizzle_test_error("returned %s", drizzle_con_error(con));
305
306
drizzle_result_set_column_count(&state->result, 2);
307
308
case SERVER_STATE_RESULT_HEADER:
309
drizzle_test("drizzle_handshake_result_write");
310
ret= drizzle_result_write(con, &state->result, false);
311
if (ret == DRIZZLE_RETURN_IO_WAIT)
312
{
313
state->state = SERVER_STATE_RESULT_HEADER;
314
return;
315
}
316
else if (ret != DRIZZLE_RETURN_OK)
317
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
318
319
drizzle_test("drizzle_column_create");
320
if (drizzle_column_create(&state->result, &state->column) == NULL)
321
drizzle_test_error("returned %s", drizzle_con_error(con));
322
323
drizzle_column_set_catalog(&state->column, "test_catalog");
324
drizzle_column_set_db(&state->column, "test_db");
325
drizzle_column_set_table(&state->column, "test_table");
326
drizzle_column_set_orig_table(&state->column, "test_orig_table");
327
drizzle_column_set_name(&state->column, "test_column_1");
328
drizzle_column_set_orig_name(&state->column, "test_orig_column_1");
329
drizzle_column_set_charset(&state->column, 8);
330
drizzle_column_set_size(&state->column, 32);
331
drizzle_column_set_type(&state->column, DRIZZLE_COLUMN_TYPE_VARCHAR);
332
drizzle_column_set_flags(&state->column, DRIZZLE_COLUMN_FLAGS_NONE);
333
334
case SERVER_STATE_RESULT_COLUMN_1:
335
drizzle_test("drizzle_column_write");
336
ret= drizzle_column_write(&state->result, &state->column);
337
if (ret == DRIZZLE_RETURN_IO_WAIT)
338
{
339
state->state = SERVER_STATE_RESULT_COLUMN_1;
340
return;
341
}
342
else if (ret != DRIZZLE_RETURN_OK)
343
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
344
345
drizzle_column_set_name(&state->column, "test_column_2");
346
drizzle_column_set_orig_name(&state->column, "test_orig_column_2");
347
348
case SERVER_STATE_RESULT_COLUMN_2:
349
drizzle_test("drizzle_column_write");
350
ret= drizzle_column_write(&state->result, &state->column);
351
if (ret == DRIZZLE_RETURN_IO_WAIT)
352
{
353
state->state = SERVER_STATE_RESULT_COLUMN_2;
354
return;
355
}
356
else if (ret != DRIZZLE_RETURN_OK)
357
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
358
359
drizzle_test("drizzle_column_free");
360
drizzle_column_free(&state->column);
361
362
drizzle_result_set_eof(&state->result, true);
363
364
case SERVER_STATE_RESULT_COLUMN_EOF:
365
drizzle_test("drizzle_handshake_result_write");
366
ret= drizzle_result_write(con, &state->result, false);
367
if (ret == DRIZZLE_RETURN_IO_WAIT)
368
{
369
state->state = SERVER_STATE_RESULT_COLUMN_EOF;
370
return;
371
}
372
else if (ret != DRIZZLE_RETURN_OK)
373
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
374
375
drizzle_result_calc_row_size(&state->result, fields, field_sizes);
376
377
case SERVER_STATE_RESULT_ROW_1:
378
ret= drizzle_row_write(&state->result);
379
380
case SERVER_STATE_RESULT_ROW_1_FIELD_1:
381
ret= drizzle_field_write(&state->result, fields[0], field_sizes[0],
382
field_sizes[0]);
383
384
case SERVER_STATE_RESULT_ROW_1_FIELD_2:
385
ret= drizzle_field_write(&state->result, fields[1], field_sizes[1],
386
field_sizes[1]);
387
388
case SERVER_STATE_RESULT_ROW_EOF:
389
drizzle_test("drizzle_handshake_result_write");
390
ret= drizzle_result_write(con, &state->result, true);
391
if (ret == DRIZZLE_RETURN_IO_WAIT)
392
{
393
state->state = SERVER_STATE_RESULT_ROW_EOF;
394
return;
395
}
396
else if (ret != DRIZZLE_RETURN_OK)
397
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
398
399
drizzle_result_free(&state->result);
400
401
state->state = SERVER_STATE_DONE;
402
403
case SERVER_STATE_DONE:
404
return;
405
406
default:
407
drizzle_test_error("invalid server state");
408
}
409
}
410
411
static void _client(drizzle_con_st *con, client_state_st *state)
412
{
413
drizzle_return_t ret;
414
drizzle_column_st *column;
415
drizzle_row_t row;
416
size_t *field_sizes;
417
418
switch(state->state)
419
{
420
case CLIENT_STATE_START:
421
drizzle_test("drizzle_query_str");
422
(void)drizzle_query_str(con, &state->result, "SELECT", &ret);
423
if (ret == DRIZZLE_RETURN_IO_WAIT)
424
{
425
state->state = CLIENT_STATE_START;
426
return;
427
}
428
else if (ret == DRIZZLE_RETURN_COULD_NOT_CONNECT)
429
{
430
state->state= CLIENT_STATE_DONE;
431
return;
432
}
433
else if (ret != DRIZZLE_RETURN_OK)
434
{
435
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
436
}
437
438
case CLIENT_STATE_RESULT:
439
drizzle_test("drizzle_result_buffer");
440
ret = drizzle_result_buffer(&state->result);
441
if (ret == DRIZZLE_RETURN_IO_WAIT)
442
{
443
state->state = CLIENT_STATE_RESULT;
444
return;
445
}
446
else if (ret != DRIZZLE_RETURN_OK)
447
{
448
drizzle_test_error("returned %s (%d)", drizzle_con_error(con), ret);
449
}
450
451
drizzle_test("drizzle_con_protocol_version");
452
if (drizzle_con_protocol_version(con) != 10)
453
{
454
drizzle_test_error("no match");
455
}
456
457
drizzle_test("drizzle_con_server_version");
458
if (strcmp(drizzle_con_server_version(con), "test_version"))
459
drizzle_test_error("no match");
460
461
drizzle_test("drizzle_con_thread_id");
462
if (drizzle_con_thread_id(con) != 1)
463
drizzle_test_error("no match");
464
465
drizzle_test("drizzle_con_scramble");
466
if (memcmp(drizzle_con_scramble(con), "ABCDEFGHIJKLMNOPQRST",
467
DRIZZLE_MAX_SCRAMBLE_SIZE))
468
{
469
drizzle_test_error("no match");
470
}
471
472
/* We default to MySQL protocol right now, which sets this flag. */
473
drizzle_test("drizzle_con_capabilities");
474
if (drizzle_con_capabilities(con) != DRIZZLE_CAPABILITIES_PROTOCOL_41)
475
drizzle_test_error("no match");
476
477
drizzle_test("drizzle_con_charset");
478
if (drizzle_con_charset(con) != 8)
479
drizzle_test_error("no match");
480
481
drizzle_test("drizzle_con_status");
482
if (drizzle_con_status(con) != DRIZZLE_CON_STATUS_NONE)
483
drizzle_test_error("no match");
484
485
drizzle_test("drizzle_con_packet_size");
486
if (drizzle_con_max_packet_size(con) != DRIZZLE_MAX_PACKET_SIZE)
487
drizzle_test_error("no match");
488
489
drizzle_test("drizzle_column_next");
490
column= drizzle_column_next(&state->result);
491
if (column == NULL)
492
drizzle_test_error("column not found");
493
494
drizzle_test("drizzle_column_drizzle_result");
495
if (drizzle_column_drizzle_result(column) != &state->result)
496
drizzle_test_error("no match");
497
498
drizzle_test("drizzle_column_catalog");
499
if (strcmp(drizzle_column_catalog(column), "test_catalog"))
500
drizzle_test_error("no match");
501
502
drizzle_test("drizzle_column_db");
503
if (strcmp(drizzle_column_db(column), "test_db"))
504
drizzle_test_error("no match");
505
506
drizzle_test("drizzle_column_table");
507
if (strcmp(drizzle_column_table(column), "test_table"))
508
drizzle_test_error("no match");
509
510
drizzle_test("drizzle_column_orig_table");
511
if (strcmp(drizzle_column_orig_table(column), "test_orig_table"))
512
drizzle_test_error("no match");
513
514
drizzle_test("drizzle_column_name");
515
if (strcmp(drizzle_column_name(column), "test_column_1"))
516
drizzle_test_error("no match");
517
518
drizzle_test("drizzle_column_orig_name");
519
if (strcmp(drizzle_column_orig_name(column), "test_orig_column_1"))
520
drizzle_test_error("no match");
521
522
drizzle_test("drizzle_column_charset");
523
if (drizzle_column_charset(column) != 8)
524
drizzle_test_error("no match");
525
526
drizzle_test("drizzle_column_size");
527
if (drizzle_column_size(column) != 32)
528
drizzle_test_error("no match");
529
530
drizzle_test("drizzle_column_max_size");
531
if (drizzle_column_max_size(column) != 12)
532
drizzle_test_error("no match");
533
534
drizzle_test("drizzle_column_type_drizzle");
535
if (drizzle_column_type_drizzle(column) != DRIZZLE_COLUMN_TYPE_DRIZZLE_VARCHAR)
536
drizzle_test_error("no match");
537
538
drizzle_test("drizzle_column_flags");
539
if (drizzle_column_flags(column) != DRIZZLE_COLUMN_FLAGS_NONE)
540
drizzle_test_error("no match");
541
542
drizzle_test("drizzle_column_next");
543
column= drizzle_column_next(&state->result);
544
if (column == NULL)
545
drizzle_test_error("column not found");
546
547
drizzle_test("drizzle_column_name");
548
if (strcmp(drizzle_column_name(column), "test_column_2"))
549
drizzle_test_error("no match");
550
551
drizzle_test("drizzle_column_next");
552
column= drizzle_column_next(&state->result);
553
if (column != NULL)
554
drizzle_test_error("column found");
555
556
drizzle_test("drizzle_column_prev");
557
column= drizzle_column_prev(&state->result);
558
if (column == NULL)
559
drizzle_test_error("column not found");
560
561
drizzle_test("drizzle_column_name");
562
if (strcmp(drizzle_column_name(column), "test_column_2"))
563
drizzle_test_error("no match");
564
565
drizzle_test("drizzle_row_next");
566
row= drizzle_row_next(&state->result);
567
if (row == NULL)
568
drizzle_test_error("row not found");
569
570
if (strcmp(row[0], "test_field_1") || strcmp(row[1], "test_field_2"))
571
drizzle_test_error("no match");
572
573
drizzle_test("drizzle_row_field_sizes");
574
field_sizes= drizzle_row_field_sizes(&state->result);
575
if (field_sizes[0] != 12 || field_sizes[1] != 12)
576
drizzle_test_error("no match");
577
578
drizzle_test("drizzle_row_prev");
579
row = drizzle_row_prev(&state->result);
580
if (row == NULL)
581
drizzle_test_error("row not found");
582
583
if (strcmp(row[0], "test_field_1") || strcmp(row[1], "test_field_2"))
584
drizzle_test_error("no match");
585
586
drizzle_test("drizzle_row_next");
587
row = drizzle_row_next(&state->result);
588
589
drizzle_test("drizzle_row_index");
590
row = drizzle_row_index(&state->result, 0);
591
if (row == NULL)
592
drizzle_test_error("row not found");
593
594
if (strcmp(row[0], "test_field_1") || strcmp(row[1], "test_field_2"))
595
drizzle_test_error("no match");
596
597
drizzle_test("drizzle_row_index");
598
row = drizzle_row_index(&state->result, 1);
599
if (row != NULL)
600
drizzle_test_error("row found");
601
602
drizzle_test("drizzle_result_free");
603
drizzle_result_free(&state->result);
604
605
state->state = CLIENT_STATE_DONE;
606
607
case CLIENT_STATE_DONE:
608
return;
609
610
default:
611
drizzle_test_error("invalid client state");
612
}
613
}
Loggerhead is a web-based interface for Breezy
Version: 2.0.1