Add tests for internal/db package

internal/db/db.go

@@ -109,7 +109,7 @@ func GetPkgs(db *sqlx.DB, where string, args ...any) (*sqlx.Rows, error) {
 // GetPkg returns a single package that match the where conditions
 func GetPkg(db *sqlx.DB, where string, args ...any) (*Package, error) {
 	out := &Package{}
-	err := db.Get(out, "SELECT * FROM pkgs WHERE "+where+"LIMIT 1", args...)
+	err := db.Get(out, "SELECT * FROM pkgs WHERE "+where+" LIMIT 1", args...)
 	return out, err
 }
 

internal/db/db_test.go

@@ -0,0 +1,237 @@
+package db_test
+
+import (
+	"reflect"
+	"strings"
+	"testing"
+
+	"github.com/jmoiron/sqlx"
+	"go.arsenm.dev/lure/internal/db"
+	"modernc.org/sqlite"
+)
+
+var testPkg = db.Package{
+	Name:          "test",
+	Version:       "0.0.1",
+	Release:       1,
+	Epoch:         2,
+	Description:   "Test package",
+	Homepage:      "https://lure.arsenm.dev",
+	Maintainer:    "Arsen Musayelyan <arsen@arsenm.dev>",
+	Architectures: db.NewJSON([]string{"arm64", "amd64"}),
+	Licenses:      db.NewJSON([]string{"GPL-3.0-or-later"}),
+	Provides:      db.NewJSON([]string{"test"}),
+	Conflicts:     db.NewJSON([]string{"test"}),
+	Replaces:      db.NewJSON([]string{"test-old"}),
+	Depends: db.NewJSON(map[string][]string{
+		"": {"sudo"},
+	}),
+	BuildDepends: db.NewJSON(map[string][]string{
+		"":     {"golang"},
+		"arch": {"go"},
+	}),
+	Repository: "default",
+}
+
+func getDB(t *testing.T) (*sqlx.DB, error) {
+	t.Helper()
+
+	gdb, err := sqlx.Open("sqlite", ":memory:")
+	if err != nil {
+		return nil, err
+	}
+
+	err = db.Init(gdb)
+	return gdb, err
+}
+
+func TestInit(t *testing.T) {
+	gdb, err := sqlx.Open("sqlite", ":memory:")
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+	defer gdb.Close()
+
+	err = db.Init(gdb)
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+
+	_, err = gdb.Exec("SELECT * FROM pkgs")
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+}
+
+func TestInsertPackage(t *testing.T) {
+	gdb, err := getDB(t)
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+	defer gdb.Close()
+
+	err = db.InsertPackage(gdb, testPkg)
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+
+	dbPkg := db.Package{}
+	err = sqlx.Get(gdb, &dbPkg, "SELECT * FROM pkgs WHERE name = 'test' AND repository = 'default'")
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+
+	if !reflect.DeepEqual(testPkg, dbPkg) {
+		t.Errorf("Expected test package to be the same as database package")
+	}
+}
+
+func TestGetPkgs(t *testing.T) {
+	gdb, err := getDB(t)
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+	defer gdb.Close()
+
+	x1 := testPkg
+	x1.Name = "x1"
+	x2 := testPkg
+	x2.Name = "x2"
+
+	err = db.InsertPackage(gdb, x1)
+	if err != nil {
+		t.Errorf("Expected no error, got %s", err)
+	}
+
+	err = db.InsertPackage(gdb, x2)
+	if err != nil {
+		t.Errorf("Expected no error, got %s", err)
+	}
+
+	result, err := db.GetPkgs(gdb, "name LIKE 'x%'")
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+
+	for result.Next() {
+		var dbPkg db.Package
+		err = result.StructScan(&dbPkg)
+		if err != nil {
+			t.Errorf("Expected no error, got %s", err)
+		}
+
+		if !strings.HasPrefix(dbPkg.Name, "x") {
+			t.Errorf("Expected package name to start with 'x', got %s", dbPkg.Name)
+		}
+	}
+}
+
+func TestGetPkg(t *testing.T) {
+	gdb, err := getDB(t)
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+	defer gdb.Close()
+
+	x1 := testPkg
+	x1.Name = "x1"
+	x2 := testPkg
+	x2.Name = "x2"
+
+	err = db.InsertPackage(gdb, x1)
+	if err != nil {
+		t.Errorf("Expected no error, got %s", err)
+	}
+
+	err = db.InsertPackage(gdb, x2)
+	if err != nil {
+		t.Errorf("Expected no error, got %s", err)
+	}
+
+	pkg, err := db.GetPkg(gdb, "name LIKE 'x%' ORDER BY name")
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+
+	if pkg.Name != "x1" {
+		t.Errorf("Expected x1 package, got %s", pkg.Name)
+	}
+
+	if !reflect.DeepEqual(*pkg, x1) {
+		t.Errorf("Expected x1 to be %v, got %v", x1, *pkg)
+	}
+}
+
+func TestDeletePkgs(t *testing.T) {
+	gdb, err := getDB(t)
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+	defer gdb.Close()
+
+	x1 := testPkg
+	x1.Name = "x1"
+	x2 := testPkg
+	x2.Name = "x2"
+
+	err = db.InsertPackage(gdb, x1)
+	if err != nil {
+		t.Errorf("Expected no error, got %s", err)
+	}
+
+	err = db.InsertPackage(gdb, x2)
+	if err != nil {
+		t.Errorf("Expected no error, got %s", err)
+	}
+
+	err = db.DeletePkgs(gdb, "name = 'x1'")
+	if err != nil {
+		t.Errorf("Expected no error, got %s", err)
+	}
+
+	var dbPkg db.Package
+	err = gdb.Get(&dbPkg, "SELECT * FROM pkgs WHERE name LIKE 'x%' ORDER BY name LIMIT 1;")
+	if err != nil {
+		t.Errorf("Expected no error, got %s", err)
+	}
+
+	if dbPkg.Name != "x2" {
+		t.Errorf("Expected x2 package, got %s", dbPkg.Name)
+	}
+}
+
+func TestJsonArrayContains(t *testing.T) {
+	sqlite.MustRegisterScalarFunction("json_array_contains", 2, db.JsonArrayContains)
+
+	gdb, err := getDB(t)
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+	defer gdb.Close()
+
+	x1 := testPkg
+	x1.Name = "x1"
+	x2 := testPkg
+	x2.Name = "x2"
+	x2.Provides.Val = append(x2.Provides.Val, "x")
+
+	err = db.InsertPackage(gdb, x1)
+	if err != nil {
+		t.Errorf("Expected no error, got %s", err)
+	}
+
+	err = db.InsertPackage(gdb, x2)
+	if err != nil {
+		t.Errorf("Expected no error, got %s", err)
+	}
+
+	var dbPkg db.Package
+	err = gdb.Get(&dbPkg, "SELECT * FROM pkgs WHERE json_array_contains(provides, 'x');")
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+
+	if dbPkg.Name != "x2" {
+		t.Errorf("Expected x2 package, got %s", dbPkg.Name)
+	}
+}