소스 검색

test(db): write test cases for db

Mustafa Arici 8 년 전
부모
커밋
7756c0cc60
1개의 변경된 파일41개의 추가작업 그리고 0개의 파일을 삭제
  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))
+	}
+}