apparmor.d/pkg/prebuild/cli/cli_test.go

// apparmor.d - Full set of apparmor profiles
// Copyright (C) 2023-2024 Alexandre Pujol <alexandre@pujol.io>
// SPDX-License-Identifier: GPL-2.0-only

package cli

import (
	"os"
	"os/exec"
	"testing"

	"github.com/roddhjav/apparmor.d/pkg/paths"
	"github.com/roddhjav/apparmor.d/pkg/prebuild"
	"github.com/roddhjav/apparmor.d/pkg/prebuild/builder"
	"github.com/roddhjav/apparmor.d/pkg/prebuild/prepare"
)

func setTestBuildDirectories(name string) {
	testRoot := paths.New("/tmp/tests")
	prebuild.Root = testRoot.Join(name)
	prebuild.RootApparmord = prebuild.Root.Join("apparmor.d")
}

func chdirGitRoot() {
	cmd := exec.Command("git", "rev-parse", "--show-toplevel")
	out, err := cmd.Output()
	if err != nil {
		panic(err)
	}
	root := string(out)[0 : len(out)-1]
	if err := os.Chdir(root); err != nil {
		panic(err)
	}
}

func Test_Prebuild(t *testing.T) {
	tests := []struct {
		name     string
		wantErr  bool
		full     bool
		complain bool
		enforce  bool
		dist     string
	}{
		{
			name:     "Build for Archlinux",
			wantErr:  false,
			full:     false,
			complain: true,
			enforce:  false,
			dist:     "arch",
		},
		{
			name:     "Build for Ubuntu",
			wantErr:  false,
			full:     true,
			complain: false,
			enforce:  true,
			dist:     "ubuntu",
		},
		{
			name:     "Build for Debian",
			wantErr:  false,
			full:     true,
			complain: false,
			enforce:  false,
			dist:     "debian",
		},
		{
			name:     "Build for OpenSUSE Tumbleweed",
			wantErr:  false,
			full:     true,
			complain: true,
			enforce:  false,
			dist:     "opensuse",
		},
	}
	chdirGitRoot()
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			setTestBuildDirectories(tt.dist)
			prebuild.Distribution = tt.dist
			prepare.Prepares = []prepare.Task{}
			prepare.Register(
				"synchronise", "ignore", "merge",
				"configure", "setflags", "systemd-default",
			)

			if full {
				prepare.Register("fsp")
				builder.Register("fsp")
			} else {
				prepare.Register("systemd-early")
			}

			if complain {
				builder.Register("complain")
			} else if enforce {
				builder.Register("enforce")
			}

			if err := Prepare(); (err != nil) != tt.wantErr {
				t.Errorf("Prepare() error = %v, wantErr %v", err, tt.wantErr)
			}
			if err := Build(); (err != nil) != tt.wantErr {
				t.Errorf("Build() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}