Bläddra i källkod

test(db): write test cases for db

Mustafa Arici 8 år sedan
förälder
incheckning
7756c0cc60
1 ändrade filer med 41 tillägg och 0 borttagningar
  1. 41 0
      db_test.go

+ 41 - 0
db_test.go

@@ -0,0 +1,41 @@
+package ovpm
+
+import "testing"
+
+func TestDBSetup(t *testing.T) {
+	// Initialize:
+	Testing = true
+
+	// Prepare:
+	// Test:
+
+	// Create database.
+	SetupDB("sqlite3", ":memory:")
+
+	// Is database created?
+	if db == nil {
+		t.Fatalf("database is expected to be not nil but it's nil")
+	}
+}
+
+func TestDBCease(t *testing.T) {
+	// Initialize:
+	Testing = true
+
+	// Prepare:
+	SetupDB("sqlite3", ":memory:")
+	user := DBUser{Username: "testUser"}
+	db.Save(&user)
+
+	// Test:
+	// Close database.
+	CeaseDB()
+
+	var users []DBUser
+	db.Find(&users)
+
+	// Is length zero?
+	if len(users) != 0 {
+		t.Fatalf("length of user should be 0 but it's %d", len(users))
+	}
+}