normaliseeol
This commit is contained in:
Michael McGuinness
@@ -1,132 +1,132 @@
|
||||
package grpc_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"testing"
|
||||
|
||||
"github.com/docker/go-connections/nat"
|
||||
"github.com/testcontainers/testcontainers-go"
|
||||
"github.com/testcontainers/testcontainers-go/network"
|
||||
"github.com/testcontainers/testcontainers-go/wait"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/credentials/insecure"
|
||||
)
|
||||
|
||||
type containerConfig struct {
|
||||
DB dbConfig
|
||||
Network *testcontainers.DockerNetwork
|
||||
}
|
||||
|
||||
func createContainer(t *testing.T, ctx context.Context, config *containerConfig) (*grpc.ClientConn, testcontainers.Container) {
|
||||
port, err := nat.NewPort("tcp", "8080")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create port: %v", err)
|
||||
}
|
||||
|
||||
req := testcontainers.ContainerRequest{
|
||||
//TODO
|
||||
//FromDockerfile: testcontainers.FromDockerfile{
|
||||
// Context: "../../..",
|
||||
// Dockerfile: "build/package/grpc/Dockerfile",
|
||||
// KeepImage: true,
|
||||
//},
|
||||
Image: "gotemplate_grpc:latest",
|
||||
Env: map[string]string{
|
||||
"DB_USER": config.DB.User,
|
||||
"DB_PASS": config.DB.Password,
|
||||
"DB_HOST": config.DB.Host,
|
||||
"DB_NAME": config.DB.Name,
|
||||
"DB_PORT": strconv.Itoa(config.DB.Port),
|
||||
},
|
||||
ExposedPorts: []string{port.Port()},
|
||||
WaitingFor: wait.ForListeningPort(port),
|
||||
Networks: []string{config.Network.Name},
|
||||
}
|
||||
|
||||
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
||||
ContainerRequest: req,
|
||||
Started: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start container: %v", err)
|
||||
}
|
||||
|
||||
host, err := container.Host(ctx)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to extract host: %v", err)
|
||||
}
|
||||
mappedPort, err := container.MappedPort(ctx, port)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to extract port: %v", err)
|
||||
}
|
||||
|
||||
conn, err := grpc.NewClient(
|
||||
fmt.Sprintf("%s:%s", host, mappedPort.Port()),
|
||||
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
||||
)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to connect to gRPC server: %v", err)
|
||||
}
|
||||
|
||||
return conn, container
|
||||
}
|
||||
|
||||
type dbConfig struct {
|
||||
Name string
|
||||
Host string
|
||||
Port int
|
||||
User string
|
||||
Password string
|
||||
}
|
||||
|
||||
func createDB(t *testing.T, ctx context.Context, network *testcontainers.DockerNetwork) (*dbConfig, testcontainers.Container) {
|
||||
alias := "postgres"
|
||||
port, err := nat.NewPort("tcp", "5432")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create port: %v", err)
|
||||
}
|
||||
|
||||
config := dbConfig{
|
||||
Name: "gotemplate",
|
||||
Password: "pass",
|
||||
User: "postgres",
|
||||
Port: port.Int(),
|
||||
Host: alias,
|
||||
}
|
||||
|
||||
req := testcontainers.ContainerRequest{
|
||||
Image: "postgres:latest",
|
||||
Env: map[string]string{
|
||||
"POSTGRES_DB": config.Name,
|
||||
"POSTGRES_USER": config.User,
|
||||
"POSTGRES_PASSWORD": config.Password,
|
||||
},
|
||||
Networks: []string{network.Name},
|
||||
NetworkAliases: map[string][]string{
|
||||
network.Name: {alias},
|
||||
},
|
||||
ExposedPorts: []string{port.Port()},
|
||||
WaitingFor: wait.ForListeningPort(port),
|
||||
}
|
||||
|
||||
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
||||
ContainerRequest: req,
|
||||
Started: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start container: %v", err)
|
||||
}
|
||||
|
||||
return &config, container
|
||||
}
|
||||
|
||||
func createNetwork(t *testing.T, ctx context.Context) *testcontainers.DockerNetwork {
|
||||
network, err := network.New(ctx, network.WithDriver("bridge"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
return network
|
||||
}
|
||||
package grpc_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"testing"
|
||||
|
||||
"github.com/docker/go-connections/nat"
|
||||
"github.com/testcontainers/testcontainers-go"
|
||||
"github.com/testcontainers/testcontainers-go/network"
|
||||
"github.com/testcontainers/testcontainers-go/wait"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/credentials/insecure"
|
||||
)
|
||||
|
||||
type containerConfig struct {
|
||||
DB dbConfig
|
||||
Network *testcontainers.DockerNetwork
|
||||
}
|
||||
|
||||
func createContainer(t *testing.T, ctx context.Context, config *containerConfig) (*grpc.ClientConn, testcontainers.Container) {
|
||||
port, err := nat.NewPort("tcp", "8080")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create port: %v", err)
|
||||
}
|
||||
|
||||
req := testcontainers.ContainerRequest{
|
||||
//TODO
|
||||
//FromDockerfile: testcontainers.FromDockerfile{
|
||||
// Context: "../../..",
|
||||
// Dockerfile: "build/package/grpc/Dockerfile",
|
||||
// KeepImage: true,
|
||||
//},
|
||||
Image: "gotemplate_grpc:latest",
|
||||
Env: map[string]string{
|
||||
"DB_USER": config.DB.User,
|
||||
"DB_PASS": config.DB.Password,
|
||||
"DB_HOST": config.DB.Host,
|
||||
"DB_NAME": config.DB.Name,
|
||||
"DB_PORT": strconv.Itoa(config.DB.Port),
|
||||
},
|
||||
ExposedPorts: []string{port.Port()},
|
||||
WaitingFor: wait.ForListeningPort(port),
|
||||
Networks: []string{config.Network.Name},
|
||||
}
|
||||
|
||||
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
||||
ContainerRequest: req,
|
||||
Started: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start container: %v", err)
|
||||
}
|
||||
|
||||
host, err := container.Host(ctx)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to extract host: %v", err)
|
||||
}
|
||||
mappedPort, err := container.MappedPort(ctx, port)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to extract port: %v", err)
|
||||
}
|
||||
|
||||
conn, err := grpc.NewClient(
|
||||
fmt.Sprintf("%s:%s", host, mappedPort.Port()),
|
||||
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
||||
)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to connect to gRPC server: %v", err)
|
||||
}
|
||||
|
||||
return conn, container
|
||||
}
|
||||
|
||||
type dbConfig struct {
|
||||
Name string
|
||||
Host string
|
||||
Port int
|
||||
User string
|
||||
Password string
|
||||
}
|
||||
|
||||
func createDB(t *testing.T, ctx context.Context, network *testcontainers.DockerNetwork) (*dbConfig, testcontainers.Container) {
|
||||
alias := "postgres"
|
||||
port, err := nat.NewPort("tcp", "5432")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create port: %v", err)
|
||||
}
|
||||
|
||||
config := dbConfig{
|
||||
Name: "gotemplate",
|
||||
Password: "pass",
|
||||
User: "postgres",
|
||||
Port: port.Int(),
|
||||
Host: alias,
|
||||
}
|
||||
|
||||
req := testcontainers.ContainerRequest{
|
||||
Image: "postgres:latest",
|
||||
Env: map[string]string{
|
||||
"POSTGRES_DB": config.Name,
|
||||
"POSTGRES_USER": config.User,
|
||||
"POSTGRES_PASSWORD": config.Password,
|
||||
},
|
||||
Networks: []string{network.Name},
|
||||
NetworkAliases: map[string][]string{
|
||||
network.Name: {alias},
|
||||
},
|
||||
ExposedPorts: []string{port.Port()},
|
||||
WaitingFor: wait.ForListeningPort(port),
|
||||
}
|
||||
|
||||
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
||||
ContainerRequest: req,
|
||||
Started: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start container: %v", err)
|
||||
}
|
||||
|
||||
return &config, container
|
||||
}
|
||||
|
||||
func createNetwork(t *testing.T, ctx context.Context) *testcontainers.DockerNetwork {
|
||||
network, err := network.New(ctx, network.WithDriver("bridge"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
return network
|
||||
}
|
||||
|
||||
@@ -1,52 +1,52 @@
|
||||
package grpc_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
"gotemplate/api/grpc/spec"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/testcontainers/testcontainers-go"
|
||||
)
|
||||
|
||||
func TestName(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
network := createNetwork(t, ctx)
|
||||
defer testcontainers.CleanupNetwork(t, network)
|
||||
|
||||
dbconfig, dbContainer := createDB(t, ctx, network)
|
||||
defer dbContainer.Terminate(ctx)
|
||||
|
||||
conn, container := createContainer(t, ctx, &containerConfig{
|
||||
DB: *dbconfig,
|
||||
Network: network,
|
||||
})
|
||||
defer container.Terminate(ctx)
|
||||
defer conn.Close()
|
||||
|
||||
client := spec.NewNameClient(conn)
|
||||
|
||||
name := "Doczy.ai"
|
||||
|
||||
_, err := client.Get(ctx, &spec.NameMessage{
|
||||
Name: name,
|
||||
})
|
||||
|
||||
assert.NotNil(t, err)
|
||||
|
||||
response, err := client.Add(ctx, &spec.NameMessage{
|
||||
Name: name,
|
||||
})
|
||||
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, name, response.Name)
|
||||
|
||||
response, err = client.Get(ctx, &spec.NameMessage{
|
||||
Name: name,
|
||||
})
|
||||
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, name, response.Name)
|
||||
}
|
||||
package grpc_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
"gotemplate/api/grpc/spec"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/testcontainers/testcontainers-go"
|
||||
)
|
||||
|
||||
func TestName(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
network := createNetwork(t, ctx)
|
||||
defer testcontainers.CleanupNetwork(t, network)
|
||||
|
||||
dbconfig, dbContainer := createDB(t, ctx, network)
|
||||
defer dbContainer.Terminate(ctx)
|
||||
|
||||
conn, container := createContainer(t, ctx, &containerConfig{
|
||||
DB: *dbconfig,
|
||||
Network: network,
|
||||
})
|
||||
defer container.Terminate(ctx)
|
||||
defer conn.Close()
|
||||
|
||||
client := spec.NewNameClient(conn)
|
||||
|
||||
name := "Doczy.ai"
|
||||
|
||||
_, err := client.Get(ctx, &spec.NameMessage{
|
||||
Name: name,
|
||||
})
|
||||
|
||||
assert.NotNil(t, err)
|
||||
|
||||
response, err := client.Add(ctx, &spec.NameMessage{
|
||||
Name: name,
|
||||
})
|
||||
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, name, response.Name)
|
||||
|
||||
response, err = client.Get(ctx, &spec.NameMessage{
|
||||
Name: name,
|
||||
})
|
||||
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, name, response.Name)
|
||||
}
|
||||
|
||||
@@ -1,238 +1,238 @@
|
||||
package queue_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"testing"
|
||||
|
||||
"github.com/aws/aws-sdk-go-v2/aws"
|
||||
"github.com/aws/aws-sdk-go-v2/config"
|
||||
"github.com/aws/aws-sdk-go-v2/credentials"
|
||||
"github.com/aws/aws-sdk-go-v2/service/sqs"
|
||||
"github.com/aws/aws-sdk-go-v2/service/sqs/types"
|
||||
"github.com/docker/go-connections/nat"
|
||||
"github.com/testcontainers/testcontainers-go"
|
||||
"github.com/testcontainers/testcontainers-go/network"
|
||||
"github.com/testcontainers/testcontainers-go/wait"
|
||||
)
|
||||
|
||||
type containerConfig struct {
|
||||
Queue queueConfig
|
||||
DB *dbConfig
|
||||
Network *testcontainers.DockerNetwork
|
||||
}
|
||||
|
||||
func createContainer(t *testing.T, ctx context.Context, config containerConfig) testcontainers.Container {
|
||||
queueCredentials, err := config.Queue.Credentials.Retrieve(ctx)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
req := testcontainers.ContainerRequest{
|
||||
//TODO
|
||||
//FromDockerfile: testcontainers.FromDockerfile{
|
||||
// Context: "../../..",
|
||||
// Dockerfile: "build/package/queue/Dockerfile",
|
||||
// KeepImage: true,
|
||||
//},
|
||||
Image: "gotemplate_queue:latest",
|
||||
Env: map[string]string{
|
||||
"QUEUE_URL": config.Queue.URL,
|
||||
"AWS_DEFAULT_REGION": config.Queue.Region,
|
||||
"AWS_REGION": config.Queue.Region,
|
||||
"AWS_ACCESS_KEY_ID": queueCredentials.AccessKeyID,
|
||||
"AWS_SECRET_ACCESS_KEY": queueCredentials.SecretAccessKey,
|
||||
"AWS_SESSION_TOKEN": queueCredentials.SessionToken,
|
||||
"AWS_ENDPOINT_URL_SQS": config.Queue.Endpoint,
|
||||
"DB_USER": config.DB.User,
|
||||
"DB_PASS": config.DB.Password,
|
||||
"DB_HOST": config.DB.Host,
|
||||
"DB_NAME": config.DB.Name,
|
||||
"DB_PORT": strconv.Itoa(config.DB.Port),
|
||||
},
|
||||
Networks: []string{config.Network.Name},
|
||||
}
|
||||
|
||||
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
||||
ContainerRequest: req,
|
||||
Started: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start container: %v", err)
|
||||
}
|
||||
|
||||
return container
|
||||
}
|
||||
|
||||
type queueConfig struct {
|
||||
URL string
|
||||
Region string
|
||||
Endpoint string
|
||||
Credentials aws.CredentialsProvider
|
||||
}
|
||||
|
||||
type queue struct {
|
||||
Container *testcontainers.Container
|
||||
Client *sqs.Client
|
||||
Config *queueConfig
|
||||
}
|
||||
|
||||
func createQueue(t *testing.T, ctx context.Context, network *testcontainers.DockerNetwork) *queue {
|
||||
queueName := "test-queue"
|
||||
region := "us-east-1"
|
||||
alias := "localstack"
|
||||
provider := credentials.NewStaticCredentialsProvider("test", "test", "")
|
||||
|
||||
port, err := nat.NewPort("tcp", "4566")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create port: %v", err)
|
||||
}
|
||||
|
||||
req := testcontainers.ContainerRequest{
|
||||
Image: "localstack/localstack:latest",
|
||||
Env: map[string]string{
|
||||
"AWS_ACCESS_KEY_ID": provider.Value.AccessKeyID,
|
||||
"AWS_SECRET_ACCESS_KEY": provider.Value.SecretAccessKey,
|
||||
"AWS_SESSION_TOKEN": provider.Value.SessionToken,
|
||||
"AWS_DEFAULT_REGION": region,
|
||||
"AWS_REGION": region,
|
||||
"SERVICES": "sqs",
|
||||
"SKIP_SSL_CERT_DOWNLOAD": "1",
|
||||
"LOCALSTACK_HOST": alias,
|
||||
"SQS_ENDPOINT_STRATEGY": "path",
|
||||
},
|
||||
Networks: []string{network.Name},
|
||||
NetworkAliases: map[string][]string{
|
||||
network.Name: {alias},
|
||||
},
|
||||
ExposedPorts: []string{port.Port()},
|
||||
WaitingFor: wait.ForListeningPort(port),
|
||||
}
|
||||
|
||||
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
||||
ContainerRequest: req,
|
||||
Started: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start container: %v", err)
|
||||
}
|
||||
|
||||
host, err := container.Host(ctx)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to extract host: %v", err)
|
||||
}
|
||||
mappedPort, err := container.MappedPort(ctx, port)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to extract port: %v", err)
|
||||
}
|
||||
|
||||
endpoint := fmt.Sprintf("http://%s:%s", host, mappedPort.Port())
|
||||
cfg, err := config.LoadDefaultConfig(ctx, config.WithRegion(region), config.WithCredentialsProvider(provider), config.WithBaseEndpoint(endpoint))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
client := sqs.NewFromConfig(cfg)
|
||||
|
||||
queueM, err := client.CreateQueue(ctx, &sqs.CreateQueueInput{
|
||||
QueueName: aws.String(queueName),
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
endpoint = fmt.Sprintf("http://%s:%s", alias, port.Port())
|
||||
queueC := &queueConfig{
|
||||
URL: *queueM.QueueUrl,
|
||||
Region: region,
|
||||
Credentials: provider,
|
||||
Endpoint: endpoint,
|
||||
}
|
||||
|
||||
return &queue{
|
||||
Container: &container,
|
||||
Client: client,
|
||||
Config: queueC,
|
||||
}
|
||||
}
|
||||
|
||||
func createNetwork(t *testing.T, ctx context.Context) *testcontainers.DockerNetwork {
|
||||
network, err := network.New(ctx, network.WithDriver("bridge"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
return network
|
||||
}
|
||||
|
||||
type dbConfig struct {
|
||||
Name string
|
||||
Host string
|
||||
Port int
|
||||
User string
|
||||
Password string
|
||||
}
|
||||
|
||||
func createDB(t *testing.T, ctx context.Context, network *testcontainers.DockerNetwork) (*dbConfig, testcontainers.Container) {
|
||||
alias := "postgres"
|
||||
port, err := nat.NewPort("tcp", "5432")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create port: %v", err)
|
||||
}
|
||||
|
||||
config := dbConfig{
|
||||
Name: "gotemplate",
|
||||
Password: "pass",
|
||||
User: "postgres",
|
||||
Port: port.Int(),
|
||||
Host: alias,
|
||||
}
|
||||
|
||||
req := testcontainers.ContainerRequest{
|
||||
Image: "postgres:latest",
|
||||
Env: map[string]string{
|
||||
"POSTGRES_DB": config.Name,
|
||||
"POSTGRES_USER": config.User,
|
||||
"POSTGRES_PASSWORD": config.Password,
|
||||
},
|
||||
Networks: []string{network.Name},
|
||||
NetworkAliases: map[string][]string{
|
||||
network.Name: {alias},
|
||||
},
|
||||
ExposedPorts: []string{port.Port()},
|
||||
WaitingFor: wait.ForListeningPort(port),
|
||||
}
|
||||
|
||||
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
||||
ContainerRequest: req,
|
||||
Started: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start container: %v", err)
|
||||
}
|
||||
|
||||
return &config, container
|
||||
}
|
||||
|
||||
type message struct {
|
||||
URL string
|
||||
Type string
|
||||
Body string
|
||||
}
|
||||
|
||||
func sendMessage(t *testing.T, ctx context.Context, queue *sqs.Client, message message) {
|
||||
_, err := queue.SendMessage(ctx, &sqs.SendMessageInput{
|
||||
MessageAttributes: map[string]types.MessageAttributeValue{
|
||||
"type": {
|
||||
DataType: aws.String("String"),
|
||||
StringValue: aws.String(message.Type),
|
||||
},
|
||||
},
|
||||
QueueUrl: aws.String(message.URL),
|
||||
MessageBody: aws.String(message.Body),
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to send message: %v", err)
|
||||
}
|
||||
}
|
||||
package queue_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"testing"
|
||||
|
||||
"github.com/aws/aws-sdk-go-v2/aws"
|
||||
"github.com/aws/aws-sdk-go-v2/config"
|
||||
"github.com/aws/aws-sdk-go-v2/credentials"
|
||||
"github.com/aws/aws-sdk-go-v2/service/sqs"
|
||||
"github.com/aws/aws-sdk-go-v2/service/sqs/types"
|
||||
"github.com/docker/go-connections/nat"
|
||||
"github.com/testcontainers/testcontainers-go"
|
||||
"github.com/testcontainers/testcontainers-go/network"
|
||||
"github.com/testcontainers/testcontainers-go/wait"
|
||||
)
|
||||
|
||||
type containerConfig struct {
|
||||
Queue queueConfig
|
||||
DB *dbConfig
|
||||
Network *testcontainers.DockerNetwork
|
||||
}
|
||||
|
||||
func createContainer(t *testing.T, ctx context.Context, config containerConfig) testcontainers.Container {
|
||||
queueCredentials, err := config.Queue.Credentials.Retrieve(ctx)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
req := testcontainers.ContainerRequest{
|
||||
//TODO
|
||||
//FromDockerfile: testcontainers.FromDockerfile{
|
||||
// Context: "../../..",
|
||||
// Dockerfile: "build/package/queue/Dockerfile",
|
||||
// KeepImage: true,
|
||||
//},
|
||||
Image: "gotemplate_queue:latest",
|
||||
Env: map[string]string{
|
||||
"QUEUE_URL": config.Queue.URL,
|
||||
"AWS_DEFAULT_REGION": config.Queue.Region,
|
||||
"AWS_REGION": config.Queue.Region,
|
||||
"AWS_ACCESS_KEY_ID": queueCredentials.AccessKeyID,
|
||||
"AWS_SECRET_ACCESS_KEY": queueCredentials.SecretAccessKey,
|
||||
"AWS_SESSION_TOKEN": queueCredentials.SessionToken,
|
||||
"AWS_ENDPOINT_URL_SQS": config.Queue.Endpoint,
|
||||
"DB_USER": config.DB.User,
|
||||
"DB_PASS": config.DB.Password,
|
||||
"DB_HOST": config.DB.Host,
|
||||
"DB_NAME": config.DB.Name,
|
||||
"DB_PORT": strconv.Itoa(config.DB.Port),
|
||||
},
|
||||
Networks: []string{config.Network.Name},
|
||||
}
|
||||
|
||||
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
||||
ContainerRequest: req,
|
||||
Started: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start container: %v", err)
|
||||
}
|
||||
|
||||
return container
|
||||
}
|
||||
|
||||
type queueConfig struct {
|
||||
URL string
|
||||
Region string
|
||||
Endpoint string
|
||||
Credentials aws.CredentialsProvider
|
||||
}
|
||||
|
||||
type queue struct {
|
||||
Container *testcontainers.Container
|
||||
Client *sqs.Client
|
||||
Config *queueConfig
|
||||
}
|
||||
|
||||
func createQueue(t *testing.T, ctx context.Context, network *testcontainers.DockerNetwork) *queue {
|
||||
queueName := "test-queue"
|
||||
region := "us-east-1"
|
||||
alias := "localstack"
|
||||
provider := credentials.NewStaticCredentialsProvider("test", "test", "")
|
||||
|
||||
port, err := nat.NewPort("tcp", "4566")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create port: %v", err)
|
||||
}
|
||||
|
||||
req := testcontainers.ContainerRequest{
|
||||
Image: "localstack/localstack:latest",
|
||||
Env: map[string]string{
|
||||
"AWS_ACCESS_KEY_ID": provider.Value.AccessKeyID,
|
||||
"AWS_SECRET_ACCESS_KEY": provider.Value.SecretAccessKey,
|
||||
"AWS_SESSION_TOKEN": provider.Value.SessionToken,
|
||||
"AWS_DEFAULT_REGION": region,
|
||||
"AWS_REGION": region,
|
||||
"SERVICES": "sqs",
|
||||
"SKIP_SSL_CERT_DOWNLOAD": "1",
|
||||
"LOCALSTACK_HOST": alias,
|
||||
"SQS_ENDPOINT_STRATEGY": "path",
|
||||
},
|
||||
Networks: []string{network.Name},
|
||||
NetworkAliases: map[string][]string{
|
||||
network.Name: {alias},
|
||||
},
|
||||
ExposedPorts: []string{port.Port()},
|
||||
WaitingFor: wait.ForListeningPort(port),
|
||||
}
|
||||
|
||||
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
||||
ContainerRequest: req,
|
||||
Started: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start container: %v", err)
|
||||
}
|
||||
|
||||
host, err := container.Host(ctx)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to extract host: %v", err)
|
||||
}
|
||||
mappedPort, err := container.MappedPort(ctx, port)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to extract port: %v", err)
|
||||
}
|
||||
|
||||
endpoint := fmt.Sprintf("http://%s:%s", host, mappedPort.Port())
|
||||
cfg, err := config.LoadDefaultConfig(ctx, config.WithRegion(region), config.WithCredentialsProvider(provider), config.WithBaseEndpoint(endpoint))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
client := sqs.NewFromConfig(cfg)
|
||||
|
||||
queueM, err := client.CreateQueue(ctx, &sqs.CreateQueueInput{
|
||||
QueueName: aws.String(queueName),
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
endpoint = fmt.Sprintf("http://%s:%s", alias, port.Port())
|
||||
queueC := &queueConfig{
|
||||
URL: *queueM.QueueUrl,
|
||||
Region: region,
|
||||
Credentials: provider,
|
||||
Endpoint: endpoint,
|
||||
}
|
||||
|
||||
return &queue{
|
||||
Container: &container,
|
||||
Client: client,
|
||||
Config: queueC,
|
||||
}
|
||||
}
|
||||
|
||||
func createNetwork(t *testing.T, ctx context.Context) *testcontainers.DockerNetwork {
|
||||
network, err := network.New(ctx, network.WithDriver("bridge"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
return network
|
||||
}
|
||||
|
||||
type dbConfig struct {
|
||||
Name string
|
||||
Host string
|
||||
Port int
|
||||
User string
|
||||
Password string
|
||||
}
|
||||
|
||||
func createDB(t *testing.T, ctx context.Context, network *testcontainers.DockerNetwork) (*dbConfig, testcontainers.Container) {
|
||||
alias := "postgres"
|
||||
port, err := nat.NewPort("tcp", "5432")
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create port: %v", err)
|
||||
}
|
||||
|
||||
config := dbConfig{
|
||||
Name: "gotemplate",
|
||||
Password: "pass",
|
||||
User: "postgres",
|
||||
Port: port.Int(),
|
||||
Host: alias,
|
||||
}
|
||||
|
||||
req := testcontainers.ContainerRequest{
|
||||
Image: "postgres:latest",
|
||||
Env: map[string]string{
|
||||
"POSTGRES_DB": config.Name,
|
||||
"POSTGRES_USER": config.User,
|
||||
"POSTGRES_PASSWORD": config.Password,
|
||||
},
|
||||
Networks: []string{network.Name},
|
||||
NetworkAliases: map[string][]string{
|
||||
network.Name: {alias},
|
||||
},
|
||||
ExposedPorts: []string{port.Port()},
|
||||
WaitingFor: wait.ForListeningPort(port),
|
||||
}
|
||||
|
||||
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
||||
ContainerRequest: req,
|
||||
Started: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start container: %v", err)
|
||||
}
|
||||
|
||||
return &config, container
|
||||
}
|
||||
|
||||
type message struct {
|
||||
URL string
|
||||
Type string
|
||||
Body string
|
||||
}
|
||||
|
||||
func sendMessage(t *testing.T, ctx context.Context, queue *sqs.Client, message message) {
|
||||
_, err := queue.SendMessage(ctx, &sqs.SendMessageInput{
|
||||
MessageAttributes: map[string]types.MessageAttributeValue{
|
||||
"type": {
|
||||
DataType: aws.String("String"),
|
||||
StringValue: aws.String(message.Type),
|
||||
},
|
||||
},
|
||||
QueueUrl: aws.String(message.URL),
|
||||
MessageBody: aws.String(message.Body),
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to send message: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,54 +1,54 @@
|
||||
package queue_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/aws/aws-sdk-go-v2/service/sqs"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/testcontainers/testcontainers-go"
|
||||
)
|
||||
|
||||
func TestName(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
network := createNetwork(t, ctx)
|
||||
defer testcontainers.CleanupNetwork(t, network)
|
||||
|
||||
dbconfig, dbContainer := createDB(t, ctx, network)
|
||||
defer dbContainer.Terminate(ctx)
|
||||
|
||||
queue := createQueue(t, ctx, network)
|
||||
defer (*queue.Container).Terminate(ctx)
|
||||
|
||||
container := createContainer(t, ctx, containerConfig{
|
||||
Queue: *queue.Config,
|
||||
DB: dbconfig,
|
||||
Network: network,
|
||||
})
|
||||
defer container.Terminate(ctx)
|
||||
|
||||
sendMessage(t, ctx, queue.Client, message{
|
||||
URL: queue.Config.URL,
|
||||
Type: "CREATE_NAME",
|
||||
Body: "{\"name\":\"Doczy.ai\"}",
|
||||
})
|
||||
|
||||
time.Sleep(time.Second)
|
||||
|
||||
result, err := queue.Client.ReceiveMessage(ctx, &sqs.ReceiveMessageInput{
|
||||
QueueUrl: &queue.Config.URL,
|
||||
MaxNumberOfMessages: 1,
|
||||
WaitTimeSeconds: 5,
|
||||
VisibilityTimeout: 2,
|
||||
MessageAttributeNames: []string{
|
||||
"type",
|
||||
},
|
||||
})
|
||||
|
||||
assert.Nil(t, err)
|
||||
for _, message := range result.Messages {
|
||||
assert.Equal(t, "CREATE_NAME_SUCCESS", *message.MessageAttributes["type"].StringValue)
|
||||
}
|
||||
}
|
||||
package queue_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/aws/aws-sdk-go-v2/service/sqs"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/testcontainers/testcontainers-go"
|
||||
)
|
||||
|
||||
func TestName(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
network := createNetwork(t, ctx)
|
||||
defer testcontainers.CleanupNetwork(t, network)
|
||||
|
||||
dbconfig, dbContainer := createDB(t, ctx, network)
|
||||
defer dbContainer.Terminate(ctx)
|
||||
|
||||
queue := createQueue(t, ctx, network)
|
||||
defer (*queue.Container).Terminate(ctx)
|
||||
|
||||
container := createContainer(t, ctx, containerConfig{
|
||||
Queue: *queue.Config,
|
||||
DB: dbconfig,
|
||||
Network: network,
|
||||
})
|
||||
defer container.Terminate(ctx)
|
||||
|
||||
sendMessage(t, ctx, queue.Client, message{
|
||||
URL: queue.Config.URL,
|
||||
Type: "CREATE_NAME",
|
||||
Body: "{\"name\":\"Doczy.ai\"}",
|
||||
})
|
||||
|
||||
time.Sleep(time.Second)
|
||||
|
||||
result, err := queue.Client.ReceiveMessage(ctx, &sqs.ReceiveMessageInput{
|
||||
QueueUrl: &queue.Config.URL,
|
||||
MaxNumberOfMessages: 1,
|
||||
WaitTimeSeconds: 5,
|
||||
VisibilityTimeout: 2,
|
||||
MessageAttributeNames: []string{
|
||||
"type",
|
||||
},
|
||||
})
|
||||
|
||||
assert.Nil(t, err)
|
||||
for _, message := range result.Messages {
|
||||
assert.Equal(t, "CREATE_NAME_SUCCESS", *message.MessageAttributes["type"].StringValue)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user