-
Notifications
You must be signed in to change notification settings - Fork 124
Expand file tree
/
Copy pathttml_test.go
More file actions
104 lines (88 loc) · 6.56 KB
/
ttml_test.go
File metadata and controls
104 lines (88 loc) · 6.56 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package astisub_test
import (
"bytes"
"io/ioutil"
"strings"
"testing"
"github.com/asticode/go-astikit"
"github.com/asticode/go-astisub"
"github.com/stretchr/testify/assert"
)
func TestTTML(t *testing.T) {
// Open
s, err := astisub.OpenFile("./testdata/example-in.ttml")
assert.NoError(t, err)
assertSubtitleItems(t, s)
// Metadata
assert.Equal(t, &astisub.Metadata{Framerate: 25, Language: astisub.LanguageFrench, Title: "Title test", TTMLCopyright: "Copyright test"}, s.Metadata)
// Styles
assert.Equal(t, 3, len(s.Styles))
assert.Equal(t, astisub.Style{ID: "style_0", InlineStyle: &astisub.StyleAttributes{STLColor: astisub.ColorWhite, TTMLColor: astisub.ColorWhite, TTMLExtent: astikit.StrPtr("100% 10%"), TTMLFontFamily: astikit.StrPtr("sansSerif"), TTMLFontStyle: astikit.StrPtr("normal"), TTMLOrigin: astikit.StrPtr("0% 90%"), TTMLTextAlign: astikit.StrPtr("center"), WebVTTAlign: "center", WebVTTLine: "0%", WebVTTLines: 2, WebVTTPosition: &astisub.WebVTTPosition{XPosition: "90%"}, WebVTTRegionAnchor: "0%,0%", WebVTTScroll: "up", WebVTTSize: "10%", WebVTTViewportAnchor: "0%,90%", WebVTTWidth: "100%"}, Style: s.Styles["style_2"]}, *s.Styles["style_0"])
assert.Equal(t, astisub.Style{ID: "style_1", InlineStyle: &astisub.StyleAttributes{STLColor: astisub.ColorWhite, TTMLColor: astisub.ColorWhite, TTMLExtent: astikit.StrPtr("100% 13%"), TTMLFontFamily: astikit.StrPtr("sansSerif"), TTMLFontStyle: astikit.StrPtr("normal"), TTMLOrigin: astikit.StrPtr("0% 87%"), TTMLTextAlign: astikit.StrPtr("center"), WebVTTAlign: "center", WebVTTLine: "0%", WebVTTLines: 2, WebVTTPosition: &astisub.WebVTTPosition{XPosition: "87%"}, WebVTTRegionAnchor: "0%,0%", WebVTTScroll: "up", WebVTTSize: "13%", WebVTTViewportAnchor: "0%,87%", WebVTTWidth: "100%"}}, *s.Styles["style_1"])
assert.Equal(t, astisub.Style{ID: "style_2", InlineStyle: &astisub.StyleAttributes{STLColor: astisub.ColorWhite, TTMLColor: astisub.ColorWhite, TTMLExtent: astikit.StrPtr("100% 20%"), TTMLFontFamily: astikit.StrPtr("sansSerif"), TTMLFontStyle: astikit.StrPtr("normal"), TTMLOrigin: astikit.StrPtr("0% 80%"), TTMLTextAlign: astikit.StrPtr("center"), WebVTTAlign: "center", WebVTTLine: "0%", WebVTTLines: 4, WebVTTPosition: &astisub.WebVTTPosition{XPosition: "80%"}, WebVTTRegionAnchor: "0%,0%", WebVTTScroll: "up", WebVTTSize: "20%", WebVTTViewportAnchor: "0%,80%", WebVTTWidth: "100%"}}, *s.Styles["style_2"])
// Regions
assert.Equal(t, 3, len(s.Regions))
assert.Equal(t, astisub.Region{ID: "region_0", Style: s.Styles["style_0"], InlineStyle: &astisub.StyleAttributes{STLColor: astisub.ColorBlue, TTMLColor: astisub.ColorBlue}}, *s.Regions["region_0"])
assert.Equal(t, astisub.Region{ID: "region_1", Style: s.Styles["style_1"], InlineStyle: &astisub.StyleAttributes{}}, *s.Regions["region_1"])
assert.Equal(t, astisub.Region{ID: "region_2", Style: s.Styles["style_2"], InlineStyle: &astisub.StyleAttributes{}}, *s.Regions["region_2"])
// Items
assert.Equal(t, s.Regions["region_1"], s.Items[0].Region)
assert.Equal(t, s.Styles["style_1"], s.Items[0].Style)
assert.Equal(t, &astisub.StyleAttributes{STLColor: astisub.ColorRed, TTMLColor: astisub.ColorRed}, s.Items[0].InlineStyle)
assert.Equal(t, []astisub.Line{{Items: []astisub.LineItem{{Style: s.Styles["style_1"], InlineStyle: &astisub.StyleAttributes{STLColor: astisub.ColorBlack, TTMLColor: astisub.ColorBlack}, Text: "(deep rumbling)"}}}}, s.Items[0].Lines)
assert.Equal(t, []astisub.Line{{Items: []astisub.LineItem{{InlineStyle: &astisub.StyleAttributes{}, Text: "MAN:"}}}, {Items: []astisub.LineItem{{InlineStyle: &astisub.StyleAttributes{}, Text: "How did we "}, {InlineStyle: &astisub.StyleAttributes{STLColor: astisub.ColorGreen, TTMLColor: astisub.ColorGreen}, Style: s.Styles["style_1"], Text: "end up"}, {InlineStyle: &astisub.StyleAttributes{}, Text: " here?"}}}}, s.Items[1].Lines)
assert.Equal(t, []astisub.Line{{Items: []astisub.LineItem{{InlineStyle: &astisub.StyleAttributes{}, Style: s.Styles["style_1"], Text: "This place is horrible."}}}}, s.Items[2].Lines)
assert.Equal(t, []astisub.Line{{Items: []astisub.LineItem{{InlineStyle: &astisub.StyleAttributes{}, Style: s.Styles["style_1"], Text: "Smells like balls."}}}}, s.Items[3].Lines)
assert.Equal(t, []astisub.Line{{Items: []astisub.LineItem{{InlineStyle: &astisub.StyleAttributes{}, Style: s.Styles["style_2"], Text: "We don't belong"}}}, {Items: []astisub.LineItem{{InlineStyle: &astisub.StyleAttributes{}, Style: s.Styles["style_1"], Text: "in this shithole."}}}}, s.Items[4].Lines)
assert.Equal(t, []astisub.Line{{Items: []astisub.LineItem{{InlineStyle: &astisub.StyleAttributes{}, Style: s.Styles["style_2"], Text: "(computer playing"}}}, {Items: []astisub.LineItem{{InlineStyle: &astisub.StyleAttributes{}, Style: s.Styles["style_1"], Text: "electronic melody)"}}}}, s.Items[5].Lines)
// No subtitles to write
w := &bytes.Buffer{}
err = astisub.Subtitles{}.WriteToTTML(w)
assert.EqualError(t, err, astisub.ErrNoSubtitlesToWrite.Error())
// Write
c, err := ioutil.ReadFile("./testdata/example-out.ttml")
assert.NoError(t, err)
err = s.WriteToTTML(w)
assert.NoError(t, err)
assert.Equal(t, string(c), w.String())
}
func TestTTMLBreakLines(t *testing.T) {
// Open
s, err := astisub.OpenFile("./testdata/example-in-breaklines.ttml")
assert.NoError(t, err)
// Write
w := &bytes.Buffer{}
err = s.WriteToTTML(w)
assert.NoError(t, err)
c, err := ioutil.ReadFile("./testdata/example-out-breaklines.ttml")
assert.NoError(t, err)
assert.Equal(t, strings.TrimSpace(string(c)), strings.TrimSpace(w.String()))
}
func TestWriteToTTMLWithIndentOption(t *testing.T) {
// Open
s, err := astisub.OpenFile("./testdata/example-in.ttml")
assert.NoError(t, err)
// Write
w := &bytes.Buffer{}
err = s.WriteToTTML(w, astisub.WriteToTTMLWithIndentOption(""))
assert.NoError(t, err)
c, err := ioutil.ReadFile("./testdata/example-out-no-indent.ttml")
assert.NoError(t, err)
assert.Equal(t, strings.TrimSpace(string(c)), strings.TrimSpace(w.String()))
}
func TestTTMLMergeStyleAttributes(t *testing.T) {
// Open
s, err := astisub.OpenFile("./testdata/example-in-merging-style.ttml")
assert.NoError(t, err)
// Styles
assert.Equal(t, 4, len(s.Items))
assert.Equal(t, s.Items[0].Region.ID, "region_0")
assert.Equal(t, s.Items[0].Style.ID, "style_1")
assert.Equal(t, s.Items[0].InlineStyle.TTMLColor, astisub.ColorRed)
assert.Equal(t, s.Items[1].Region.ID, "region_1")
assert.Equal(t, s.Items[1].Style.ID, "style_0")
assert.Equal(t, s.Items[1].Lines[0].Items[0].Style.ID, "style_1")
assert.Equal(t, s.Items[2].Region.ID, "region_2")
assert.Equal(t, s.Items[2].Style.ID, "style_0")
assert.Equal(t, s.Items[3].InlineStyle.TTMLColor, astisub.ColorBlue)
}