mirror of
https://github.com/sablierapp/sablier.git
synced 2025-12-21 21:33:06 +01:00
0d699effc3
* add tmp test * add strategy config * strategy config * pause / unpause strategy impl * Fix tests * fix compilation * add pause / unpause tests * add doc * add config test * start if not paused * remove test files
117 lines
2.6 KiB
Go
117 lines
2.6 KiB
Go
package docker_test
|
|
|
|
import (
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/neilotoole/slogt"
|
|
"github.com/sablierapp/sablier/pkg/provider"
|
|
"github.com/sablierapp/sablier/pkg/provider/docker"
|
|
"github.com/sablierapp/sablier/pkg/sablier"
|
|
"gotest.tools/v3/assert"
|
|
)
|
|
|
|
func TestDockerClassicProvider_InstanceList(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping test in short mode.")
|
|
}
|
|
|
|
ctx := t.Context()
|
|
dind := setupDinD(t)
|
|
p, err := docker.New(ctx, dind.client, slogt.New(t), "stop")
|
|
assert.NilError(t, err)
|
|
|
|
c1, err := dind.CreateMimic(ctx, MimicOptions{
|
|
Labels: map[string]string{
|
|
"sablier.enable": "true",
|
|
},
|
|
})
|
|
assert.NilError(t, err)
|
|
|
|
i1, err := dind.client.ContainerInspect(ctx, c1.ID)
|
|
assert.NilError(t, err)
|
|
|
|
assert.NilError(t, err)
|
|
|
|
c2, err := dind.CreateMimic(ctx, MimicOptions{
|
|
Labels: map[string]string{
|
|
"sablier.enable": "true",
|
|
"sablier.group": "my-group",
|
|
},
|
|
})
|
|
assert.NilError(t, err)
|
|
|
|
i2, err := dind.client.ContainerInspect(ctx, c2.ID)
|
|
assert.NilError(t, err)
|
|
|
|
got, err := p.InstanceList(ctx, provider.InstanceListOptions{
|
|
All: true,
|
|
})
|
|
assert.NilError(t, err)
|
|
|
|
want := []sablier.InstanceConfiguration{
|
|
{
|
|
Name: strings.TrimPrefix(i1.Name, "/"),
|
|
Group: "default",
|
|
},
|
|
{
|
|
Name: strings.TrimPrefix(i2.Name, "/"),
|
|
Group: "my-group",
|
|
},
|
|
}
|
|
// Assert go is equal to want
|
|
// Sort both array to ensure they are equal
|
|
sort.Slice(got, func(i, j int) bool {
|
|
return strings.Compare(got[i].Name, got[j].Name) < 0
|
|
})
|
|
sort.Slice(want, func(i, j int) bool {
|
|
return strings.Compare(want[i].Name, want[j].Name) < 0
|
|
})
|
|
assert.DeepEqual(t, got, want)
|
|
}
|
|
|
|
func TestDockerClassicProvider_GetGroups(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping test in short mode.")
|
|
}
|
|
|
|
ctx := t.Context()
|
|
dind := setupDinD(t)
|
|
p, err := docker.New(ctx, dind.client, slogt.New(t), "stop")
|
|
assert.NilError(t, err)
|
|
|
|
c1, err := dind.CreateMimic(ctx, MimicOptions{
|
|
Labels: map[string]string{
|
|
"sablier.enable": "true",
|
|
},
|
|
})
|
|
assert.NilError(t, err)
|
|
|
|
i1, err := dind.client.ContainerInspect(ctx, c1.ID)
|
|
assert.NilError(t, err)
|
|
|
|
assert.NilError(t, err)
|
|
|
|
c2, err := dind.CreateMimic(ctx, MimicOptions{
|
|
Labels: map[string]string{
|
|
"sablier.enable": "true",
|
|
"sablier.group": "my-group",
|
|
},
|
|
})
|
|
assert.NilError(t, err)
|
|
|
|
i2, err := dind.client.ContainerInspect(ctx, c2.ID)
|
|
assert.NilError(t, err)
|
|
|
|
got, err := p.InstanceGroups(ctx)
|
|
assert.NilError(t, err)
|
|
|
|
want := map[string][]string{
|
|
"default": {strings.TrimPrefix(i1.Name, "/")},
|
|
"my-group": {strings.TrimPrefix(i2.Name, "/")},
|
|
}
|
|
|
|
assert.DeepEqual(t, got, want)
|
|
}
|