feat: initial version

cmd/service/service_test.go

@@ -0,0 +1,69 @@
+package main
+
+import (
+	"os"
+	"syscall"
+	"testing"
+	"time"
+
+	mocks "gitlab.com/unboundsoftware/apex-mocks"
+)
+
+func TestMainFunc_Success(t *testing.T) {
+	os.Args = []string{"s3uploader", "--port", "7777", "--bucket", "test-bucket-somewhere", "--return-url", "https://example.org"}
+	go func() {
+		time.Sleep(time.Second)
+		_ = syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
+	}()
+	main()
+}
+
+func TestMainFunc_Invalid_AWS_Config(t *testing.T) {
+	os.Args = []string{"s3uploader", "--port", "7777", "--bucket", "test-bucket-somewhere", "--return-url", "https://example.org"}
+	_ = os.Setenv("AWS_STS_REGIONAL_ENDPOINTS", "unknown_value")
+	defer func() {
+		_ = os.Unsetenv("AWS_STS_REGIONAL_ENDPOINTS")
+	}()
+	main()
+}
+
+func Test_start(t *testing.T) {
+	type args struct {
+		port   int
+		bucket string
+		url    string
+	}
+	tests := []struct {
+		name       string
+		args       args
+		wantErr    bool
+		wantLogged []string
+	}{
+		{
+			name: "invalid port",
+			args: args{
+				port:   77777,
+				bucket: "some-bucket",
+				url:    "https://example.org",
+			},
+			wantErr: false,
+			wantLogged: []string{
+				"info: Serving HTTP API on :77777",
+				"error: HTTP server failed",
+				"info: Shutdown of HTTP server complete",
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			logger := mocks.New()
+			CLI.Port = tt.args.port
+			CLI.Bucket = tt.args.bucket
+			CLI.ReturnURL = tt.args.url
+			if err := start(logger.Logger); (err != nil) != tt.wantErr {
+				t.Errorf("start() error = %v, wantErr %v", err, tt.wantErr)
+			}
+			logger.Check(t, tt.wantLogged)
+		})
+	}
+}