mirror of
https://github.com/aquasecurity/trivy.git
synced 2025-12-19 02:32:41 -08:00
ca6f196001
* Skip downloading DB if a remote DB is not updated * Apply suggestions from code review Co-authored-by: Teppei Fukuda <knqyf263@gmail.com> * update github.com/aquasecurity/trivy-db version * fix lint * Use UTC datetime * display DownloadedAt info in debug log * refactor(db): merge isLatestDB into isNewDB Co-authored-by: Teppei Fukuda <knqyf263@gmail.com>
123 lines
2.7 KiB
Go
123 lines
2.7 KiB
Go
package internal
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/spf13/afero"
|
|
|
|
"github.com/aquasecurity/trivy-db/pkg/db"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_showVersion(t *testing.T) {
|
|
type args struct {
|
|
cacheDir string
|
|
outputFormat string
|
|
version string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
createDB bool
|
|
expectedOutput string
|
|
}{
|
|
{
|
|
name: "happy path, table output",
|
|
args: args{
|
|
outputFormat: "table",
|
|
version: "v1.2.3",
|
|
},
|
|
expectedOutput: `Version: v1.2.3
|
|
Vulnerability DB:
|
|
Type: Light
|
|
Version: 42
|
|
UpdatedAt: 2020-03-16 23:40:20 +0000 UTC
|
|
NextUpdate: 2020-03-16 23:57:00 +0000 UTC
|
|
DownloadedAt: 2020-03-16 23:40:20 +0000 UTC
|
|
`,
|
|
createDB: true,
|
|
},
|
|
{
|
|
name: "happy path, JSON output",
|
|
args: args{
|
|
outputFormat: "json",
|
|
version: "1.2.3",
|
|
},
|
|
expectedOutput: `{"Version":"1.2.3","VulnerabilityDB":{"Version":42,"Type":1,"NextUpdate":"2020-03-16T23:57:00Z","UpdatedAt":"2020-03-16T23:40:20Z","DownloadedAt":"2020-03-16T23:40:20Z"}}
|
|
`,
|
|
createDB: true,
|
|
},
|
|
{
|
|
name: "sad path, no DB is available",
|
|
args: args{
|
|
outputFormat: "json",
|
|
version: "1.2.3",
|
|
},
|
|
expectedOutput: `{"Version":"1.2.3"}
|
|
`,
|
|
},
|
|
{
|
|
name: "sad path, bogus cache dir",
|
|
args: args{
|
|
outputFormat: "json",
|
|
version: "1.2.3",
|
|
cacheDir: "/foo/bar/bogus",
|
|
},
|
|
expectedOutput: `{"Version":"1.2.3"}
|
|
`,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
var cacheDir string
|
|
switch {
|
|
case tt.args.cacheDir != "":
|
|
cacheDir = tt.args.cacheDir
|
|
default:
|
|
cacheDir, _ = ioutil.TempDir("", "Test_showVersion-*")
|
|
defer os.RemoveAll(cacheDir)
|
|
}
|
|
|
|
if tt.createDB {
|
|
fs := afero.NewOsFs()
|
|
err := os.MkdirAll(filepath.Join(cacheDir, "db"), os.ModePerm)
|
|
require.NoError(t, err)
|
|
metadataFile := filepath.Join(cacheDir, "db", "metadata.json")
|
|
|
|
b, err := json.Marshal(db.Metadata{
|
|
Version: 42,
|
|
Type: 1,
|
|
NextUpdate: time.Unix(1584403020, 0),
|
|
UpdatedAt: time.Unix(1584402020, 0),
|
|
DownloadedAt: time.Unix(1584402020, 0),
|
|
})
|
|
require.NoError(t, err)
|
|
err = afero.WriteFile(fs, metadataFile, b, 0600)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
fw := new(bytes.Buffer)
|
|
showVersion(cacheDir, tt.args.outputFormat, tt.args.version, fw)
|
|
assert.Equal(t, tt.expectedOutput, fw.String(), tt.name)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestNewCommands(t *testing.T) {
|
|
NewApp("test")
|
|
NewClientCommand()
|
|
NewFilesystemCommand()
|
|
NewImageCommand()
|
|
NewRepositoryCommand()
|
|
NewServerCommand()
|
|
}
|