feat: add support for multiple packages in one alr.sh

pkg/repos/pull_internal_test.go

@@ -0,0 +1,173 @@
+// ALR - Any Linux Repository
+// Copyright (C) 2025 Евгений Храмов
+//
+// 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
+
+import (
+	"context"
+	"io"
+	"os"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	"gitea.plemya-x.ru/Plemya-x/ALR/internal/config"
+	"gitea.plemya-x.ru/Plemya-x/ALR/internal/db"
+	"gitea.plemya-x.ru/Plemya-x/ALR/internal/types"
+)
+
+type TestALRConfig struct{}
+
+func (c *TestALRConfig) GetPaths(ctx context.Context) *config.Paths {
+	return &config.Paths{
+		DBPath: ":memory:",
+	}
+}
+
+func (c *TestALRConfig) Repos(ctx context.Context) []types.Repo {
+	return []types.Repo{
+		{
+			Name: "test",
+			URL:  "https://test",
+		},
+	}
+}
+
+func createReadCloserFromString(input string) io.ReadCloser {
+	reader := strings.NewReader(input)
+	return struct {
+		io.Reader
+		io.Closer
+	}{
+		Reader: reader,
+		Closer: io.NopCloser(reader),
+	}
+}
+
+func TestUpdatePkg(t *testing.T) {
+	type testCase struct {
+		name   string
+		file   string
+		verify func(context.Context, *db.Database)
+	}
+
+	repo := types.Repo{
+		Name: "test",
+		URL:  "https://test",
+	}
+
+	for _, tc := range []testCase{
+		{
+			name: "single package",
+			file: `name=foo
+version='0.0.1'
+release=1
+desc="main desc"
+deps=('sudo')
+build_deps=('golang')
+`,
+			verify: func(ctx context.Context, database *db.Database) {
+				result, err := database.GetPkgs(ctx, "1 = 1")
+				assert.NoError(t, err)
+				pkgCount := 0
+				for result.Next() {
+					var dbPkg db.Package
+					err = result.StructScan(&dbPkg)
+					if err != nil {
+						t.Errorf("Expected no error, got %s", err)
+					}
+
+					assert.Equal(t, "foo", dbPkg.Name)
+					assert.Equal(t, db.NewJSON(map[string]string{"": "main desc"}), dbPkg.Description)
+					assert.Equal(t, db.NewJSON(map[string][]string{"": {"sudo"}}), dbPkg.Depends)
+					pkgCount++
+				}
+				assert.Equal(t, 1, pkgCount)
+			},
+		},
+		{
+			name: "multiple package",
+			file: `basepkg_name=foo
+name=(
+	bar
+	buz
+)
+version='0.0.1'
+release=1
+desc="main desc"
+deps=('sudo')
+build_deps=('golang')
+		
+meta_bar() {
+	desc="foo desc"
+}
+			
+meta_buz() {
+	deps+=('doas')
+}
+`,
+			verify: func(ctx context.Context, database *db.Database) {
+				result, err := database.GetPkgs(ctx, "1 = 1")
+				assert.NoError(t, err)
+
+				pkgCount := 0
+				for result.Next() {
+					var dbPkg db.Package
+					err = result.StructScan(&dbPkg)
+					if err != nil {
+						t.Errorf("Expected no error, got %s", err)
+					}
+					if dbPkg.Name == "bar" {
+						assert.Equal(t, db.NewJSON(map[string]string{"": "foo desc"}), dbPkg.Description)
+						assert.Equal(t, db.NewJSON(map[string][]string{"": {"sudo"}}), dbPkg.Depends)
+					}
+
+					if dbPkg.Name == "buz" {
+						assert.Equal(t, db.NewJSON(map[string]string{"": "main desc"}), dbPkg.Description)
+						assert.Equal(t, db.NewJSON(map[string][]string{"": {"sudo", "doas"}}), dbPkg.Depends)
+					}
+					pkgCount++
+				}
+				assert.Equal(t, 2, pkgCount)
+			},
+		},
+	} {
+		t.Run(tc.name, func(t *testing.T) {
+			cfg := &TestALRConfig{}
+			ctx := context.Background()
+
+			database := db.New(&TestALRConfig{})
+			database.Init(ctx)
+
+			rs := New(cfg, database)
+
+			path, err := os.MkdirTemp("", "test-update-pkg")
+			assert.NoError(t, err)
+			defer os.RemoveAll(path)
+
+			runner, err := rs.processRepoChangesRunner(path, path)
+			assert.NoError(t, err)
+
+			err = rs.updatePkg(ctx, repo, runner, createReadCloserFromString(
+				tc.file,
+			))
+			assert.NoError(t, err)
+
+			tc.verify(ctx, database)
+		})
+	}
+}