mirror of
https://github.com/fleetdm/fleet
synced 2026-05-14 04:28:42 +00:00
6ff2c449cf
Addresses https://github.com/fleetdm/fleet/issues/10112 Improved test coverage, also fixed a bug related to vulnerability processing for mac office apps.
122 lines
3.2 KiB
Go
122 lines
3.2 KiB
Go
package io
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFSClient(t *testing.T) {
|
|
t.Run("Delete", func(t *testing.T) {
|
|
path := t.TempDir()
|
|
sut := NewFSClient(path)
|
|
|
|
t.Run("file does not exists", func(t *testing.T) {
|
|
err := sut.Delete(MetadataFileName{filename: "something"})
|
|
require.Error(t, err)
|
|
})
|
|
})
|
|
|
|
t.Run("MacOfficeReleaseNotes", func(t *testing.T) {
|
|
t.Run("directory does not exists", func(t *testing.T) {
|
|
sut := NewFSClient("asdf")
|
|
_, err := sut.MSRCBulletins()
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("when files contain the wrong date format", func(t *testing.T) {
|
|
path := t.TempDir()
|
|
sut := NewFSClient(path)
|
|
|
|
asset1 := filepath.Join(path, fmt.Sprintf("%smacoffice-2022000_10_10.json", macOfficeReleaseNotesPrefix))
|
|
|
|
f1, err := os.Create(asset1)
|
|
require.NoError(t, err)
|
|
f1.Close()
|
|
|
|
_, err = sut.MacOfficeReleaseNotes()
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("returns a list of file matching the MacOffice file prefix", func(t *testing.T) {
|
|
path := t.TempDir()
|
|
sut := NewFSClient(path)
|
|
|
|
file1 := filepath.Join(path, "my_lyrics-2021_10_10.json")
|
|
asset1 := filepath.Join(path, fmt.Sprintf("%smacoffice-2022_10_10.json", macOfficeReleaseNotesPrefix))
|
|
asset2 := filepath.Join(path, fmt.Sprintf("%smacoffice_11-2022_10_10.json", macOfficeReleaseNotesPrefix))
|
|
|
|
f1, err := os.Create(asset1)
|
|
require.NoError(t, err)
|
|
f1.Close()
|
|
|
|
f2, err := os.Create(asset2)
|
|
require.NoError(t, err)
|
|
f2.Close()
|
|
|
|
f3, err := os.Create(file1)
|
|
require.NoError(t, err)
|
|
f3.Close()
|
|
|
|
r, err := sut.MacOfficeReleaseNotes()
|
|
require.NoError(t, err)
|
|
|
|
a, err := NewMacOfficeRelNotesMetadata(filepath.Base(file1))
|
|
require.NoError(t, err)
|
|
b, err := NewMacOfficeRelNotesMetadata(filepath.Base(asset1))
|
|
require.NoError(t, err)
|
|
c, err := NewMacOfficeRelNotesMetadata(filepath.Base(asset2))
|
|
require.NoError(t, err)
|
|
|
|
require.NotContains(t, r, a)
|
|
require.Contains(t, r, b)
|
|
require.Contains(t, r, c)
|
|
})
|
|
})
|
|
|
|
t.Run("#MSRCBulletins", func(t *testing.T) {
|
|
t.Run("directory does not exists", func(t *testing.T) {
|
|
sut := NewFSClient("asdf")
|
|
_, err := sut.MSRCBulletins()
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("returns a list of file matching the MSRC file prefix", func(t *testing.T) {
|
|
path := t.TempDir()
|
|
sut := NewFSClient(path)
|
|
|
|
file1 := filepath.Join(path, "my_lyrics-2021_10_10.json")
|
|
bulletin1 := filepath.Join(path, fmt.Sprintf("%sWindows_10-2022_10_10.json", mSRCFilePrefix))
|
|
bulletin2 := filepath.Join(path, fmt.Sprintf("%sWindows_11-2022_10_10.json", mSRCFilePrefix))
|
|
|
|
f1, err := os.Create(bulletin1)
|
|
require.NoError(t, err)
|
|
f1.Close()
|
|
|
|
f2, err := os.Create(bulletin2)
|
|
require.NoError(t, err)
|
|
f2.Close()
|
|
|
|
f3, err := os.Create(file1)
|
|
require.NoError(t, err)
|
|
f3.Close()
|
|
|
|
r, err := sut.MSRCBulletins()
|
|
require.NoError(t, err)
|
|
|
|
a, err := NewMSRCMetadata(filepath.Base(file1))
|
|
require.NoError(t, err)
|
|
b, err := NewMSRCMetadata(filepath.Base(bulletin1))
|
|
require.NoError(t, err)
|
|
c, err := NewMSRCMetadata(filepath.Base(bulletin2))
|
|
require.NoError(t, err)
|
|
|
|
require.NotContains(t, r, a)
|
|
require.Contains(t, r, b)
|
|
require.Contains(t, r, c)
|
|
})
|
|
})
|
|
}
|