aptly/cmd/snapshot_pull_variant_test.go

package cmd

import (
	"encoding/json"
	"os"
	"path/filepath"

	ctx "github.com/aptly-dev/aptly/context"
	"github.com/aptly-dev/aptly/deb"
	"github.com/aptly-dev/aptly/utils"
	"github.com/smira/flag"

	"gopkg.in/check.v1"
)

type SnapshotPullVariantSuite struct{}

var _ = check.Suite(&SnapshotPullVariantSuite{})

// Exercise the real pull entry point and persisted references, without package
// files or mirrors. Architecture selection remains base-architecture selection.
func (*SnapshotPullVariantSuite) TestSnapshotPullArchitectureVariant(c *check.C) {
	normal := &deb.Package{Name: "pull-test", Version: "1", Architecture: "amd64"}
	variant := &deb.Package{Name: "pull-test", Version: "1", Architecture: "amd64", ArchitectureVariant: "amd64v3"}
	normal2 := &deb.Package{Name: "pull-test", Version: "2", Architecture: "amd64"}
	variant2 := &deb.Package{Name: "pull-test", Version: "2", Architecture: "amd64", ArchitectureVariant: "amd64v3"}
	cases := []struct {
		name                 string
		dest, source, want   []*deb.Package
		noRemove, allMatches bool
	}{
		{"normal", nil, []*deb.Package{normal}, []*deb.Package{normal}, false, false},
		{"variant", nil, []*deb.Package{variant}, []*deb.Package{variant}, false, false},
		{"both", nil, []*deb.Package{normal, variant}, []*deb.Package{normal, variant}, false, false},
		{"variant_into_normal", []*deb.Package{normal}, []*deb.Package{variant}, []*deb.Package{normal, variant}, false, false},
		{"normal_into_variant", []*deb.Package{variant}, []*deb.Package{normal}, []*deb.Package{normal, variant}, false, false},
		{"both_all_matches", nil, []*deb.Package{normal, variant}, []*deb.Package{normal, variant}, false, true},
		{"both_no_remove", nil, []*deb.Package{normal, variant}, []*deb.Package{normal, variant}, true, false},
		{"variant_into_normal_no_remove", []*deb.Package{normal}, []*deb.Package{variant}, []*deb.Package{normal, variant}, true, false},
		{"normal_into_variant_no_remove", []*deb.Package{variant}, []*deb.Package{normal}, []*deb.Package{normal, variant}, true, false},
		{"variant_into_normal_all_matches", []*deb.Package{normal}, []*deb.Package{variant}, []*deb.Package{normal, variant}, false, true},
		{"normal_into_variant_all_matches", []*deb.Package{variant}, []*deb.Package{normal}, []*deb.Package{normal, variant}, false, true},
		{"normal_replace", []*deb.Package{normal}, []*deb.Package{normal2}, []*deb.Package{normal2}, false, false},
		{"normal_no_remove", []*deb.Package{normal}, []*deb.Package{normal2}, []*deb.Package{normal, normal2}, true, false},
		{"normal_latest", nil, []*deb.Package{normal, normal2}, []*deb.Package{normal2}, false, false},
		{"normal_all_matches", nil, []*deb.Package{normal, normal2}, []*deb.Package{normal, normal2}, false, true},
		{"variant_replace_preserves_normal", []*deb.Package{normal, variant}, []*deb.Package{variant2}, []*deb.Package{normal, variant2}, false, false},
		{"latest_per_identity", nil, []*deb.Package{normal, normal2, variant, variant2}, []*deb.Package{normal2, variant2}, false, false},
		{"all_versions_both_identities", nil, []*deb.Package{normal, normal2, variant, variant2}, []*deb.Package{normal, normal2, variant, variant2}, false, true},
	}
	for _, tc := range cases {
		c.Logf("case: %s", tc.name)
		func() {
			savedConfig := utils.Config
			defer func() { utils.Config = savedConfig }()
			dir := c.MkDir()
			configPath := filepath.Join(dir, "aptly.conf")
			configData, err := json.Marshal(map[string]interface{}{"rootDir": dir, "architectures": []string{}})
			c.Assert(err, check.IsNil)
			c.Assert(os.WriteFile(configPath, configData, 0600), check.IsNil)
			flags := flag.NewFlagSet("pull-test", flag.ContinueOnError)
			flags.String("config", configPath, "")
			// Empty destinations need explicit architecture; populated destinations
			// exercise automatic base-architecture discovery.
			arch := ""
			if len(tc.dest) == 0 {
				arch = "amd64"
			}
			flags.String("architectures", arch, "")
			flags.Bool("no-lock", false, "")
			flags.Int("db-open-attempts", 1, "")
			flags.Bool("no-deps", true, "")
			flags.Bool("no-remove", tc.noRemove, "")
			flags.Bool("all-matches", tc.allMatches, "")
			flags.Bool("dry-run", false, "")
			testContext, err := ctx.NewContext(flags)
			c.Assert(err, check.IsNil)
			defer testContext.Shutdown()
			factory := testContext.NewCollectionFactory()
			refs := func(packages []*deb.Package) *deb.PackageRefList {
				list := deb.NewPackageList()
				for _, p := range packages {
					c.Assert(list.Add(p), check.IsNil)
				}
				return deb.NewPackageRefListFromPackageList(list)
			}
			for _, p := range append(append([]*deb.Package{}, tc.dest...), tc.source...) {
				c.Assert(factory.PackageCollection().Update(p), check.IsNil)
			}
			for name, packages := range map[string][]*deb.Package{"base": tc.dest, "source": tc.source} {
				c.Assert(factory.SnapshotCollection().Add(deb.NewSnapshotFromRefList(name, nil, refs(packages), "")), check.IsNil)
			}
			savedContext := context
			context = testContext
			defer func() { context = savedContext }()
			c.Assert(aptlySnapshotPull(makeCmdSnapshotPull(), []string{"base", "source", "result", "pull-test"}), check.IsNil)
			snapshots := testContext.NewCollectionFactory().SnapshotCollection()
			result, err := snapshots.ByName("result")
			c.Assert(err, check.IsNil)
			c.Assert(snapshots.LoadComplete(result), check.IsNil)
			got, want := result.RefList().Strings(), refs(tc.want).Strings()
			c.Check(got, check.DeepEquals, want, check.Commentf("case %s: persisted package identities", tc.name))
		}()
	}
}