mirror of
https://github.com/empayre/fleet.git
synced 2024-11-06 17:05:18 +00:00
67 lines
2.8 KiB
Go
67 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/fleetdm/fleet/v4/orbit/pkg/packaging"
|
|
"github.com/fleetdm/fleet/v4/orbit/pkg/update"
|
|
"github.com/fleetdm/fleet/v4/pkg/nettest"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPackage(t *testing.T) {
|
|
nettest.Run(t)
|
|
|
|
updateOpt := update.DefaultOptions
|
|
updateOpt.RootDirectory = t.TempDir()
|
|
updatesData, err := packaging.InitializeUpdates(updateOpt)
|
|
require.NoError(t, err)
|
|
|
|
// --type is required
|
|
runAppCheckErr(t, []string{"package", "deb"}, "Required flag \"type\" not set")
|
|
|
|
// if you provide -fleet-url & --enroll-secret are required together
|
|
runAppCheckErr(t, []string{"package", "--type=deb", "--fleet-url=https://localhost:8080"}, "--enroll-secret and --fleet-url must be provided together")
|
|
runAppCheckErr(t, []string{"package", "--type=deb", "--enroll-secret=foobar"}, "--enroll-secret and --fleet-url must be provided together")
|
|
|
|
// --insecure and --fleet-certificate are mutually exclusive
|
|
runAppCheckErr(t, []string{"package", "--type=deb", "--insecure", "--fleet-certificate=test123"}, "--insecure and --fleet-certificate may not be provided together")
|
|
|
|
// Test invalid PEM file provided in --fleet-certificate.
|
|
certDir := t.TempDir()
|
|
fleetCertificate := filepath.Join(certDir, "fleet.pem")
|
|
err = os.WriteFile(fleetCertificate, []byte("undefined"), os.FileMode(0o644))
|
|
require.NoError(t, err)
|
|
runAppCheckErr(t, []string{"package", "--type=deb", fmt.Sprintf("--fleet-certificate=%s", fleetCertificate)}, fmt.Sprintf("failed to read certificate %q: invalid PEM file", fleetCertificate))
|
|
|
|
if runtime.GOOS != "linux" {
|
|
runAppCheckErr(t, []string{"package", "--type=msi", "--native-tooling"}, "native tooling is only available in Linux")
|
|
}
|
|
|
|
t.Run("deb", func(t *testing.T) {
|
|
runAppForTest(t, []string{"package", "--type=deb", "--insecure", "--disable-open-folder"})
|
|
info, err := os.Stat(fmt.Sprintf("fleet-osquery_%s_amd64.deb", updatesData.OrbitVersion))
|
|
require.NoError(t, err)
|
|
require.Greater(t, info.Size(), int64(0)) // TODO verify contents
|
|
})
|
|
|
|
t.Run("rpm", func(t *testing.T) {
|
|
runAppForTest(t, []string{"package", "--type=rpm", "--insecure", "--disable-open-folder"})
|
|
info, err := os.Stat(fmt.Sprintf("fleet-osquery-%s.x86_64.rpm", updatesData.OrbitVersion))
|
|
require.NoError(t, err)
|
|
require.Greater(t, info.Size(), int64(0)) // TODO verify contents
|
|
})
|
|
|
|
// fleet-osquery.msi
|
|
// runAppForTest(t, []string{"package", "--type=msi", "--insecure"}) TODO: this is currently failing on Github runners due to permission issues
|
|
// info, err = os.Stat("orbit-osquery_0.0.3.msi")
|
|
// require.NoError(t, err)
|
|
// require.Greater(t, info.Size(), int64(0))
|
|
|
|
// runAppForTest(t, []string{"package", "--type=pkg", "--insecure"}) TODO: had a hard time getting xar installed on Ubuntu
|
|
}
|