tests/container_listing_test.go

Rama master · réplica de lectura del repositorio original.

package main

import (
	"testing"

	"github.com/shellhub-io/shellhub/pkg/api/query"
	"github.com/shellhub-io/shellhub/pkg/models"
	"github.com/shellhub-io/shellhub/tests/environment"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

// TestContainerListing enrolls by hand one device reporting the connector platform, as a
// connector registers each container it serves, and one reporting the native platform, as an
// agent does, then reads back where the API lists each: /api/containers serves the connector's
// devices alone, and /api/devices every device but those.
func TestContainerListing(t *testing.T) {
	t.Parallel()

	compose := newSSHEnvironment(t, t.Context(), legacyNamespace)

	containerReq := newDeviceAuthRequest(t, "listed-container", "02:00:00:00:a1:01")
	containerReq.Info.Platform = "connector"
	container := enroll(t, compose, containerReq)
	require.Equal(t, "connector", container.Info.Platform)

	native := enroll(t, compose, newDeviceAuthRequest(t, "listed-native", "02:00:00:00:a1:02"))
	require.Equal(t, "native", native.Info.Platform)

	t.Run("pending devices are listed apart", func(t *testing.T) {
		for _, status := range []models.DeviceStatus{models.DeviceStatusEmpty, models.DeviceStatusPending} {
			requireListedApart(t, compose, status, container.UID, native.UID)
		}
	})

	t.Run("accepted devices are listed apart", func(t *testing.T) {
		compose.UpdateDeviceStatus(t, container.UID, environment.DeviceActionAccept)
		compose.UpdateDeviceStatus(t, native.UID, environment.DeviceActionAccept)

		for _, status := range []models.DeviceStatus{models.DeviceStatusEmpty, models.DeviceStatusAccepted} {
			requireListedApart(t, compose, status, container.UID, native.UID)
		}
	})

	t.Run("a filter naming a device from the other listing does not cross the split", func(t *testing.T) {
		containers := compose.ListContainersMatching(t, nameIsEither(native.Name, container.Name))
		assert.Equal(t, []string{container.UID}, deviceUIDs(containers))

		devices := compose.ListDevicesMatching(t, nameIsEither(container.Name, native.Name))
		assert.Equal(t, []string{native.UID}, deviceUIDs(devices))
	})
}

func nameIsEither(first, second string) []query.Filter {
	return []query.Filter{
		{Type: query.FilterTypeProperty, Params: &query.FilterProperty{Name: "name", Operator: "eq", Value: first}},
		{Type: query.FilterTypeOperator, Params: &query.FilterOperator{Name: "or"}},
		{Type: query.FilterTypeProperty, Params: &query.FilterProperty{Name: "name", Operator: "eq", Value: second}},
	}
}

func requireListedApart(t *testing.T, compose *environment.DockerCompose, status models.DeviceStatus, container, native string) {
	t.Helper()

	containers := deviceUIDs(compose.ListContainers(t, status))
	assert.Contains(t, containers, container, "the container is missing from the containers listed under status %q", status)
	assert.NotContains(t, containers, native, "the native device is among the containers listed under status %q", status)

	devices := deviceUIDs(compose.ListDevices(t, status))
	assert.Contains(t, devices, native, "the native device is missing from the devices listed under status %q", status)
	assert.NotContains(t, devices, container, "the container is among the devices listed under status %q", status)
}