From 98bdc2079442b79780db04bb8d05a2c329577e10 Mon Sep 17 00:00:00 2001
From: Alexander Sulfrian <alexander@sulfrian.net>
Date: Sun, 10 Jan 2010 21:47:42 +0100
Subject: added function for testing weather a table contains a column or not

---
 test/base/database.cpp | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'test')

diff --git a/test/base/database.cpp b/test/base/database.cpp
index fdf4a3dd..72b44094 100644
--- a/test/base/database.cpp
+++ b/test/base/database.cpp
@@ -43,6 +43,9 @@ namespace usdx
 		CPPUNIT_TEST(testFormatDate);
 		CPPUNIT_TEST(testTableExists);
 		CPPUNIT_TEST(testTableNotExists);
+		CPPUNIT_TEST(testTableColumnExists);
+		CPPUNIT_TEST(testTableColumnNotExists);
+		CPPUNIT_TEST(testTableColumnTableNotExists);
 		CPPUNIT_TEST_SUITE_END();
 	private:
 		StatDatabase *db;
@@ -102,6 +105,21 @@ namespace usdx
 		{
 			CPPUNIT_ASSERT( false == StatDatabase::get_instance()->sqlite_table_exists("abc") );
 		}
+
+		void testTableColumnExists()
+		{
+			CPPUNIT_ASSERT( true == StatDatabase::get_instance()->sqlite_table_contains_column("us_songs", "Title") );
+		}
+
+		void testTableColumnNotExists()
+		{
+			CPPUNIT_ASSERT( false == StatDatabase::get_instance()->sqlite_table_contains_column("us_songs", "title") );
+		}
+
+		void testTableColumnTableNotExists()
+		{
+			CPPUNIT_ASSERT( false == StatDatabase::get_instance()->sqlite_table_contains_column("abc", "Title") );
+		}
 	};
 
 	CPPUNIT_TEST_SUITE_REGISTRATION(StatDatabaseTest);
-- 
cgit v1.2.3