111 lines
2.8 KiB
Go
111 lines
2.8 KiB
Go
//go:build integration
|
|
|
|
package mediaserver_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.netflux.io/rob/octoplex/container"
|
|
"git.netflux.io/rob/octoplex/mediaserver"
|
|
"git.netflux.io/rob/octoplex/testhelpers"
|
|
"github.com/docker/docker/client"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const component = "mediaserver"
|
|
|
|
func TestIntegrationMediaServerStartStop(t *testing.T) {
|
|
logger := testhelpers.NewTestLogger()
|
|
apiClient, err := client.NewClientWithOpts(client.FromEnv)
|
|
require.NoError(t, err)
|
|
|
|
containerClient, err := container.NewClient(t.Context(), apiClient, logger)
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { require.NoError(t, containerClient.Close()) })
|
|
|
|
running, err := containerClient.ContainerRunning(t.Context(), containerClient.ContainersWithLabels(map[string]string{container.LabelComponent: component}))
|
|
require.NoError(t, err)
|
|
assert.False(t, running)
|
|
|
|
// We need to avoid clashing with other integration tests, e.g. multiplexer.
|
|
const (
|
|
apiPort = 9999
|
|
rtmpPort = 1937
|
|
)
|
|
|
|
rtmpURL := fmt.Sprintf("rtmp://localhost:%d/live", rtmpPort)
|
|
|
|
mediaServer := mediaserver.StartActor(t.Context(), mediaserver.StartActorParams{
|
|
RTMPPort: rtmpPort,
|
|
APIPort: apiPort,
|
|
FetchIngressStateInterval: 500 * time.Millisecond,
|
|
ChanSize: 1,
|
|
ContainerClient: containerClient,
|
|
Logger: logger,
|
|
})
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { mediaServer.Close() })
|
|
testhelpers.ChanDiscard(mediaServer.C())
|
|
|
|
require.Eventually(
|
|
t,
|
|
func() bool {
|
|
running, err = containerClient.ContainerRunning(t.Context(), containerClient.ContainersWithLabels(map[string]string{container.LabelComponent: component}))
|
|
return err == nil && running
|
|
},
|
|
time.Second*10,
|
|
time.Second,
|
|
"container not in RUNNING state",
|
|
)
|
|
|
|
state := mediaServer.State()
|
|
assert.False(t, state.Live)
|
|
assert.Equal(t, rtmpURL, state.RTMPURL)
|
|
|
|
testhelpers.StreamFLV(t, rtmpURL)
|
|
|
|
require.Eventually(
|
|
t,
|
|
func() bool {
|
|
currState := mediaServer.State()
|
|
return currState.Live &&
|
|
!currState.LiveChangedAt.IsZero() &&
|
|
currState.Container.HealthState == "healthy"
|
|
},
|
|
time.Second*5,
|
|
time.Second,
|
|
"actor not healthy and/or in LIVE state",
|
|
)
|
|
|
|
require.Eventually(
|
|
t,
|
|
func() bool {
|
|
currState := mediaServer.State()
|
|
return len(currState.Tracks) == 1 && currState.Tracks[0] == "H264"
|
|
},
|
|
time.Second*5,
|
|
time.Second,
|
|
"tracks not updated",
|
|
)
|
|
|
|
require.Eventually(
|
|
t,
|
|
func() bool {
|
|
currState := mediaServer.State()
|
|
return currState.Container.RxRate > 500
|
|
},
|
|
time.Second*10,
|
|
time.Second,
|
|
"rxRate not updated",
|
|
)
|
|
|
|
mediaServer.Close()
|
|
|
|
running, err = containerClient.ContainerRunning(t.Context(), containerClient.ContainersWithLabels(map[string]string{container.LabelComponent: component}))
|
|
require.NoError(t, err)
|
|
assert.False(t, running)
|
|
}
|