Line data    Source code

       1             : #include "common.h"
       2             : 
       3             : /* Test cursors */
       4             : 
       5          10 : TEST_MAIN()
       6             : {
       7             : #define ROWS 3
       8             : #define C_LEN 10
       9             : 
      10             :         SQLUINTEGER n[ROWS];
      11             :         char c[ROWS][C_LEN];
      12             :         SQLLEN c_len[ROWS], n_len[ROWS];
      13             : 
      14             :         SQLUSMALLINT statuses[ROWS];
      15             :         SQLUSMALLINT i;
      16             :         SQLULEN num_row;
      17             :         int i_test;
      18             : 
      19             :         typedef struct _test
      20             :         {
      21             :                 SQLUSMALLINT type;
      22             :                 SQLINTEGER irow;
      23             :                 int start;
      24             :                 int num;
      25             :         } TEST;
      26             : 
      27             :         static const TEST tests[] = {
      28             :                 {SQL_FETCH_NEXT, 0, 1, 3},
      29             :                 {SQL_FETCH_NEXT, 0, 4, 2},
      30             :                 {SQL_FETCH_PRIOR, 0, 1, 3},
      31             :                 {SQL_FETCH_NEXT, 0, 4, 2},
      32             :                 {SQL_FETCH_NEXT, 0, -1, -1},
      33             :                 {SQL_FETCH_FIRST, 0, 1, 3},
      34             :                 {SQL_FETCH_ABSOLUTE, 3, 3, 3},
      35             :                 {SQL_FETCH_RELATIVE, 1, 4, 2},
      36             :                 {SQL_FETCH_LAST, 0, 3, 3}
      37             :         };
      38          10 :         const int num_tests = TDS_VECTOR_SIZE(tests);
      39             : 
      40          10 :         odbc_connect();
      41          10 :         odbc_check_cursor();
      42             : 
      43             :         /* create test table */
      44           8 :         odbc_command("IF OBJECT_ID('tempdb..#test') IS NOT NULL DROP TABLE #test");
      45           8 :         odbc_command("CREATE TABLE #test(i int, c varchar(6))");
      46           8 :         odbc_command("INSERT INTO #test(i, c) VALUES(1, 'a')");
      47           8 :         odbc_command("INSERT INTO #test(i, c) VALUES(2, 'bb')");
      48           8 :         odbc_command("INSERT INTO #test(i, c) VALUES(3, 'ccc')");
      49           8 :         odbc_command("INSERT INTO #test(i, c) VALUES(4, 'dddd')");
      50           8 :         odbc_command("INSERT INTO #test(i, c) VALUES(5, 'eeeee')");
      51             : 
      52             :         /* set cursor options */
      53           8 :         odbc_reset_statement();
      54           8 :         CHKSetStmtAttr(SQL_ATTR_CONCURRENCY, (SQLPOINTER) SQL_CONCUR_ROWVER, 0, "S");
      55           8 :         CHKSetStmtAttr(SQL_ATTR_CURSOR_SCROLLABLE, (SQLPOINTER) SQL_SCROLLABLE, 0, "S");
      56           8 :         CHKSetStmtAttr(SQL_ATTR_CURSOR_TYPE, (SQLPOINTER) SQL_CURSOR_DYNAMIC, 0, "S");
      57           8 :         CHKSetStmtAttr(SQL_ATTR_ROW_ARRAY_SIZE, (SQLPOINTER) ROWS, 0, "S");
      58           8 :         CHKSetStmtAttr(SQL_ATTR_ROW_STATUS_PTR, (SQLPOINTER) statuses, 0, "S");
      59           8 :         CHKSetStmtAttr(SQL_ATTR_ROWS_FETCHED_PTR, &num_row, 0, "S");
      60             : 
      61             :         /* */
      62           8 :         CHKExecDirect(T("SELECT i, c FROM #test"), SQL_NTS, "S");
      63             : 
      64             :         /* bind some rows at a time */
      65           8 :         CHKBindCol(1, SQL_C_ULONG, n, 0, n_len, "S");
      66           8 :         CHKBindCol(2, SQL_C_CHAR, c, C_LEN, c_len, "S");
      67             : 
      68          88 :         for (i_test = 0; i_test < num_tests; ++i_test) {
      69          72 :                 const TEST *t = &tests[i_test];
      70             : 
      71          72 :                 printf("Test %d\n", i_test + 1);
      72             : 
      73          72 :                 if (t->start == -1) {
      74           8 :                         CHKFetchScroll(t->type, t->irow, "No");
      75             :                 } else {
      76          64 :                         CHKFetchScroll(t->type, t->irow, "S");
      77             : 
      78          64 :                         if (t->start < 1) {
      79           0 :                                 fprintf(stderr, "Rows not expected\n");
      80           0 :                                 exit(1);
      81             :                         }
      82             : 
      83             :                         /* print, just for debug */
      84         168 :                         for (i = 0; i < num_row; ++i)
      85         168 :                                 printf("row %d i %d c %s\n", (int) (i + 1), (int) n[i], c[i]);
      86          64 :                         printf("---\n");
      87             : 
      88          64 :                         if (num_row != t->num) {
      89           0 :                                 fprintf(stderr, "Expected %d rows, got %d\n", t->num, (int) num_row);
      90           0 :                                 exit(1);
      91             :                         }
      92             : 
      93         168 :                         for (i = 0; i < num_row; ++i) {
      94             :                                 char name[10];
      95             : 
      96         168 :                                 memset(name, 0, sizeof(name));
      97         168 :                                 memset(name, 'a' - 1 + i + t->start, i + t->start);
      98         168 :                                 if (n[i] != i + t->start || c_len[i] != strlen(name) || strcmp(c[i], name) != 0) {
      99           0 :                                         fprintf(stderr, "Wrong row returned\n");
     100           0 :                                         fprintf(stderr, "\tn %d %d\n", (int) n[i], i + t->start);
     101           0 :                                         fprintf(stderr, "\tc len %d %d\n", (int) c_len[i], (int) strlen(name));
     102           0 :                                         fprintf(stderr, "\tc %s %s\n", c[i], name);
     103           0 :                                         exit(1);
     104             :                                 }
     105             :                         }
     106             :                 }
     107             :         }
     108             : 
     109           8 :         odbc_reset_statement();
     110             : 
     111           8 :         odbc_disconnect();
     112           8 :         return 0;
     113             : }