forked from External/mediamtx
73ddb21e63
* implement native recording (#1399) * support saving VP9 tracks * support saving MPEG-1 audio tracks * switch segment when codec parameters change * allow to disable recording on a path basis * allow disabling recording cleaner * support recording MPEG-1/2/4 video tracks * add microseconds to file names * add tests
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package record
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCleaner(t *testing.T) {
|
|
timeNow = func() time.Time {
|
|
return time.Date(2009, 0o5, 20, 22, 15, 25, 427000, time.UTC)
|
|
}
|
|
|
|
dir, err := os.MkdirTemp("", "mediamtx-cleaner")
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(dir)
|
|
|
|
recordPath := filepath.Join(dir, "%path/%Y-%m-%d_%H-%M-%S-%f")
|
|
|
|
err = os.Mkdir(filepath.Join(dir, "mypath"), 0o755)
|
|
require.NoError(t, err)
|
|
|
|
err = os.WriteFile(filepath.Join(dir, "mypath", "2008-05-20_22-15-25-000125.mp4"), []byte{1}, 0o644)
|
|
require.NoError(t, err)
|
|
|
|
err = os.WriteFile(filepath.Join(dir, "mypath", "2009-05-20_22-15-25-000427.mp4"), []byte{1}, 0o644)
|
|
require.NoError(t, err)
|
|
|
|
c := NewCleaner(
|
|
recordPath,
|
|
10*time.Second,
|
|
nilLogger{},
|
|
)
|
|
defer c.Close()
|
|
|
|
time.Sleep(500 * time.Millisecond)
|
|
|
|
_, err = os.Stat(filepath.Join(dir, "mypath", "2008-05-20_22-15-25-000125.mp4"))
|
|
require.Error(t, err)
|
|
|
|
_, err = os.Stat(filepath.Join(dir, "mypath", "2009-05-20_22-15-25-000427.mp4"))
|
|
require.NoError(t, err)
|
|
}
|