rob/octoplex
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

test(integration): add coverage for custom stream key

Browse Source
This commit is contained in:
Rob Watson 2025-04-11 09:52:18 +02:00
parent b69e1ecf86
commit f7f9843c4b
2 changed files with 24 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/app/integration_helpers_test.go
View File

@ -89,7 +89,7 @@ func contentsIncludes(contents []string, search string) bool {
} }
func setupConfigService(t *testing.T, cfg config.Config) *config.Service { func setupConfigService(t *testing.T, cfg config.Config) *config.Service {
tmpDir, err := os.MkdirTemp("", "octoplex_"+t.Name()) tmpDir, err := os.MkdirTemp("", "octoplex_"+strings.ReplaceAll(t.Name(), "/", "_"))
require.NoError(t, err) require.NoError(t, err)
t.Cleanup(func() { os.RemoveAll(tmpDir) }) t.Cleanup(func() { os.RemoveAll(tmpDir) })
configService, err := config.NewService(func() (string, error) { return tmpDir, nil }, 1) configService, err := config.NewService(func() (string, error) { return tmpDir, nil }, 1)

34
internal/app/integration_test.go
View File

@ -3,6 +3,7 @@
package app_test package app_test
import ( import (
"cmp"
"context" "context"
"errors" "errors"
"fmt" "fmt"
@ -33,9 +34,21 @@ import (
) )
func TestIntegration(t *testing.T) { func TestIntegration(t *testing.T) {
t.Run("with default stream key", func(t *testing.T) {
testIntegration(t, "")
})
t.Run("with custom stream key", func(t *testing.T) {
testIntegration(t, "s0meK3y")
})
}
func testIntegration(t *testing.T, streamKey string) {
ctx, cancel := context.WithTimeout(t.Context(), 10*time.Minute) ctx, cancel := context.WithTimeout(t.Context(), 10*time.Minute)
defer cancel() defer cancel()
wantStreamKey := cmp.Or(streamKey, "live")
destServer, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{ destServer, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
ContainerRequest: testcontainers.ContainerRequest{ ContainerRequest: testcontainers.ContainerRequest{
Image: "bluenviron/mediamtx:latest", Image: "bluenviron/mediamtx:latest",
@ -75,10 +88,10 @@ func TestIntegration(t *testing.T) {
} }
const destHost = "172.17.0.1" const destHost = "172.17.0.1"
destURL1 := fmt.Sprintf("rtmp://%s:%d/live/dest1", destHost, destServerPort.Int()) destURL1 := fmt.Sprintf("rtmp://%s:%d/%s/dest1", destHost, destServerPort.Int(), wantStreamKey)
destURL2 := fmt.Sprintf("rtmp://%s:%d/live/dest2", destHost, destServerPort.Int()) destURL2 := fmt.Sprintf("rtmp://%s:%d/%s/dest2", destHost, destServerPort.Int(), wantStreamKey)
configService := setupConfigService(t, config.Config{ configService := setupConfigService(t, config.Config{
Sources: config.Sources{RTMP: config.RTMPSource{Enabled: true, StreamKey: "live"}}, Sources: config.Sources{RTMP: config.RTMPSource{Enabled: true, StreamKey: streamKey}},
// Load one destination from config, add the other in-app. // Load one destination from config, add the other in-app.
Destinations: []config.Destination{{Name: "Local server 1", URL: destURL1}}, Destinations: []config.Destination{{Name: "Local server 1", URL: destURL1}},
}) })
@ -120,14 +133,15 @@ func TestIntegration(t *testing.T) {
printScreen(getContents, "After starting the mediaserver") printScreen(getContents, "After starting the mediaserver")
// Start streaming a test video to the app: // Start streaming a test video to the app:
testhelpers.StreamFLV(t, "rtmp://localhost:1935/live") testhelpers.StreamFLV(t, "rtmp://localhost:1935/"+wantStreamKey)
require.EventuallyWithT( require.EventuallyWithT(
t, t,
func(t *assert.CollectT) { func(t *assert.CollectT) {
contents := getContents() contents := getContents()
require.True(t, len(contents) > 2, "expected at least 3 lines of output") require.True(t, len(contents) > 4, "expected at least 5 lines of output")
assert.Contains(t, contents[1], "URL rtmp://localhost:1935/"+wantStreamKey, "expected mediaserver status to be receiving")
assert.Contains(t, contents[2], "Status receiving", "expected mediaserver status to be receiving") assert.Contains(t, contents[2], "Status receiving", "expected mediaserver status to be receiving")
assert.Contains(t, contents[3], "Tracks H264", "expected mediaserver tracks to be H264") assert.Contains(t, contents[3], "Tracks H264", "expected mediaserver tracks to be H264")
assert.Contains(t, contents[4], "Health healthy", "expected mediaserver to be healthy") assert.Contains(t, contents[4], "Health healthy", "expected mediaserver to be healthy")
@ -154,7 +168,7 @@ func TestIntegration(t *testing.T) {
t, t,
func(t *assert.CollectT) { func(t *assert.CollectT) {
contents := getContents() contents := getContents()
require.True(t, len(contents) > 2, "expected at least 3 lines of output") require.True(t, len(contents) > 4, "expected at least 5 lines of output")
assert.Contains(t, contents[2], "Status receiving", "expected mediaserver status to be receiving") assert.Contains(t, contents[2], "Status receiving", "expected mediaserver status to be receiving")
assert.Contains(t, contents[3], "Tracks H264", "expected mediaserver tracks to be H264") assert.Contains(t, contents[3], "Tracks H264", "expected mediaserver tracks to be H264")
@ -182,7 +196,7 @@ func TestIntegration(t *testing.T) {
t, t,
func(t *assert.CollectT) { func(t *assert.CollectT) {
contents := getContents() contents := getContents()
require.True(t, len(contents) > 2, "expected at least 3 lines of output") require.True(t, len(contents) > 4, "expected at least 5 lines of output")
assert.Contains(t, contents[2], "Status receiving", "expected mediaserver status to be receiving") assert.Contains(t, contents[2], "Status receiving", "expected mediaserver status to be receiving")
assert.Contains(t, contents[3], "Tracks H264", "expected mediaserver tracks to be H264") assert.Contains(t, contents[3], "Tracks H264", "expected mediaserver tracks to be H264")
@ -210,7 +224,7 @@ func TestIntegration(t *testing.T) {
t, t,
func(t *assert.CollectT) { func(t *assert.CollectT) {
contents := getContents() contents := getContents()
require.True(t, len(contents) > 2, "expected at least 3 lines of output") require.True(t, len(contents) > 4, "expected at least 5 lines of output")
assert.Contains(t, contents[2], "Status receiving", "expected mediaserver status to be receiving") assert.Contains(t, contents[2], "Status receiving", "expected mediaserver status to be receiving")
assert.Contains(t, contents[3], "Tracks H264", "expected mediaserver tracks to be H264") assert.Contains(t, contents[3], "Tracks H264", "expected mediaserver tracks to be H264")
@ -230,15 +244,13 @@ func TestIntegration(t *testing.T) {
printScreen(getContents, "After removing the second destination") printScreen(getContents, "After removing the second destination")
// Stop remaining destination. // Stop remaining destination.
// It is currently necessary to press down to re-focus the destination:
sendKey(screen, tcell.KeyDown, ' ')
sendKey(screen, tcell.KeyRune, ' ') sendKey(screen, tcell.KeyRune, ' ')
require.EventuallyWithT( require.EventuallyWithT(
t, t,
func(t *assert.CollectT) { func(t *assert.CollectT) {
contents := getContents() contents := getContents()
require.True(t, len(contents) > 2, "expected at least 3 lines of output") require.True(t, len(contents) > 4, "expected at least 5 lines of output")
require.Contains(t, contents[2], "Local server 1", "expected local server 1 to be present") require.Contains(t, contents[2], "Local server 1", "expected local server 1 to be present")
assert.Contains(t, contents[2], "exited", "expected local server 1 to have exited") assert.Contains(t, contents[2], "exited", "expected local server 1 to have exited")