refactor(internal): export internal packages (#887)

* refactor: export internal packages * refactor(server): define Server * refactor: fix lint issues * test(integration): fix imports
2025-12-21 14:50:53 -08:00 · 2021-03-14 17:04:01 +02:00
parent 8b3b5d0290
commit c26a3e481f
40 changed files with 104 additions and 89 deletions
--- a/pkg/commands/app_test.go
+++ b/pkg/commands/app_test.go
@@ -0,0 +1,120 @@
+package commands
+
+import (
+	"bytes"
+	"encoding/json"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+
+	"github.com/spf13/afero"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/aquasecurity/trivy-db/pkg/db"
+)
+
+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()
+}