CppCMS C++ Web Framework Code

///////////////////////////////////////////////////////////////////////////////
//                                                                             
//  Copyright (C) 2010-2011  Artyom Beilis (Tonkikh) <artyomtnk@yahoo.com>     
//                                                                             
//  Distributed under:
//
//                   the Boost Software License, Version 1.0.
//              (See accompanying file LICENSE_1_0.txt or copy at 
//                     http://www.boost.org/LICENSE_1_0.txt)
//
//  or (at your opinion) under:
//
//                               The MIT License
//                 (See accompanying file MIT.txt or a copy at
//              http://www.opensource.org/licenses/mit-license.php)
//
///////////////////////////////////////////////////////////////////////////////
#include <cppdb/driver_manager.h>
#include <cppdb/conn_manager.h>
#include "test.h"
#include "dummy_driver.h" 

#if ( defined(WIN32) || defined(_WIN32) || defined(__WIN32) ) && !defined(__CYGWIN__)
# ifndef NOMINMAX
#  define NOMINMAX
# endif
# include <windows.h>
void sleep(int x)
{
	Sleep(1000*x);
}
#else
# include <stdlib.h>
# include <unistd.h>
#endif

void test_driver_manager()
{
	cppdb::ref_ptr<cppdb::backend::connection> c1,c2,c3,c4;
	cppdb::driver_manager &dm = cppdb::driver_manager::instance();
	std::cout << "Testing drivers collection" << std::endl;
	cppdb::connections_manager &cm = cppdb::connections_manager::instance();
	dm.install_driver("dummy",new dummy::loadable_driver());
	TEST(dummy::drivers==1);
	dm.collect_unused();
	TEST(dummy::drivers==0);
	dm.install_driver("dummy",new dummy::loadable_driver());
	c1=dm.connect("dummy:");
	TEST(dummy::connections==1);
	dm.collect_unused();
	TEST(dummy::drivers==1);
	TEST(dummy::connections==1);
	c2=dm.connect("dummy:");
	dm.collect_unused();
	TEST(dummy::connections==2);
	TEST(dummy::drivers==1);
	c1=0;
	TEST(dummy::connections==1);
	c2=0;
	TEST(dummy::connections==0);
	TEST(dummy::drivers==1);
	dm.collect_unused();
	TEST(dummy::drivers==0);
	THROWS(c1=dm.connect("dummy:"),cppdb::cppdb_error);
	std::cout << "Testing connection pooling" << std::endl;
	dm.install_driver("dummy",new dummy::loadable_driver());
	c1=cm.open("dummy:@pool_size=2;@pool_max_idle=2");
	TEST(dummy::connections==1);
	c2=cm.open("dummy:@pool_size=2;@pool_max_idle=2");
	TEST(dummy::connections==2);
	c3=cm.open("dummy:@pool_size=2;@pool_max_idle=2");
	TEST(dummy::connections==3);
	c1.reset();
	TEST(dummy::connections==3);
	c2.reset();
	TEST(dummy::connections==3);
	c3.reset();
	TEST(dummy::connections==2);
	c3=cm.open("dummy:@pool_size=2;@pool_max_idle=2");
	TEST(dummy::connections==2);
	TEST(dummy::drivers==1);
	cm.gc();
	dm.collect_unused();
	TEST(dummy::connections==2);
	TEST(dummy::drivers==1);
	TEST(dummy::connections==2);
	TEST(dummy::drivers==1);
	cm.gc();
	dm.collect_unused();
	TEST(dummy::connections==2);
	TEST(dummy::drivers==1);
	sleep(3);
	TEST(dummy::connections==2);
	TEST(dummy::drivers==1);
	cm.gc();
	dm.collect_unused();
	TEST(dummy::connections==1);
	TEST(dummy::drivers==1);
	c3.reset();
	TEST(dummy::connections==1);
	TEST(dummy::drivers==1);
	sleep(3);
	cm.gc();
	dm.collect_unused();
	TEST(dummy::connections==0);
	TEST(dummy::drivers==0);
}

void test_stmt_cache()
{
	cppdb::ref_ptr<cppdb::backend::connection> c;
	cppdb::ref_ptr<cppdb::backend::statement> s1,s2,s3;

	cppdb::driver_manager &dm = cppdb::driver_manager::instance();
	dm.install_driver("dummy",new dummy::loadable_driver());
	c=dm.connect("dummy:@use_prepared=off");
	s1=c->prepare("test1");
	s2=c->prepare("test2");
	TEST(dummy::statements==2);
	s1.reset();
	TEST(dummy::statements==1);
	s2.reset();
	TEST(dummy::statements==0);
	c=dm.connect("dummy:@use_prepared=on;@stmt_cache_size=3");
	s1=c->prepare("test1");
	s1=c->prepare("test2");
	s1=c->prepare("test3");
	TEST(dummy::statements==3);
	s1=c->prepare("test4");
	TEST(dummy::statements==4);
	s1=c->prepare("test5");
	TEST(dummy::statements==4);
	s1.reset();
	s1=c->prepare("test3");
	TEST(dummy::statements==3);
	s1.reset();
	TEST(dummy::statements==3);
	c->clear_cache();
	TEST(dummy::statements==0);
	s1=c->prepare("test");
	s1.reset();
	s1=c->prepare("test");
	TEST(dummy::statements==1);
	s1=c->prepare("test1");
	TEST(dummy::statements==2);
	s2=c->prepare("test1");
	TEST(dummy::statements==3);
	s1.reset();
	s2.reset();
	TEST(dummy::statements==2);

}

int main()
{
	try {
		test_driver_manager();
	}
	CATCH_BLOCK()
	try {
		test_stmt_cache();
	}
	CATCH_BLOCK()
	SUMMARY();

}