54 lines
2.7 KiB
Go
54 lines
2.7 KiB
Go
package edgegroups
|
|
|
|
import (
|
|
"testing"
|
|
|
|
portainer "github.com/portainer/portainer/api"
|
|
"github.com/portainer/portainer/api/internal/testhelpers"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_getEndpointTypes(t *testing.T) {
|
|
endpoints := []portainer.Endpoint{
|
|
{ID: 1, Type: portainer.DockerEnvironment},
|
|
{ID: 2, Type: portainer.AgentOnDockerEnvironment},
|
|
{ID: 3, Type: portainer.AzureEnvironment},
|
|
{ID: 4, Type: portainer.EdgeAgentOnDockerEnvironment},
|
|
{ID: 5, Type: portainer.KubernetesLocalEnvironment},
|
|
{ID: 6, Type: portainer.AgentOnKubernetesEnvironment},
|
|
{ID: 7, Type: portainer.EdgeAgentOnKubernetesEnvironment},
|
|
}
|
|
|
|
datastore := testhelpers.NewDatastore(testhelpers.WithEndpoints(endpoints))
|
|
|
|
tests := []struct {
|
|
endpointIds []portainer.EndpointID
|
|
expected []portainer.EndpointType
|
|
}{
|
|
{endpointIds: []portainer.EndpointID{1}, expected: []portainer.EndpointType{portainer.DockerEnvironment}},
|
|
{endpointIds: []portainer.EndpointID{2}, expected: []portainer.EndpointType{portainer.AgentOnDockerEnvironment}},
|
|
{endpointIds: []portainer.EndpointID{3}, expected: []portainer.EndpointType{portainer.AzureEnvironment}},
|
|
{endpointIds: []portainer.EndpointID{4}, expected: []portainer.EndpointType{portainer.EdgeAgentOnDockerEnvironment}},
|
|
{endpointIds: []portainer.EndpointID{5}, expected: []portainer.EndpointType{portainer.KubernetesLocalEnvironment}},
|
|
{endpointIds: []portainer.EndpointID{6}, expected: []portainer.EndpointType{portainer.AgentOnKubernetesEnvironment}},
|
|
{endpointIds: []portainer.EndpointID{7}, expected: []portainer.EndpointType{portainer.EdgeAgentOnKubernetesEnvironment}},
|
|
{endpointIds: []portainer.EndpointID{7, 2}, expected: []portainer.EndpointType{portainer.EdgeAgentOnKubernetesEnvironment, portainer.AgentOnDockerEnvironment}},
|
|
{endpointIds: []portainer.EndpointID{6, 4, 1}, expected: []portainer.EndpointType{portainer.AgentOnKubernetesEnvironment, portainer.EdgeAgentOnDockerEnvironment, portainer.DockerEnvironment}},
|
|
{endpointIds: []portainer.EndpointID{1, 2, 3}, expected: []portainer.EndpointType{portainer.DockerEnvironment, portainer.AgentOnDockerEnvironment, portainer.AzureEnvironment}},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
ans, err := getEndpointTypes(datastore.Endpoint(), test.endpointIds)
|
|
assert.NoError(t, err, "getEndpointTypes shouldn't fail")
|
|
|
|
assert.ElementsMatch(t, test.expected, ans, "getEndpointTypes expected to return %b for %v, but returned %b", test.expected, test.endpointIds, ans)
|
|
}
|
|
}
|
|
|
|
func Test_getEndpointTypes_failWhenEndpointDontExist(t *testing.T) {
|
|
datastore := testhelpers.NewDatastore(testhelpers.WithEndpoints([]portainer.Endpoint{}))
|
|
|
|
_, err := getEndpointTypes(datastore.Endpoint(), []portainer.EndpointID{1})
|
|
assert.Error(t, err, "getEndpointTypes should fail")
|
|
}
|