s3_test.go

//go:build s3
// +build s3

package main

import (
	"context"
	"net"
	"net/http"
	"net/http/httptest"
	"os"
	"strings"
	"testing"
	"time"

	"github.com/aerokube/selenoid/event"
	"github.com/aerokube/selenoid/session"
	"github.com/aerokube/selenoid/upload"
	assert "github.com/stretchr/testify/require"
)

var (
	s3Srv *httptest.Server
)

func init() {
	s3Srv = httptest.NewServer(s3Mux())
	dialer := &net.Dialer{
		Timeout:   30 * time.Second,
		KeepAlive: 30 * time.Second,
		DualStack: true,
	}
	http.DefaultTransport.(*http.Transport).DialContext = func(ctx context.Context, network, addr string) (net.Conn, error) {
		if strings.Contains(addr, "s3-mock.example.com") {
			addr = s3Srv.Listener.Addr().String()
		}
		return dialer.DialContext(ctx, network, addr)
	}
}

func s3Mux() http.Handler {
	mux := http.NewServeMux()
	mux.HandleFunc("/", func(_ http.ResponseWriter, _ *http.Request) {})
	return mux
}

var testSession = &session.Session{
	Quota: "some-user",
	Caps: session.Caps{
		Name:     "internet explorer",
		Version:  "11",
		Platform: "WINDOWS",
	},
}

func TestS3Uploader(t *testing.T) {
	uploader := &upload.S3Uploader{
		Endpoint:          "http://s3-mock.example.com",
		Region:            "us-west-1",
		AccessKey:         "some-access-key",
		SecretKey:         "some-secret-key",
		BucketName:        "test-bucket",
		KeyPattern:        "$fileName",
		ReducedRedundancy: true,
	}
	uploader.Init()
	f, _ := os.CreateTemp("", "some-file")
	input := event.CreatedFile{
		Event: event.Event{
			RequestId: 4342,
			SessionId: "some-session-id",
			Session:   testSession,
		},
		Name: f.Name(),
		Type: "log",
	}
	uploaded, err := uploader.Upload(input)
	assert.NoError(t, err)
	assert.True(t, uploaded)
}

func TestGetKey(t *testing.T) {
	const testPattern = "$quota/$sessionId_$browserName_$browserVersion_$platformName/$fileType$fileExtension"
	input := event.CreatedFile{
		Event: event.Event{
			SessionId: "some-Session-id",
			Session:   testSession,
			RequestId: 12345,
		},

		Name: "/path/to/Some-File.txt",
		Type: "log",
	}

	key := upload.GetS3Key(testPattern, input)
	assert.Equal(t, key, "some-user/some-Session-id_internet-explorer_11_windows/log.txt")

	input.Session.Caps.Name = ""
	input.Session.Caps.DeviceName = "internet explorer"
	key = upload.GetS3Key(testPattern, input)
	assert.Equal(t, key, "some-user/some-Session-id_internet-explorer_11_windows/log.txt")

	input.Session.Caps.S3KeyPattern = "$quota/$fileType$fileExtension"
	key = upload.GetS3Key(testPattern, input)
	assert.Equal(t, key, "some-user/log.txt")

	input.Session.Caps.S3KeyPattern = "$fileName"
	key = upload.GetS3Key(testPattern, input)
	assert.Equal(t, key, "Some-File.txt")
}

func TestFileMatches(t *testing.T) {
	matches, err := upload.FileMatches("", "", "any-file-name")
	assert.NoError(t, err)
	assert.True(t, matches)

	matches, err = upload.FileMatches("[", "", "/path/to/file.mp4")
	assert.Error(t, err)
	assert.False(t, matches)

	matches, err = upload.FileMatches("", "[", "/path/to/file.mp4")
	assert.Error(t, err)
	assert.False(t, matches)

	matches, err = upload.FileMatches("*.mp4", "", "/path/to/file.mp4")
	assert.NoError(t, err)
	assert.True(t, matches)

	matches, err = upload.FileMatches("*.mp4", "", "/path/to/file.log")
	assert.NoError(t, err)
	assert.False(t, matches)

	matches, err = upload.FileMatches("*.mp4", "", "/path/to/file.log")
	assert.NoError(t, err)
	assert.False(t, matches)

	matches, err = upload.FileMatches("", "*.log", "/path/to/file.log")
	assert.NoError(t, err)
	assert.False(t, matches)
}