initial public commit

daos/table_test.go

@@ -0,0 +1,81 @@
+package daos_test
+
+import (
+	"testing"
+
+	"github.com/pocketbase/pocketbase/tests"
+	"github.com/pocketbase/pocketbase/tools/list"
+)
+
+func TestHasTable(t *testing.T) {
+	app, _ := tests.NewTestApp()
+	defer app.Cleanup()
+
+	scenarios := []struct {
+		tableName string
+		expected  bool
+	}{
+		{"", false},
+		{"test", false},
+		{"_admins", true},
+		{"demo3", true},
+		{"DEMO3", true}, // table names are case insensitives by default
+	}
+
+	for i, scenario := range scenarios {
+		result := app.Dao().HasTable(scenario.tableName)
+		if result != scenario.expected {
+			t.Errorf("(%d) Expected %v, got %v", i, scenario.expected, result)
+		}
+	}
+}
+
+func TestGetTableColumns(t *testing.T) {
+	app, _ := tests.NewTestApp()
+	defer app.Cleanup()
+
+	scenarios := []struct {
+		tableName string
+		expected  []string
+	}{
+		{"", nil},
+		{"_params", []string{"id", "key", "value", "created", "updated"}},
+	}
+
+	for i, scenario := range scenarios {
+		columns, _ := app.Dao().GetTableColumns(scenario.tableName)
+
+		if len(columns) != len(scenario.expected) {
+			t.Errorf("(%d) Expected columns %v, got %v", i, scenario.expected, columns)
+		}
+
+		for _, c := range columns {
+			if !list.ExistInSlice(c, scenario.expected) {
+				t.Errorf("(%d) Didn't expect column %s", i, c)
+			}
+		}
+	}
+}
+
+func TestDeleteTable(t *testing.T) {
+	app, _ := tests.NewTestApp()
+	defer app.Cleanup()
+
+	scenarios := []struct {
+		tableName   string
+		expectError bool
+	}{
+		{"", true},
+		{"test", true},
+		{"_admins", false},
+		{"demo3", false},
+	}
+
+	for i, scenario := range scenarios {
+		err := app.Dao().DeleteTable(scenario.tableName)
+		hasErr := err != nil
+		if hasErr != scenario.expectError {
+			t.Errorf("(%d) Expected hasErr %v, got %v", i, scenario.expectError, hasErr)
+		}
+	}
+}