Test DatabaseHelper

Author: hpique

AndroidBillingLibraryTest/src/net/robotmedia/billing/model/BillingDBTest.java

@@ -16,6 +16,7 @@
 package net.robotmedia.billing.model;
 
 import android.database.Cursor;
+import android.database.sqlite.SQLiteDatabase;
 import android.test.AndroidTestCase;
 import android.test.suitebuilder.annotation.SmallTest;
 
@@ -130,4 +131,71 @@ public void testQueryTransactionsStringPurchaseState() throws Exception {
 		cursor2.close();
 	}
 
+	@SmallTest
+	public void testDatabaseHelperConstructor() throws Exception {
+		BillingDB.DatabaseHelper helper = new BillingDB.DatabaseHelper(this.getContext());
+		assertEquals(helper.getReadableDatabase().getVersion(), BillingDB.DATABASE_VERSION);
+	}
+	
+	private void testColumn(SQLiteDatabase db, String column, String expectedType, boolean result) {
+		Cursor cursor = db.rawQuery("PRAGMA table_info(" +  BillingDB.TABLE_TRANSACTIONS + ")", null);
+		boolean found = false;
+		while (cursor.moveToNext()) {
+			final String name = cursor.getString(1);
+			if (name.equals(column)) {
+				final String type = cursor.getString(2);
+				assertEquals(type, expectedType);
+				found = true;
+			}
+		}
+		cursor.close();
+		assertEquals(found, result);
+	}
+	
+	@SmallTest
+	public void testDatabaseHelperOnCreate() throws Exception {
+		BillingDB.DatabaseHelper helper = new BillingDB.DatabaseHelper(this.getContext());
+		SQLiteDatabase db = SQLiteDatabase.create(null);
+		helper.onCreate(db);
+		Cursor cursor = db.query("sqlite_master", new String[] {"name"}, "type='table' AND name='" + BillingDB.TABLE_TRANSACTIONS + "'", null, null, null, null);
+		assertTrue(cursor.getCount() > 0);
+		cursor.close();
+		testColumn(db, BillingDB.COLUMN__ID, "TEXT", true);
+		testColumn(db, BillingDB.COLUMN_PRODUCT_ID, "TEXT", true);
+		testColumn(db, BillingDB.COLUMN_STATE, "TEXT", true);
+		testColumn(db, BillingDB.COLUMN_PURCHASE_TIME, "TEXT", true);
+		testColumn(db, BillingDB.COLUMN_DEVELOPER_PAYLOAD, "TEXT", true);
+		testColumn(db, BillingDB.COLUMN_SIGNED_DATA, "TEXT", true);
+		testColumn(db, BillingDB.COLUMN_SIGNATURE, "TEXT", true);
+	}
+	
+	@SmallTest
+	public void testDatabaseHelperOnUpgradeCurrentVersion() throws Exception {
+		BillingDB.DatabaseHelper helper = new BillingDB.DatabaseHelper(this.getContext());
+		SQLiteDatabase db = helper.getWritableDatabase();
+		helper.onUpgrade(db, BillingDB.DATABASE_VERSION, BillingDB.DATABASE_VERSION);
+
+	}
+	
+	private SQLiteDatabase createVersion1Database() {
+		final SQLiteDatabase db = SQLiteDatabase.create(null);
+		db.execSQL("CREATE TABLE " + BillingDB.TABLE_TRANSACTIONS + "(" +
+				BillingDB.COLUMN__ID + " TEXT PRIMARY KEY, " +
+				BillingDB.COLUMN_PRODUCT_ID + " INTEGER, " +
+				BillingDB.COLUMN_STATE + " TEXT, " +
+				BillingDB.COLUMN_PURCHASE_TIME + " TEXT, " +
+				BillingDB.COLUMN_DEVELOPER_PAYLOAD + " INTEGER)");
+		return db;
+	}
+	
+	@SmallTest
+	public void testDatabaseHelperOnUpgradeVersion1To2() throws Exception {
+		BillingDB.DatabaseHelper helper = new BillingDB.DatabaseHelper(this.getContext());
+		SQLiteDatabase db = createVersion1Database();
+		testColumn(db, BillingDB.COLUMN_SIGNED_DATA, "TEXT", false);
+		testColumn(db, BillingDB.COLUMN_SIGNATURE, "TEXT", false);
+		helper.onUpgrade(db, 1, 2);
+		testColumn(db, BillingDB.COLUMN_SIGNED_DATA, "TEXT", true);
+		testColumn(db, BillingDB.COLUMN_SIGNATURE, "TEXT", true);
+	}
 }