Skip to content

Commit 23b7656

Browse files
Dominik KoteckiDominik Kotecki
authored andcommitted
add config tests
1 parent 24c7f1e commit 23b7656

File tree

2 files changed

+33
-16
lines changed

2 files changed

+33
-16
lines changed

redis/config_test.go

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
package redis
2+
3+
import (
4+
"testing"
5+
6+
"github.com/stretchr/testify/assert"
7+
)
8+
9+
// BenchmarkToSlice-4 1302279 897.8 ns/op 688 B/op 12 allocs/op
10+
func TestRedisConfigBuilder(t *testing.T) {
11+
builder := NewRedisContainerConfigurationBuilder()
12+
13+
config := builder.WithImage("redis:6").WithPort(6379).Build()
14+
assert.NotNil(t, config)
15+
assert.Equal(t, "redis:6", config.image)
16+
assert.Equal(t, 6379, config.port)
17+
}
18+
19+
func BenchmarkToSlice(b *testing.B) {
20+
for n := 0; n < b.N; n++ {
21+
builder := NewRedisContainerConfigurationBuilder()
22+
23+
_ = builder.WithImage("redis:6").WithPort(6379).WithPassword("kremowki").WithUsername("janpawel2").Build()
24+
}
25+
}

redis/redis.go

Lines changed: 8 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ import (
1111

1212
type redisContainer struct {
1313
testcontainers.Container
14+
config *RedisContainerConfiguration
1415
}
1516

1617
func StartContainer(ctx context.Context, config *RedisContainerConfiguration) (*redisContainer, error) {
@@ -26,30 +27,21 @@ func StartContainer(ctx context.Context, config *RedisContainerConfiguration) (*
2627
if err != nil {
2728
return nil, err
2829
}
30+
return &redisContainer{container, config}, nil
31+
}
2932

30-
mappedPort, err := container.MappedPort(ctx)
33+
func (container *redisContainer) Url(ctx context.Context) (string, error) {
34+
mappedPort, err := container.MappedPort(ctx, nat.Port(fmt.Sprint(container.config.port)))
3135
if err != nil {
32-
return nil, err
36+
return "", err
3337
}
3438

3539
hostIP, err := container.Host(ctx)
3640
if err != nil {
37-
return nil, err
41+
return "", err
3842
}
3943

4044
uri := fmt.Sprintf("redis://%s:%s", hostIP, mappedPort.Port())
4145

42-
return &redisContainer{Container: container, URI: uri}, nil
43-
}
44-
45-
func (container *redisContainer) Url(ctx context.Context) (string, error) {
46-
host, err := container.Host(ctx)
47-
if err != nil {
48-
return "", err
49-
}
50-
port, err := container.MappedPort(ctx, nat.Port(fmt.Sprint(container.config.port)))
51-
if err != nil {
52-
return "", err
53-
}
54-
return fmt.Sprintf("mongodb://%s:%s", host, port.Port()), nil
46+
return uri, nil
5547
}

0 commit comments

Comments
 (0)