ChineseSubFinder/internal/pkg/sub_helper/dialogue_merger_test.go

package sub_helper

import (
	"github.com/allanpk716/ChineseSubFinder/internal/logic/sub_parser/ass"
	"github.com/allanpk716/ChineseSubFinder/internal/logic/sub_parser/srt"
	"github.com/allanpk716/ChineseSubFinder/internal/pkg"
	"github.com/allanpk716/ChineseSubFinder/internal/pkg/sub_parser_hub"
	"path"
	"testing"
)

func Test_isFirstLetterIsEngUpper(t *testing.T) {
	type args struct {
		instring string
	}
	tests := []struct {
		name string
		args args
		want bool
	}{
		{name: "0", args: args{instring: "A"}, want: true},
		{name: "1", args: args{instring: "a"}, want: false},
		{name: "2", args: args{instring: "哈"}, want: false},
		{name: "3", args: args{instring: ""}, want: false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := isFirstLetterIsEngUpper(tt.args.instring); got != tt.want {
				t.Errorf("isFirstLetterIsEngUpper() = %v, want %v", got, tt.want)
			}
		})
	}
}

func Test_isFirstLetterIsEngLower(t *testing.T) {
	type args struct {
		instring string
	}
	tests := []struct {
		name string
		args args
		want bool
	}{
		{name: "0", args: args{instring: "A"}, want: false},
		{name: "1", args: args{instring: "a"}, want: true},
		{name: "2", args: args{instring: "哈"}, want: false},
		{name: "3", args: args{instring: ""}, want: false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := isFirstLetterIsEngLower(tt.args.instring); got != tt.want {
				t.Errorf("isFirstLetterIsEngLower() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestNewDialogueMerger(t *testing.T) {

	testDataPath := "../../../TestData/FixTimeline"
	testRootDir, err := pkg.CopyTestData(testDataPath)
	if err != nil {
		t.Fatal(err)
	}

	subParserHub := sub_parser_hub.NewSubParserHub(ass.NewParser(), srt.NewParser())
	bFind, infoBase, err := subParserHub.DetermineFileTypeFromFile(filepath.Join(testRootDir, "2line-The Card Counter (2021) WEBDL-1080p.chinese(inside).ass"))
	if err != nil {
		t.Fatal(err)
	}
	if bFind == false {
		t.Fatal("not find")
	}

	merger := NewDialogueMerger()
	for _, ex := range infoBase.DialoguesEx {
		merger.Add(ex)
	}
	newEx := merger.Get()
	println(len(newEx))
}