ffmpeg_helper_test.go 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package ffmpeg_helper
  2. import (
  3. "github.com/allanpk716/ChineseSubFinder/internal/pkg"
  4. "io/ioutil"
  5. "path/filepath"
  6. "testing"
  7. )
  8. func TestGetSubTileIndexList(t *testing.T) {
  9. videoFile := "X:\\连续剧\\瑞克和莫蒂 (2013)\\Season 5\\Rick and Morty - S05E10 - Rickmurai Jack WEBRip-1080p.mkv"
  10. err := GetSubTileIndexList(videoFile)
  11. if err != nil {
  12. t.Fatal(err)
  13. }
  14. }
  15. func Test_parseJsonString2GetAudioAndSubs(t *testing.T) {
  16. testDataPath := "../../../TestData/ffmpeg"
  17. testRootDir, err := pkg.CopyTestData(testDataPath)
  18. if err != nil {
  19. t.Fatal(err)
  20. }
  21. type args struct {
  22. input string
  23. }
  24. tests := []struct {
  25. name string
  26. args args
  27. wantErr bool
  28. }{
  29. {name: "R&M S05E10", args: args{input: readString(filepath.Join(testRootDir, "R&M S05E10-video_stream.json"))}, wantErr: false},
  30. }
  31. for _, tt := range tests {
  32. t.Run(tt.name, func(t *testing.T) {
  33. if err := parseJsonString2GetAudioAndSubs(tt.args.input); (err != nil) != tt.wantErr {
  34. t.Errorf("parseJsonString2GetAudioAndSubs() error = %v, wantErr %v", err, tt.wantErr)
  35. }
  36. })
  37. }
  38. }
  39. func readString(filePath string) string {
  40. bytes, err := ioutil.ReadFile(filePath)
  41. if err != nil {
  42. return ""
  43. }
  44. return string(bytes)
  45. }