Browse Source

Create s3_test_main.go

feature/modernize-s3-tests
Chris Lu 2 months ago
parent
commit
8328da5d76
  1. 51
      test/s3/basic/s3_test_main.go

51
test/s3/basic/s3_test_main.go

@ -0,0 +1,51 @@
package basic
import (
"fmt"
"os"
"strings"
"testing"
"github.com/seaweedfs/seaweedfs/test/s3/testutil"
)
var testServer *testutil.TestServer
// TestMain sets up and tears down the test environment using weed mini
func TestMain(m *testing.M) {
var exitCode int
fmt.Println("TestMain: Starting test setup...")
// Check if we should manage the server or use an existing one
if os.Getenv("USE_EXTERNAL_SERVER") != "true" {
// Start server using weed mini
fmt.Println("TestMain: Starting weed mini server...")
config := testutil.DefaultServerConfig(nil)
config.AccessKey = "some_access_key1"
config.SecretKey = "some_secret_key1"
fmt.Println("TestMain: Setting environment variables...")
// Set AWS credentials for tests
os.Setenv("AWS_ACCESS_KEY_ID", config.AccessKey)
os.Setenv("AWS_SECRET_ACCESS_KEY", config.SecretKey)
os.Setenv("AWS_REGION", "us-east-1")
fmt.Printf("TestMain: Set AWS_ACCESS_KEY_ID=%s, AWS_SECRET_ACCESS_KEY=%s, AWS_REGION=us-east-1\n",
config.AccessKey, strings.Repeat("*", len(config.SecretKey)))
var err error
testServer, err = testutil.StartServer(config)
if err != nil {
fmt.Printf("TestMain: ERROR - Failed to start test server: %v\n", err)
os.Exit(1)
}
fmt.Println("TestMain: Server started successfully")
defer testServer.Stop()
}
// Run tests
fmt.Println("TestMain: Running tests...")
exitCode = m.Run()
fmt.Printf("TestMain: Tests completed with exit code %d\n", exitCode)
os.Exit(exitCode)
}
Loading…
Cancel
Save