fleet/orbit/pkg/packaging/windows_test.go

92 lines
3.0 KiB
Go

package packaging
import (
"fmt"
"path/filepath"
"testing"
"time"
"github.com/stretchr/testify/require"
)
func TestCreateVersionInfo(t *testing.T) {
manifestPath := filepath.Join(t.TempDir(), "somefile.json")
t.Run("invalid version parts", func(t *testing.T) {
parts := []string{"1", "a", "3", "c"}
result, err := createVersionInfo(parts, manifestPath)
require.ErrorContains(t, err, "error parsing version part")
require.Nil(t, result)
})
t.Run("creates a VersionInfo struct", func(t *testing.T) {
parts := []string{"1", "2", "3", "0"}
result, err := createVersionInfo(parts, manifestPath)
require.NoError(t, err)
require.NotNil(t, result)
require.Equal(t, result.FixedFileInfo.FileVersion.Major, 1)
require.Equal(t, result.FixedFileInfo.FileVersion.Minor, 2)
require.Equal(t, result.FixedFileInfo.FileVersion.Patch, 3)
require.Equal(t, result.FixedFileInfo.FileVersion.Build, 0)
require.Equal(t, result.FixedFileInfo.ProductVersion.Major, 1)
require.Equal(t, result.FixedFileInfo.ProductVersion.Minor, 2)
require.Equal(t, result.FixedFileInfo.ProductVersion.Patch, 3)
require.Equal(t, result.FixedFileInfo.ProductVersion.Build, 0)
require.Equal(t, result.FixedFileInfo.FileFlagsMask, "3f")
require.Equal(t, result.FixedFileInfo.FileFlags, "00")
require.Equal(t, result.FixedFileInfo.FileOS, "040004")
require.Equal(t, result.FixedFileInfo.FileType, "01")
require.Equal(t, result.FixedFileInfo.FileSubType, "00")
require.Equal(t, result.StringFileInfo.Comments, "Fleet osquery")
require.Equal(t, result.StringFileInfo.CompanyName, "Fleet Device Management (fleetdm.com)")
require.Equal(t, result.StringFileInfo.FileDescription, "Fleet osquery installer")
require.Equal(t, result.StringFileInfo.FileVersion, "1.2.3.0")
require.Equal(t, result.StringFileInfo.LegalCopyright, fmt.Sprintf("%d Fleet Device Management Inc.", time.Now().Year()))
require.Equal(t, result.StringFileInfo.ProductName, "Fleet osquery")
require.Equal(t, result.StringFileInfo.ProductVersion, "1.2.3.0")
require.Equal(t, result.ManifestPath, manifestPath)
})
}
func TestWriteResourceSyso(t *testing.T) {
t.Run("removes intermediary manifest.xml file", func(t *testing.T) {
path := t.TempDir()
opt := Options{Version: "1.2.3"}
err := writeResourceSyso(opt, path)
require.NoError(t, err)
require.NoFileExists(t, filepath.Join(path, "manifest.xml"))
})
}
func TestSanitizeVersion(t *testing.T) {
testCases := []struct {
Version string
Parts []string
ErrorsOut bool
}{
{Version: "4.13.0", Parts: []string{"4", "13", "0", "0"}},
{Version: "4.13.0.1", Parts: []string{"4", "13", "0", "1"}},
{Version: "4.13.0.1.2", Parts: []string{"4", "13", "0", "1"}},
{Version: "4", ErrorsOut: true},
{Version: "4.13", ErrorsOut: true},
{Version: "bad bad", ErrorsOut: true},
}
for _, tC := range testCases {
result, err := SanitizeVersion(tC.Version)
if tC.ErrorsOut {
require.Error(t, err)
}
require.Equal(t, tC.Parts, result)
}
}