mirror of
https://github.com/roddhjav/apparmor.d.git
synced 2025-01-30 23:05:11 +01:00
build: define new unified build interfaces.
This commit is contained in:
parent
b6aed5cd8d
commit
62099d325d
2 changed files with 111 additions and 0 deletions
47
pkg/prebuild/cfg/core.go
Normal file
47
pkg/prebuild/cfg/core.go
Normal file
|
@ -0,0 +1,47 @@
|
||||||
|
// apparmor.d - Full set of apparmor profiles
|
||||||
|
// Copyright (C) 2021-2024 Alexandre Pujol <alexandre@pujol.io>
|
||||||
|
// SPDX-License-Identifier: GPL-2.0-only
|
||||||
|
|
||||||
|
package cfg
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
type BaseInterface interface {
|
||||||
|
Message() string
|
||||||
|
Name() string
|
||||||
|
Usage() string
|
||||||
|
}
|
||||||
|
|
||||||
|
type Base struct {
|
||||||
|
Msg string
|
||||||
|
Keyword string
|
||||||
|
Help string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b Base) Name() string {
|
||||||
|
return b.Keyword
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b Base) Usage() string {
|
||||||
|
return b.Help
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b Base) Message() string {
|
||||||
|
return b.Msg
|
||||||
|
}
|
||||||
|
|
||||||
|
func Help[T BaseInterface](name string, tasks map[string]T) string {
|
||||||
|
res := fmt.Sprintf("%s tasks:\n", name)
|
||||||
|
for _, t := range tasks {
|
||||||
|
res += fmt.Sprintf(" %s - %s\n", t.Name(), t.Message())
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
|
||||||
|
func Usage[T BaseInterface](name string, tasks map[string]T) string {
|
||||||
|
res := fmt.Sprintf("%s\n", name)
|
||||||
|
for _, t := range tasks {
|
||||||
|
res += fmt.Sprintf(" %s\n", t.Usage())
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
64
pkg/prebuild/cfg/core_test.go
Normal file
64
pkg/prebuild/cfg/core_test.go
Normal file
|
@ -0,0 +1,64 @@
|
||||||
|
// apparmor.d - Full set of apparmor profiles
|
||||||
|
// Copyright (C) 2021-2024 Alexandre Pujol <alexandre@pujol.io>
|
||||||
|
// SPDX-License-Identifier: GPL-2.0-only
|
||||||
|
|
||||||
|
package cfg
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestBase_Helpers(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
b Base
|
||||||
|
want string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "base",
|
||||||
|
b: Base{Keyword: "test", Help: "test", Msg: "test"},
|
||||||
|
want: "test",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if got := tt.b.Name(); got != tt.want {
|
||||||
|
t.Errorf("Base.Name() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
if got := tt.b.Usage(); got != tt.want {
|
||||||
|
t.Errorf("Base.Usage() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
if got := tt.b.Message(); got != tt.want {
|
||||||
|
t.Errorf("Base.Message() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestHelp(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
tasks map[string]Base
|
||||||
|
want string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "one",
|
||||||
|
tasks: map[string]Base{
|
||||||
|
"one": {Keyword: "one", Help: "one", Msg: "one"},
|
||||||
|
"two": {Keyword: "two", Help: "two", Msg: "two"},
|
||||||
|
},
|
||||||
|
want: `one`,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if got := Help(tt.name, tt.tasks); !strings.Contains(got, tt.want) {
|
||||||
|
t.Errorf("Help() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
if got := Usage(tt.name, tt.tasks); !strings.Contains(got, tt.want) {
|
||||||
|
t.Errorf("Usage() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue