Initial commit

pkg/repos/find_test.go

@@ -0,0 +1,148 @@
+/*
+ * LURE - Linux User REpository
+ * Copyright (C) 2023 Elara Musayelyan
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package repos_test
+
+import (
+	"context"
+	"reflect"
+	"strings"
+	"testing"
+
+	"lure.sh/lure/internal/db"
+	"lure.sh/lure/internal/types"
+	"lure.sh/lure/pkg/repos"
+)
+
+func TestFindPkgs(t *testing.T) {
+	_, err := db.Open(":memory:")
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+	defer db.Close()
+
+	setCfgDirs(t)
+	defer removeCacheDir(t)
+
+	ctx := context.Background()
+
+	err = repos.Pull(ctx, []types.Repo{
+		{
+			Name: "default",
+			URL:  "https://github.com/Arsen6331/lure-repo.git",
+		},
+	})
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+
+	found, notFound, err := repos.FindPkgs([]string{"itd", "nonexistentpackage1", "nonexistentpackage2"})
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+
+	if !reflect.DeepEqual(notFound, []string{"nonexistentpackage1", "nonexistentpackage2"}) {
+		t.Errorf("Expected 'nonexistentpackage{1,2} not to be found")
+	}
+
+	if len(found) != 1 {
+		t.Errorf("Expected 1 package found, got %d", len(found))
+	}
+
+	itdPkgs, ok := found["itd"]
+	if !ok {
+		t.Fatalf("Expected 'itd' packages to be found")
+	}
+
+	if len(itdPkgs) < 2 {
+		t.Errorf("Expected two 'itd' packages to be found")
+	}
+
+	for i, pkg := range itdPkgs {
+		if !strings.HasPrefix(pkg.Name, "itd") {
+			t.Errorf("Expected package name of all found packages to start with 'itd', got %s on element %d", pkg.Name, i)
+		}
+	}
+}
+
+func TestFindPkgsEmpty(t *testing.T) {
+	_, err := db.Open(":memory:")
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+	defer db.Close()
+
+	setCfgDirs(t)
+	defer removeCacheDir(t)
+
+	err = db.InsertPackage(db.Package{
+		Name:       "test1",
+		Repository: "default",
+		Version:    "0.0.1",
+		Release:    1,
+		Description: db.NewJSON(map[string]string{
+			"en": "Test package 1",
+			"ru": "Проверочный пакет 1",
+		}),
+		Provides: db.NewJSON([]string{""}),
+	})
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+
+	err = db.InsertPackage(db.Package{
+		Name:       "test2",
+		Repository: "default",
+		Version:    "0.0.1",
+		Release:    1,
+		Description: db.NewJSON(map[string]string{
+			"en": "Test package 2",
+			"ru": "Проверочный пакет 2",
+		}),
+		Provides: db.NewJSON([]string{"test"}),
+	})
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+
+	found, notFound, err := repos.FindPkgs([]string{"test", ""})
+	if err != nil {
+		t.Fatalf("Expected no error, got %s", err)
+	}
+
+	if len(notFound) != 0 {
+		t.Errorf("Expected all packages to be found")
+	}
+
+	if len(found) != 1 {
+		t.Errorf("Expected 1 package found, got %d", len(found))
+	}
+
+	testPkgs, ok := found["test"]
+	if !ok {
+		t.Fatalf("Expected 'test' packages to be found")
+	}
+
+	if len(testPkgs) != 1 {
+		t.Errorf("Expected one 'test' package to be found, got %d", len(testPkgs))
+	}
+
+	if testPkgs[0].Name != "test2" {
+		t.Errorf("Expected 'test2' package, got '%s'", testPkgs[0].Name)
+	}
+}