Line data    Source code

       1             : #include "common.h"
       2             : #include <freetds/bool.h>
       3             : 
       4             : static int
       5          20 : Test(bool discard_test)
       6             : {
       7          20 :         ODBC_BUF *odbc_buf = NULL;
       8             :         SQLINTEGER out_buf;
       9             :         SQLLEN out_len;
      10             :         SQLLEN rows;
      11          20 :         int retcode = 0;
      12             :         SQLTCHAR buf[512];
      13             :         SQLTCHAR sqlstate[6];
      14             : 
      15          20 :         const char *createErrorProcedure = "CREATE PROCEDURE testerror AS\n"
      16             :                 "SELECT value FROM TestTransaction\n" "SELECT value / (value-value) FROM TestTransaction\n";
      17             : 
      18             :         /* select after insert is required to test data discarding */
      19             :         char createProcedure[512];
      20             : 
      21          20 :         sprintf(createProcedure,
      22             :                 "CREATE PROCEDURE testinsert @value INT AS\n"
      23             :                 "INSERT INTO TestTransaction VALUES ( @value )\n%s", discard_test ? "SELECT * FROM TestTransaction\n" : "");
      24             : 
      25             :         /* create stored proc */
      26          20 :         odbc_command_with_result(odbc_stmt, "DROP PROCEDURE testinsert");
      27             : 
      28          20 :         odbc_command(createProcedure);
      29             : 
      30             :         /* create stored proc that generates an error */
      31          20 :         odbc_command_with_result(odbc_stmt, "DROP PROCEDURE testerror");
      32             : 
      33          20 :         odbc_command(createErrorProcedure);
      34             : 
      35             :         /* Start transaction */
      36          20 :         CHKSetConnectAttr(SQL_ATTR_AUTOCOMMIT, (void *) SQL_AUTOCOMMIT_OFF, 0, "S");
      37             : 
      38             :         /* Insert a value */
      39          20 :         odbc_command("EXEC testinsert 1");
      40             : 
      41             :         /* we should be able to read row count */
      42          20 :         CHKRowCount(&rows, "S");
      43             : 
      44             :         /* Commit transaction */
      45          20 :         CHKEndTran(SQL_HANDLE_DBC, odbc_conn, SQL_COMMIT, "S");
      46             : 
      47          20 :         SQLCloseCursor(odbc_stmt);
      48             : 
      49             :         /* Start transaction */
      50          20 :         CHKSetConnectAttr(SQL_ATTR_AUTOCOMMIT, (void *) SQL_AUTOCOMMIT_OFF, 0, "S");
      51             : 
      52             :         /* Insert another value */
      53          20 :         odbc_command("EXEC testinsert 2");
      54             : 
      55             :         /* Roll back transaction */
      56          20 :         CHKEndTran(SQL_HANDLE_DBC, odbc_conn, SQL_ROLLBACK, "S");
      57             : 
      58             :         /* TODO test row inserted */
      59             : 
      60          20 :         CHKSetConnectAttr(SQL_ATTR_AUTOCOMMIT, (void *) SQL_AUTOCOMMIT_ON, 0, "S");
      61             : 
      62             :         /* generate an error */
      63          20 :         odbc_command("EXEC testerror");
      64          20 :         CHKBindCol(1, SQL_C_SLONG, &out_buf, sizeof(out_buf), &out_len, "S");
      65             : 
      66          70 :         while (CHKFetch("SNo") == SQL_SUCCESS) {
      67          30 :                 printf("\t%ld\n", (long int) out_buf);
      68          30 :                 if (out_buf != 1) {
      69           0 :                         fprintf(stderr, "error: expected to select 1 got %ld\n", (long int) out_buf);
      70           0 :                         retcode = 1;
      71           0 :                         goto cleanup;
      72             :                 }
      73             :         }
      74             : 
      75          20 :         CHKMoreResults("E");
      76             : 
      77          20 :         CHKGetDiagRec(SQL_HANDLE_STMT, odbc_stmt, 1, sqlstate, NULL, buf, TDS_VECTOR_SIZE(buf), NULL, "SI");
      78          20 :         printf("err=%s\n", C(buf));
      79             : 
      80          20 :         CHKMoreResults("No");
      81             : 
      82          20 :       cleanup:
      83             :         /* drop table */
      84          20 :         odbc_command_with_result(odbc_stmt, "DROP PROCEDURE testinsert");
      85          20 :         odbc_command_with_result(odbc_stmt, "DROP PROCEDURE testerror");
      86             : 
      87          20 :         ODBC_FREE();
      88          20 :         return retcode;
      89             : }
      90             : 
      91             : int
      92          10 : main(void)
      93             : {
      94          10 :         int retcode = 0;
      95             : 
      96          10 :         odbc_connect();
      97             : 
      98             :         /* create table */
      99          10 :         odbc_command_with_result(odbc_stmt, "DROP TABLE TestTransaction");
     100          10 :         odbc_command("CREATE TABLE TestTransaction ( value INT )");
     101             : 
     102             :         if (!retcode)
     103          10 :                 retcode = Test(true);
     104          10 :         if (!retcode)
     105          10 :                 retcode = Test(false);
     106             : 
     107             :         /* drop table */
     108          10 :         odbc_command_with_result(odbc_stmt, "DROP TABLE TestTransaction");
     109             : 
     110          10 :         odbc_disconnect();
     111             : 
     112          10 :         printf("Done.\n");
     113             :         return retcode;
     114             : }