mirror of
https://github.com/datarhei/core.git
synced 2025-10-16 04:50:44 +08:00
202 lines
7.4 KiB
Go
202 lines
7.4 KiB
Go
package probe
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestProber(t *testing.T) {
|
|
prober := New(Config{}).(*prober)
|
|
|
|
rawdata := `ffmpeg version 4.0.2 Copyright (c) 2000-2018 the FFmpeg developers
|
|
built with Apple LLVM version 9.1.0 (clang-902.0.39.2)
|
|
configuration: --prefix=/usr/local/Cellar/ffmpeg/4.0.2 --enable-shared --enable-pthreads --enable-version3 --enable-hardcoded-tables --enable-avresample --cc=clang --host-cflags= --host-ldflags= --enable-gpl --enable-libmp3lame --enable-libx264 --enable-libx265 --enable-libxvid --enable-opencl --enable-videotoolbox --disable-lzma
|
|
libavutil 56. 14.100 / 56. 14.100
|
|
libavcodec 58. 18.100 / 58. 18.100
|
|
libavformat 58. 12.100 / 58. 12.100
|
|
libavdevice 58. 3.100 / 58. 3.100
|
|
libavfilter 7. 16.100 / 7. 16.100
|
|
libavresample 4. 0. 0 / 4. 0. 0
|
|
libswscale 5. 1.100 / 5. 1.100
|
|
libswresample 3. 1.100 / 3. 1.100
|
|
libpostproc 55. 1.100 / 55. 1.100
|
|
Input #0, lavfi, from 'testsrc=size=1280x720:rate=25':
|
|
Duration: N/A, start: 0.000000, bitrate: N/A
|
|
Stream #0:0: Video: rawvideo (RGB[24] / 0x18424752), rgb24, 1280x720 [SAR 1:1 DAR 16:9], 25 tbr, 25 tbn, 25 tbc
|
|
Input #1, lavfi, from 'anullsrc=r=44100:cl=stereo':
|
|
Duration: N/A, start: 0.000000, bitrate: 705 kb/s
|
|
Stream #1:0: Audio: pcm_u8, 44100 Hz, stereo, u8, 705 kb/s
|
|
Input #2, playout, from 'playout:rtmp://l5gn74l5-vpu.livespotting.com/live/0chl6hu7_360?token=m5ZuiCQYRlIon8':
|
|
Duration: N/A, start: 0.000000, bitrate: 265 kb/s
|
|
Stream #2:0: Video: h264 (Constrained Baseline), yuvj420p(pc, progressive), 640x360 [SAR 1:1 DAR 16:9], 265 kb/s, 10 fps, 10 tbr, 1000k tbn, 20 tbc
|
|
Input #3, mov,mp4,m4a,3gp,3g2,mj2, from 'movie.mp4':
|
|
Metadata:
|
|
major_brand : isom
|
|
minor_version : 512
|
|
compatible_brands: isomiso2avc1mp41
|
|
encoder : Lavf58.20.100
|
|
Duration: 00:01:02.28, start: 0.000000, bitrate: 5895 kb/s
|
|
Stream #3:0(eng): Video: h264 (Main) (avc1 / 0x31637661), yuvj420p(pc, bt709), 2560x1440 [SAR 1:1 DAR 16:9], 5894 kb/s, 23.98 fps, 25 tbr, 90k tbn, 50 tbc (default)
|
|
Stream #3:1(por): Subtitle: subrip
|
|
Input #4, mpegts, from 'srt://localhost:6000?mode=caller&transtype=live&streamid=#!:m=request,r=ingest/ad045490-8233-4f31-a296-ea5771a340ac&passphrase=foobarfoobar':
|
|
Duration: N/A, start: 71.786667, bitrate: N/A
|
|
Program 1
|
|
Metadata:
|
|
service_name : Service01
|
|
service_provider: FFmpeg
|
|
Stream #4:0[0x100]: Video: h264 (Main) ([27][0][0][0] / 0x001B), yuv420p(tv, smpte170m/bt709/bt709, progressive), 1920x1080 [SAR 1:1 DAR 16:9], 25 tbr, 90k tbn
|
|
Stream #4:1[0x101]: Audio: aac (LC) ([15][0][0][0] / 0x000F), 48000 Hz, stereo, fltp, 162 kb/s
|
|
Stream mapping:
|
|
Stream #0:0 -> #0:0 (rawvideo (native) -> h264 (libx264))
|
|
Stream #1:0 -> #0:1 (pcm_u8 (native) -> aac (native))
|
|
Press [q] to stop, [?] for help`
|
|
|
|
data := strings.Split(rawdata, "\n")
|
|
|
|
for _, d := range data {
|
|
prober.Parse([]byte(d))
|
|
}
|
|
|
|
prober.ResetStats()
|
|
|
|
require.Equal(t, 7, len(prober.inputs))
|
|
|
|
i := prober.inputs[0]
|
|
|
|
require.Equal(t, "testsrc=size=1280x720:rate=25", i.Address)
|
|
require.Equal(t, "lavfi", i.Format)
|
|
require.Equal(t, uint64(0), i.Index)
|
|
require.Equal(t, uint64(0), i.Stream)
|
|
require.Equal(t, "und", i.Language)
|
|
require.Equal(t, "video", i.Type)
|
|
require.Equal(t, "rawvideo", i.Codec)
|
|
require.Equal(t, 0.0, i.Bitrate)
|
|
require.Equal(t, 0.0, i.Duration)
|
|
require.Equal(t, 0.0, i.FPS)
|
|
require.Equal(t, "rgb24", i.Pixfmt)
|
|
require.Equal(t, uint64(1280), i.Width)
|
|
require.Equal(t, uint64(720), i.Height)
|
|
|
|
i = prober.inputs[1]
|
|
|
|
require.Equal(t, "anullsrc=r=44100:cl=stereo", i.Address)
|
|
require.Equal(t, "lavfi", i.Format)
|
|
require.Equal(t, uint64(1), i.Index)
|
|
require.Equal(t, uint64(0), i.Stream)
|
|
require.Equal(t, "und", i.Language)
|
|
require.Equal(t, "audio", i.Type)
|
|
require.Equal(t, "pcm_u8", i.Codec)
|
|
require.Equal(t, 705.0, i.Bitrate)
|
|
require.Equal(t, 0.0, i.Duration)
|
|
require.Equal(t, uint64(44100), i.Sampling)
|
|
require.Equal(t, "stereo", i.Layout)
|
|
|
|
i = prober.inputs[2]
|
|
|
|
require.Equal(t, "playout:rtmp://l5gn74l5-vpu.livespotting.com/live/0chl6hu7_360?token=m5ZuiCQYRlIon8", i.Address)
|
|
require.Equal(t, "playout", i.Format)
|
|
require.Equal(t, uint64(2), i.Index)
|
|
require.Equal(t, uint64(0), i.Stream)
|
|
require.Equal(t, "und", i.Language)
|
|
require.Equal(t, "video", i.Type)
|
|
require.Equal(t, "h264", i.Codec)
|
|
require.Equal(t, 265.0, i.Bitrate)
|
|
require.Equal(t, 0.0, i.Duration)
|
|
require.Equal(t, 10.0, i.FPS)
|
|
require.Equal(t, "yuvj420p", i.Pixfmt)
|
|
require.Equal(t, uint64(640), i.Width)
|
|
require.Equal(t, uint64(360), i.Height)
|
|
|
|
i = prober.inputs[3]
|
|
|
|
require.Equal(t, "movie.mp4", i.Address)
|
|
require.Equal(t, "mov,mp4,m4a,3gp,3g2,mj2", i.Format)
|
|
require.Equal(t, uint64(3), i.Index)
|
|
require.Equal(t, uint64(0), i.Stream)
|
|
require.Equal(t, "eng", i.Language)
|
|
require.Equal(t, "video", i.Type)
|
|
require.Equal(t, "h264", i.Codec)
|
|
require.Equal(t, 5894.0, i.Bitrate)
|
|
require.Equal(t, 62.28, i.Duration)
|
|
require.Equal(t, 23.98, i.FPS)
|
|
require.Equal(t, "yuvj420p", i.Pixfmt)
|
|
require.Equal(t, uint64(2560), i.Width)
|
|
require.Equal(t, uint64(1440), i.Height)
|
|
|
|
i = prober.inputs[4]
|
|
|
|
require.Equal(t, "movie.mp4", i.Address)
|
|
require.Equal(t, "mov,mp4,m4a,3gp,3g2,mj2", i.Format)
|
|
require.Equal(t, uint64(3), i.Index)
|
|
require.Equal(t, uint64(1), i.Stream)
|
|
require.Equal(t, "por", i.Language)
|
|
require.Equal(t, "subtitle", i.Type)
|
|
require.Equal(t, "subrip", i.Codec)
|
|
|
|
i = prober.inputs[5]
|
|
|
|
require.Equal(t, "srt://localhost:6000?mode=caller&transtype=live&streamid=#!:m=request,r=ingest/ad045490-8233-4f31-a296-ea5771a340ac&passphrase=foobarfoobar", i.Address)
|
|
require.Equal(t, "mpegts", i.Format)
|
|
require.Equal(t, uint64(4), i.Index)
|
|
require.Equal(t, uint64(0), i.Stream)
|
|
require.Equal(t, "und", i.Language)
|
|
require.Equal(t, "video", i.Type)
|
|
require.Equal(t, "h264", i.Codec)
|
|
require.Equal(t, 0.0, i.Bitrate)
|
|
require.Equal(t, 0.0, i.Duration)
|
|
require.Equal(t, 0.0, i.FPS)
|
|
require.Equal(t, "yuv420p", i.Pixfmt)
|
|
require.Equal(t, uint64(1920), i.Width)
|
|
require.Equal(t, uint64(1080), i.Height)
|
|
|
|
i = prober.inputs[6]
|
|
|
|
require.Equal(t, "srt://localhost:6000?mode=caller&transtype=live&streamid=#!:m=request,r=ingest/ad045490-8233-4f31-a296-ea5771a340ac&passphrase=foobarfoobar", i.Address)
|
|
require.Equal(t, "mpegts", i.Format)
|
|
require.Equal(t, uint64(4), i.Index)
|
|
require.Equal(t, uint64(1), i.Stream)
|
|
require.Equal(t, "und", i.Language)
|
|
require.Equal(t, "audio", i.Type)
|
|
require.Equal(t, "aac", i.Codec)
|
|
require.Equal(t, 162.0, i.Bitrate)
|
|
require.Equal(t, 0.0, i.Duration)
|
|
require.Equal(t, uint64(48000), i.Sampling)
|
|
require.Equal(t, "stereo", i.Layout)
|
|
}
|
|
|
|
func TestJSON(t *testing.T) {
|
|
prober := New(Config{}).(*prober)
|
|
|
|
prober.Parse([]byte("foobar"))
|
|
prober.Parse([]byte(`ffmpeg.inputs:[{"url":"https://cdn.livespotting.com/vpu/e9slfpe3/z60wzayk.m3u8","format":"playout","index":0,"stream":0,"type":"video","codec":"h264","coder":"h264","bitrate_kbps":0,"duration_sec":0.000000,"language":"und","fps":20.666666,"pix_fmt":"yuvj420p","width":1280,"height":720}]`))
|
|
|
|
prober.ResetStats()
|
|
|
|
probe := prober.Probe()
|
|
|
|
require.Equal(t, "foobar", probe.Log[0])
|
|
require.Equal(t, []ProbeIO{
|
|
{
|
|
Address: "https://cdn.livespotting.com/vpu/e9slfpe3/z60wzayk.m3u8",
|
|
Index: 0,
|
|
Stream: 0,
|
|
Language: "und",
|
|
Format: "playout",
|
|
Type: "video",
|
|
Codec: "h264",
|
|
Coder: "h264",
|
|
Bitrate: 0,
|
|
Duration: 0,
|
|
Pixfmt: "yuvj420p",
|
|
Width: 1280,
|
|
Height: 720,
|
|
FPS: 20.666666,
|
|
Sampling: 0,
|
|
Layout: "",
|
|
Channels: 0,
|
|
},
|
|
}, probe.Streams)
|
|
}
|