fleet/server/vulnerabilities/goval_dictionary/sync_test.go

package goval_dictionary

import (
	"github.com/fleetdm/fleet/v4/server/fleet"
	"github.com/fleetdm/fleet/v4/server/vulnerabilities/oval"
	"github.com/stretchr/testify/require"
	"testing"
	"time"
)

func TestSync(t *testing.T) {
	t.Run("#whatToDownload", func(t *testing.T) {
		osVersions := fleet.OSVersions{
			CountsUpdatedAt: time.Now(),
			OSVersions: []fleet.OSVersion{
				{
					HostsCount: 1,
					Platform:   "ubuntu",
					Name:       "Ubuntu 20.4.0",
				},
				{
					HostsCount: 1,
					Platform:   "amzn",
					Name:       "Amazon Linux 2.0.0",
				},
			},
		}

		result := whatToDownload(&osVersions)
		require.Len(t, result, 1)
		require.Contains(t, result, oval.NewPlatform("amzn", "Amazon Linux 2.0.0"))
		require.NotContains(t, result, oval.NewPlatform("ubuntu", "Ubuntu 20.4.0"))
	})
}