mirror of
https://github.com/asticode/go-astiav.git
synced 2024-05-28 13:53:18 +08:00
64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
package astiav
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"unsafe"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestClass(t *testing.T) {
|
|
c := FindDecoder(CodecIDMjpeg)
|
|
require.NotNil(t, c)
|
|
cc := AllocCodecContext(c)
|
|
require.NotNil(t, cc)
|
|
defer cc.Free()
|
|
|
|
cl := cc.Class()
|
|
require.NotNil(t, cl)
|
|
require.Equal(t, ClassCategoryDecoder, cl.Category())
|
|
require.Equal(t, "mjpeg", cl.ItemName())
|
|
require.Equal(t, "AVCodecContext", cl.Name())
|
|
require.Equal(t, fmt.Sprintf("mjpeg [AVCodecContext] @ %p", cc.c), cl.String())
|
|
// TODO Test parent
|
|
}
|
|
|
|
func TestClassers(t *testing.T) {
|
|
cl := len(classers.p)
|
|
f := AllocFilterGraph()
|
|
c := FindDecoder(CodecIDMjpeg)
|
|
require.NotNil(t, c)
|
|
cc := AllocCodecContext(c)
|
|
require.NotNil(t, cc)
|
|
bufferSink := FindFilterByName("buffersink")
|
|
require.NotNil(t, bufferSink)
|
|
fc, err := f.NewFilterContext(bufferSink, "filter_out", nil)
|
|
require.NoError(t, err)
|
|
fmc1 := AllocFormatContext()
|
|
fmc2 := AllocFormatContext()
|
|
require.NoError(t, fmc2.OpenInput("testdata/video.mp4", nil, nil))
|
|
path := filepath.Join(t.TempDir(), "iocontext.txt")
|
|
ic, err := OpenIOContext(path, NewIOContextFlags(IOContextFlagWrite))
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(path)
|
|
ssc, err := CreateSoftwareScaleContext(1, 1, PixelFormatRgba, 2, 2, PixelFormatRgba, NewSoftwareScaleContextFlags())
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, cl+8, len(classers.p))
|
|
v, ok := classers.get(unsafe.Pointer(f.c))
|
|
require.True(t, ok)
|
|
require.Equal(t, f, v)
|
|
|
|
cc.Free()
|
|
fc.Free()
|
|
f.Free()
|
|
fmc1.Free()
|
|
fmc2.CloseInput()
|
|
require.NoError(t, ic.Close())
|
|
ssc.Free()
|
|
require.Equal(t, cl, len(classers.p))
|
|
}
|