zimuku_test.go 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. package zimuku
  2. import (
  3. series_helper2 "github.com/allanpk716/ChineseSubFinder/internal/logic/series_helper"
  4. "testing"
  5. )
  6. func TestSupplier_GetSubListFromKeyword(t *testing.T) {
  7. //imdbId1 := "tt3228774"
  8. videoName := "黑白魔女库伊拉"
  9. s := NewSupplier()
  10. outList, err := s.getSubListFromKeyword(videoName)
  11. if err != nil {
  12. t.Error(err)
  13. }
  14. println(outList)
  15. for i, sublist := range outList {
  16. println(i, sublist.Name, sublist.Ext, sublist.Language.String(), sublist.Score, len(sublist.Data))
  17. }
  18. }
  19. func TestSupplier_GetSubListFromFile(t *testing.T) {
  20. movie1 := "XLen:\\电影\\The Devil All the Time (2020)\\The Devil All the Time (2020) WEBDL-1080p.mkv"
  21. //movie1 := "XLen:\\电影\\龙猫 (1988)\\龙猫 (1988) 1080p DTS.mkv"
  22. //movie1 := "XLen:\\电影\\消失爱人 (2016)\\消失爱人 (2016) 720p AAC.rmvb"
  23. //movie1:= "XLen:\\电影\\Spiral From the Book of Saw (2021)\\Spiral From the Book of Saw (2021) WEBDL-1080p.mkv"
  24. //movie1 := "XLen:\\电影\\机动战士Z高达:星之继承者 (2005)\\机动战士Z高达:星之继承者 (2005) 1080p TrueHD.mkv"
  25. //movie1 := "XLen:\\连续剧\\The Bad Batch\\Season 1\\The Bad Batch - S01E01 - Aftermath WEBDL-1080p.mkv"
  26. s := NewSupplier()
  27. outList, err := s.getSubListFromMovie(movie1)
  28. if err != nil {
  29. t.Error(err)
  30. }
  31. println(outList)
  32. for i, sublist := range outList {
  33. println(i, sublist.Name, sublist.Ext, sublist.Language.String(), sublist.Score, len(sublist.Data))
  34. }
  35. }
  36. func TestSupplier_GetSubListFromFile4Series(t *testing.T) {
  37. ser := "XLen:\\连续剧\\The Bad Batch" // tt12708542
  38. //ser := "XLen:\\连续剧\\杀死伊芙 (2018)" // tt12708542
  39. //ser := "XLen:\\连续剧\\Money.Heist"
  40. // 读取本地的视频和字幕信息
  41. seriesInfo, err := series_helper2.ReadSeriesInfoFromDir(ser, nil, false)
  42. if err != nil {
  43. t.Fatal(err)
  44. }
  45. s := NewSupplier()
  46. outList, err := s.GetSubListFromFile4Series(seriesInfo)
  47. if err != nil {
  48. t.Fatal(err)
  49. }
  50. println(outList)
  51. for i, sublist := range outList {
  52. println(i, sublist.Name, sublist.Ext, sublist.Language.String(), sublist.Score, len(sublist.Data))
  53. }
  54. }
  55. func TestSupplier_getSubListFromKeyword(t *testing.T) {
  56. //imdbID := "tt12708542" // 星球大战:残次品
  57. //imdbID := "tt7016936" // 杀死伊芙
  58. //imdbID := "tt2990738" // 恐怖直播
  59. //imdbID := "tt3032476" // 风骚律师
  60. //imdbID := "tt6468322" // 纸钞屋
  61. //imdbID := "tt15299712" // 云南虫谷
  62. //imdbID := "tt3626476" // Vacation Friends (2021)
  63. imdbID := "tt11192306" // Superman.and.Lois
  64. subhd := NewSupplier()
  65. subInfos, err := subhd.getSubListFromKeyword(imdbID)
  66. if err != nil {
  67. t.Fatal(err)
  68. }
  69. for i, sublist := range subInfos {
  70. println(i, sublist.Name, sublist.Ext, sublist.Language.String(), sublist.Score, len(sublist.Data))
  71. }
  72. }