/*
 * Copyright (c) 2010-2012 Postgres-XC Development Group
 */

#include <sys/types.h>
#include <unistd.h>

#include "gtm/libpq-fe.h"
#include "gtm/gtm_c.h"
#include "gtm/gtm_client.h"

#include "test_common.h"

pthread_key_t threadinfo_key;

void setUp()
{
    conn = PQconnectGTM(connect_string);
    if (conn == NULL) {
        client_log(("Error in connection\n"));
        exit(1);
    }
    client_log(("PGconnectGTM() ok.\n"));
}

void tearDown()
{
    GTMPQfinish(conn);
}

void test_txn_01()
{
    GlobalTransactionId gxid;
    int rc;

    SETUP();

    gxid = begin_transaction(conn, GTM_ISOLATION_SERIALIZABLE, timestamp);
    _ASSERT(gxid != InvalidGlobalTransactionId);

    rc = prepare_transaction(conn, gxid);
    _ASSERT(rc >= 0);

    rc = abort_transaction(conn, gxid);
    _ASSERT(rc >= 0);

    TEARDOWN();
}

void test_txn_02()
{
    GlobalTransactionId gxid;
    int rc;

    SETUP();

    gxid = begin_transaction(conn, GTM_ISOLATION_SERIALIZABLE, timestamp);
    _ASSERT(gxid != InvalidGlobalTransactionId);

    rc = prepare_transaction(conn, gxid);
    _ASSERT(rc >= 0);

    rc = commit_transaction(conn, gxid);
    _ASSERT(rc >= 0);

    TEARDOWN();
}

void test_txn_03()
{
    GlobalTransactionId gxid;
    int rc;

    SETUP();

    gxid = begin_transaction_autovacuum(conn, GTM_ISOLATION_SERIALIZABLE);
    _ASSERT(gxid != InvalidGlobalTransactionId);

    rc = commit_transaction(conn, gxid);
    _ASSERT(rc >= 0);

    TEARDOWN();
}

void test_txn_11()
{
    GTM_IsolationLevel isolevel = GTM_ISOLATION_SERIALIZABLE;
    char* gid = "test";
    GlobalTransactionId gxid = InvalidGlobalTransactionId;
    GlobalTransactionId prepared_gxid = InvalidGlobalTransactionId;
    int datanodecnt = 0;
    char** datanodes = NULL;
    int coordcnt = 0;
    char** coordinators = NULL;
    int rc;

    SETUP();

    gxid = begin_transaction(conn, isolevel, timestamp);
    _ASSERT(gxid != InvalidGlobalTransactionId);
    printf("test_txn_11: gxid=%d\n", gxid);

    rc = start_prepared_transaction(conn, gxid, gid, datanodecnt, datanodes, coordcnt, coordinators);
    _ASSERT(rc >= 0);

    rc = get_gid_data(conn, isolevel, gid, &gxid, &prepared_gxid, &datanodecnt, datanodes, &coordcnt, coordinators);
    printf("test_txn_11: prepared_gxid=%d\n", prepared_gxid);
    _ASSERT(rc >= 0);

    rc = commit_prepared_transaction(conn, gxid, prepared_gxid);
    _ASSERT(rc >= 0);

    TEARDOWN();
}

void test_txn_51()
{
    GlobalTransactionId gxid;
    int rc;

    SETUP();

    gxid = begin_transaction(conn, GTM_ISOLATION_SERIALIZABLE, timestamp);
    _ASSERT(gxid == InvalidGlobalTransactionId);

    TEARDOWN();
}

void test_txn_52()
{
    GlobalTransactionId gxid;
    int rc;

    SETUP();

    rc = prepare_transaction(conn, InvalidGlobalTransactionId);
    _ASSERT(rc == -1);

    TEARDOWN();
}

void test_txn_53()
{
    GlobalTransactionId gxid;
    int rc;

    SETUP();

    rc = commit_transaction(conn, InvalidGlobalTransactionId);
    _ASSERT(rc == -1);

    TEARDOWN();
}

void test_txn_54()
{
    GlobalTransactionId gxid;
    int rc;

    SETUP();

    rc = abort_transaction(conn, InvalidGlobalTransactionId);
    _ASSERT(rc == -1);

    TEARDOWN();
}

int main(int argc, char* argv[])
{
    sprintf(connect_string, "host=localhost port=6666 node_name=one remote_type=%d", GTM_NODE_GTM);

    test_txn_01();
    test_txn_02();
    test_txn_03();

    test_txn_11();

    /*
     * connect to standby. must be prevented.
     */
    sprintf(connect_string, "host=localhost port=6667 node_name=one remote_type=%d", GTM_NODE_GTM);

    test_txn_51();
    test_txn_52();
    test_txn_53();
    test_txn_54();

    /*
     * promote GTM standby to become GTM active.
     */
    system("./promote.sh");
    sleep(1);

    test_txn_01();
    test_txn_02();
    test_txn_03();

    test_txn_11();

    return 0;
}