rpmtools_test.go

package rpmtools

import (
	"fmt"
	"io"
	"io/ioutil"
	"net/http"
	"os"
	"path/filepath"
	"sort"
	"testing"
)

type test struct {
	url          string
	specName     string
	sourceTags   []string
	patchTags    []string
	bundledNames []string
}

var outDir string

var tests = []test{
	{
		"https://kojipkgs.fedoraproject.org//packages/python-urllib3/1.25.8/5.fc33/src/python-urllib3-1.25.8-5.fc33.src.rpm",
		"python-urllib3.spec",
		[]string{"https://github.com/urllib3/urllib3/archive/1.25.8/urllib3-1.25.8.tar.gz", "ssl_match_hostname_py3.py"},
		[]string{"CVE-2021-33503.patch"},
		[]string{},
	},
	{
		"https://kojipkgs.fedoraproject.org//packages/rizin/0.2.0/2.fc33/src/rizin-0.2.0-2.fc33.src.rpm",
		"rizin.spec",
		[]string{"https://github.com/rizinorg/rizin/releases/download/v0.2.0/rizin-src-v0.2.0.tar.xz"},
		[]string{"rizin-avoid-symbols-clashing.patch"},
		[]string{"spp", "sdb", "js0n", "openbsdregex", "tcc", "binutils", "vavrdisasm"},
	},
	{
		"https://kojipkgs.fedoraproject.org//packages/python-cvxopt/1.2.6/1.fc34/src/python-cvxopt-1.2.6-1.fc34.src.rpm",
		"python-cvxopt.spec",
		[]string{"https://github.com/cvxopt/cvxopt/archive/1.2.6/cvxopt-1.2.6.tar.gz"},
		[]string{"python-cvxopt-setup.patch", "python-cvxopt-doc.patch", "python-cvxopt-signed.patch"},
		[]string{"jquery"},
	},
	{
		"https://kojipkgs.fedoraproject.org//packages/yasm/1.3.0/12.fc33/src/yasm-1.3.0-12.fc33.src.rpm",
		"yasm.spec",
		[]string{"http://www.tortall.net/projects/yasm/releases/yasm-1.3.0.tar.gz"},
		[]string{"0001-Update-elf-objfmt.c.patch"},
		[]string{"md5-plumb", "md5-plumb"},
	},
	{
		"https://kojipkgs.fedoraproject.org//packages/redis/6.2.5/1.fc34/src/redis-6.2.5-1.fc34.src.rpm",
		"redis.spec",
		[]string{"https://download.redis.io/releases/redis-6.2.5.tar.gz", "redis.logrotate", "redis-sentinel.service", "redis.service", "redis-shutdown", "redis-limit-systemd", "macros.redis", "https://github.com/redis/redis-doc/archive/6a9c6310b7291d802cf2fbc45b742e97e2804413/redis-doc-6a9c631.tar.gz"},
		[]string{"0001-1st-man-pageis-for-redis-cli-redis-benchmark-redis-c.patch"},
		[]string{"hiredis", "jemalloc", "lua-libs", "linenoise", "lzf"},
	},
	{
		"https://kojipkgs.fedoraproject.org//packages/gnome-shell-extension-pop-shell/1.2.0%5E3.ab87042/2.fc33/src/gnome-shell-extension-pop-shell-1.2.0%5E3.ab87042-2.fc33.src.rpm",
		"gnome-shell-extension-pop-shell.spec",
		[]string{
			"https://github.com/pop-os/shell/archive/ab87042d2579c9ad9bb3271584b0281d97da7baa/pop-shell-ab87042.tar.gz",
			"50_org.gnome.desktop.wm.keybindings.pop-shell.gschema.override",
			"50_org.gnome.mutter.pop-shell.gschema.override",
			"50_org.gnome.mutter.wayland.pop-shell.gschema.override",
			"50_org.gnome.settings-daemon.plugins.media-keys.pop-shell.gschema.override",
			"50_org.gnome.shell.pop-shell.gschema.override",
		},
		[]string{"0001-Remove-schema-handling-from-transpile.sh.patch"},
		[]string{"npm(mathjs)", "npm(js-levenshtein)"},
	},
	{
		"https://kojipkgs.fedoraproject.org//packages/golang/1.16.6/2.fc34/src/golang-1.16.6-2.fc34.src.rpm",
		"golang.spec",
		[]string{"https://storage.googleapis.com/golang/go1.16.6.src.tar.gz", "fedora.go", "golang-gdbinit"},
		[]string{"0001-Don-t-use-the-bundled-tzdata-at-runtime-except-for-t.patch", "0002-syscall-expose-IfInfomsg.X__ifi_pad-on-s390x.patch", "0003-cmd-go-disable-Google-s-proxy-and-sumdb.patch", "ppc64le-vdso-fix.patch"},
		[]string{
			"golang(github.com/google/pprof)",
			"golang(github.com/ianlancetaylor/demangle)",
			"golang(golang.org/x/arch)",
			"golang(golang.org/x/crypto)",
			"golang(golang.org/x/mod)",
			"golang(golang.org/x/net)",
			"golang(golang.org/x/sys)",
			"golang(golang.org/x/text)",
			"golang(golang.org/x/tools)",
			"golang(golang.org/x/xerrors)",
		},
	},
}

func SortCompare(a []string, b []string) bool {
	if len(a) != len(b) {
		return false
	}

	sort.Strings(a)
	sort.Strings(b)

	for i, v := range a {
		if v != b[i] {
			return false
		}
	}
	return true
}

func equalSpecTags(a, b []SpecTag) bool {
	if len(a) != len(b) {
		return false
	}
	for i, v := range a {
		if v != b[i] {
			return false
		}
	}
	return true
}

func TestMain(m *testing.M) {
	var err error
	outDir, err = ioutil.TempDir("", "")
	if err != nil {
		return
	}
	m.Run()
	os.Remove(outDir)
}

func TestRpmSpecFromFile(t *testing.T) {
	for _, test := range tests {
		t.Run(fmt.Sprintf("url=%s", test.url), func(t *testing.T) {
			url := test.url
			resp, err := http.Get(url)
			if err != nil {
				t.Fatal(err)
			}
			defer resp.Body.Close()

			out, err := ioutil.TempFile("", "")
			if err != nil {
				t.Fatal(err)
			}
			defer out.Close()

			_, err = io.Copy(out, resp.Body)
			if err != nil {
				t.Fatal(err)
			}

			rpmSpec, err := RpmSpecFromFile(out.Name(), outDir)
			if err != nil {
				t.Fatal(err)
			}

			sourcesLocationExp := filepath.Join(outDir, "SOURCES")
			srpmLocationExp := filepath.Join(outDir, "SRPMS")
			buildLocationExp := filepath.Join(outDir, "BUILD")
			outLocationExp := outDir
			if rpmSpec.SourcesLocation != sourcesLocationExp {
				t.Errorf("SourcesLocation is wrong: %s, exp: %s", rpmSpec.SourcesLocation, sourcesLocationExp)
			}
			if rpmSpec.SrpmLocation != srpmLocationExp {
				t.Errorf("SRPMLocation is wrong: %s, exp: %s", rpmSpec.SrpmLocation, srpmLocationExp)
			}
			if rpmSpec.BuildLocation != buildLocationExp {
				t.Errorf("BuildLocation is wrong: %s, exp: %s", rpmSpec.SrpmLocation, buildLocationExp)
			}
			if rpmSpec.OutLocation != outLocationExp {
				t.Errorf("OutLocation is wrong: %s, exp: %s", rpmSpec.SrpmLocation, buildLocationExp)
			}
			if filepath.Base(rpmSpec.SpecLocation) != test.specName {
				t.Errorf("Spec file was wrongly detected: %s, exp: %s", filepath.Base(rpmSpec.SpecLocation), test.specName)
			}

			source0, err := rpmSpec.GetSource0()
			if err != nil {
				t.Fatal(err)
			}
			if source0 != test.sourceTags[0] {
				t.Errorf("Source0 was not correctly found: %s, exp: %s", source0, test.sourceTags[0])
			}

			if !equalSpecTags(rpmSpec.Tags["sources"], rpmSpec.SourcesTags) {
				t.Errorf("tags['sources'] different from SourcesTags")
			}
			if !equalSpecTags(rpmSpec.Tags["patches"], rpmSpec.PatchTags) {
				t.Errorf("tags['patches'] different from PatchTags")
			}
			if !equalSpecTags(rpmSpec.Tags["requires"], rpmSpec.RequiresTags) {
				t.Errorf("tags['requires'] different from RequiresTags")
			}

			sourcesTags := make([]string, len(rpmSpec.SourcesTags))
			for i, s := range rpmSpec.SourcesTags {
				sourcesTags[i] = s.TagValue
			}
			if !SortCompare(sourcesTags, test.sourceTags) {
				t.Errorf("Sources tags are wrong: %v, exp: %v", sourcesTags, test.sourceTags)
			}

			patchesTags := make([]string, len(rpmSpec.PatchTags))
			for i, s := range rpmSpec.PatchTags {
				patchesTags[i] = s.TagValue
			}
			if !SortCompare(patchesTags, test.patchTags) {
				t.Errorf("Patch tags are wrong: %v, exp: %v", patchesTags, test.patchTags)
			}

			bundledTags := make([]string, len(rpmSpec.BundledCodes))
			for i, s := range rpmSpec.BundledCodes {
				bundledTags[i] = s.Name
			}
			if !SortCompare(bundledTags, test.bundledNames) {
				t.Errorf("Provides: bundled(...) tags are wrong: %v, exp: %v", bundledTags, test.bundledNames)
			}

			rpmSpec.Cleanup()
		})
	}
}