/*
 * 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()
{
	connect1();
}

void
tearDown()
{
	GTMPQfinish(conn);
}

void
test_node_01()
{
	int rc;

	SETUP();

	rc = node_register(conn, GTM_NODE_DATANODE,  6666, "one", "/tmp/pgxc/data/gtm");
	_ASSERT( rc >= 0 );

	TEARDOWN();
}

void
test_node_02()
{
	int rc;

	SETUP();

	rc = node_unregister(conn, GTM_NODE_DATANODE, "One");
	_ASSERT( rc >= 0 );

	TEARDOWN();
}

void
test_node_03()
{
	GTM_PGXCNodeInfo *data;
	int rc, i;

	SETUP();

	data = (GTM_PGXCNodeInfo *)malloc( sizeof(GTM_PGXCNodeInfo)*128 );
	memset(data, 0, sizeof(GTM_PGXCNodeInfo)*128);

	rc = get_node_list(conn, data, 128);
	_ASSERT( rc>=0 );

	for (i=0 ; i<rc ; i++)
	{
		print_nodeinfo(data[i]);
	}

	free(data);

	TEARDOWN();
}

void
test_node_04()
{
	GTM_PGXCNodeInfo *data;
	int rc, i;

	SETUP();

	data = (GTM_PGXCNodeInfo *)malloc( sizeof(GTM_PGXCNodeInfo)*128 );
	memset(data, 0, sizeof(GTM_PGXCNodeInfo)*128);

	rc = node_register(conn, GTM_NODE_DATANODE, 6666, "one", "/tmp/pgxc/data/gtm");
	_ASSERT( rc>=0 );

	rc = get_node_list(conn, data, 128);
	_ASSERT( rc>=0 );

	for (i=0 ; i<rc ; i++)
	{
		print_nodeinfo(data[i]);
	}

	free(data);

	TEARDOWN();
}


void
test_node_05()
{
	int rc, i;

	SETUP();

	rc = node_unregister(conn, GTM_NODE_DATANODE, "One zero one");

	rc = node_register(conn, GTM_NODE_DATANODE, 6666, "One zero one", "/tmp/pgxc/data/gtm");
	_ASSERT( rc>=0 );

	sleep(5);

	rc = backend_disconnect(conn, true, GTM_NODE_DATANODE, "One Zero one");
	_ASSERT( rc>=0 );

	rc = node_unregister(conn, GTM_NODE_DATANODE, "One zero one");
	_ASSERT( rc>=0 );

	TEARDOWN();
}


int
main(int argc, char *argv[])
{
	test_node_01();
	test_node_02();

	test_node_03();
	test_node_04();

	test_node_05();

	return 0;
}