srt_test.go

package subtitles

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNewFromSRT(t *testing.T) {
	in := "1\n" +
		"00:00:04,630 --> 00:00:06,018\n" +
		"Go ninja!\n" +
		"\n" +
		"2\n" +
		"00:00:10,000 --> 00:00:11,000\n" +
		"Subtitles By MrCool\n" +
		"\n" +
		"\n" + // NOTE make sure we allow slightly sloppy input
		"3\n" +
		"00:01:09,630 --> 00:01:11,005\n" +
		"No ninja!\n"

	expected := Subtitle{[]Caption{{
		1,
		makeTime(0, 0, 4, 630),
		makeTime(0, 0, 6, 18),
		[]string{"Go ninja!"},
	}, {
		2,
		makeTime(0, 0, 10, 0),
		makeTime(0, 0, 11, 0),
		[]string{"Subtitles By MrCool"},
	}, {
		3,
		makeTime(0, 1, 9, 630),
		makeTime(0, 1, 11, 005),
		[]string{"No ninja!"},
	}}}

	res, err := NewFromSRT(in)
	assert.Equal(t, nil, err)
	assert.Equal(t, expected, res)
}

func TestNewFromSRTWithMacLinebreaks(t *testing.T) {
	in := "1\r" +
		"00:00:04,630 --> 00:00:06,018\r" +
		"Go ninja!\r" +
		"\r" +
		"3\r" +
		"00:01:09,630 --> 00:01:11,005\r" +
		"No ninja!\r"

	expected := Subtitle{[]Caption{{
		1,
		makeTime(0, 0, 4, 630),
		makeTime(0, 0, 6, 18),
		[]string{"Go ninja!"},
	}, {
		2,
		makeTime(0, 1, 9, 630),
		makeTime(0, 1, 11, 005),
		[]string{"No ninja!"},
	}}}

	utf8 := ConvertToUTF8([]byte(in))

	res, err := NewFromSRT(utf8)
	assert.Equal(t, nil, err)
	assert.Equal(t, expected, res)
}

func TestNewFromSRTSkipEmpty(t *testing.T) {
	in := "1\n" +
		"00:00:04,630 --> 00:00:06,018\n" +
		"Go ninja!\n" +
		"\n" +
		"2\n" +
		"00:00:10,000 --> 00:00:11,000\n" +
		"\n" +
		"\n" +
		"3\n" +
		"00:01:09,630 --> 00:01:11,005\n" +
		"No ninja!\n"

	expected := Subtitle{[]Caption{{
		1,
		makeTime(0, 0, 4, 630),
		makeTime(0, 0, 6, 18),
		[]string{"Go ninja!"},
	}, {
		2,
		makeTime(0, 1, 9, 630),
		makeTime(0, 1, 11, 005),
		[]string{"No ninja!"},
	}}}

	res, err := NewFromSRT(in)
	assert.Equal(t, nil, err)
	assert.Equal(t, expected, res)
}

func TestNewFromSRTCrlf(t *testing.T) {
	in := "1\n" +
		"00:00:04,630 --> 00:00:06,018\r\n" +
		"Go ninja!\r\n" +
		"\r\n"

	expected := Subtitle{[]Caption{{
		1,
		makeTime(0, 0, 4, 630),
		makeTime(0, 0, 6, 18),
		[]string{"Go ninja!"},
	}}}

	res, err := NewFromSRT(in)
	assert.Equal(t, nil, err)
	assert.Equal(t, expected, res)
}

func TestParseExtraLineBreak(t *testing.T) {
	in := "1\n" +
		"00:00:04,630 --> 00:00:06,018\r\n" +
		// NOTE: should not be line break here, but some files has,
		// so lets make sure we handle it
		"\r\n" +
		"Go ninja!\r\n" +
		"\r\n"

	expected := Subtitle{[]Caption{{
		1,
		makeTime(0, 0, 4, 630),
		makeTime(0, 0, 6, 18),
		[]string{"Go ninja!"},
	}}}

	res, err := NewFromSRT(in)
	assert.Equal(t, nil, err)
	assert.Equal(t, expected, res)
}

func TestParseWeirdTimestamp(t *testing.T) {
	in := "1\r\n" +
		"00:14:52.00 --> 00:14:57,500\r\n" +
		"Go ninja!\r\n"
	expected := Subtitle{[]Caption{{
		1,
		makeTime(0, 14, 52, 0),
		makeTime(0, 14, 57, 500),
		[]string{"Go ninja!"},
	}}}

	res, err := NewFromSRT(in)
	assert.Equal(t, nil, err)
	assert.Equal(t, expected, res)
}

func TestParseWeirdTrailingTabs(t *testing.T) {
	in := "1\r\n" +
		"00:14:52.00 --> 00:14:57,500\r\n" +
		"Go ninja!\r\n" +
		"\r\n" +
		"										\r\n" +
		"\r\n"
	expected := Subtitle{[]Caption{{
		1,
		makeTime(0, 14, 52, 0),
		makeTime(0, 14, 57, 500),
		[]string{"Go ninja!"},
	}}}

	res, err := NewFromSRT(in)
	assert.Equal(t, nil, err)
	assert.Equal(t, expected, res)
}

func TestAsSRT(t *testing.T) {
	in := Subtitle{[]Caption{{
		1,
		makeTime(0, 0, 4, 630),
		makeTime(0, 0, 6, 18),
		[]string{"Go ninja!"},
	}, {
		2,
		makeTime(0, 1, 9, 630),
		makeTime(0, 1, 11, 005),
		[]string{"No ninja!"},
	}}}
	expected := "1\n" +
		"00:00:04,630 --> 00:00:06,018\n" +
		"Go ninja!\n" +
		"\n" +
		"2\n" +
		"00:01:09,630 --> 00:01:11,005\n" +
		"No ninja!\n\n"
	assert.Equal(t, expected, in.AsSRT())
}

func TestParseLatin1Srt(t *testing.T) {
	in := "1\r\n" +
		"00:14:52.00 --> 00:14:57,500\r\n" +
		"Hall\xe5 ninja!\r\n"

	expected := Subtitle{[]Caption{{
		1,
		makeTime(0, 14, 52, 0),
		makeTime(0, 14, 57, 500),
		[]string{"Hallå ninja!"},
	}}}

	utf8 := ConvertToUTF8([]byte(in))

	res, err := NewFromSRT(utf8)
	assert.Equal(t, nil, err)
	assert.Equal(t, expected, res)
}

func TestParseUTF16BESrt(t *testing.T) {
	in := []byte{
		0xfe, 0xff, // UTF16 BE BOM
		0, '1',
		0, '\r', 0, '\n',
		0, '0', 0, '0', 0, ':', 0, '0', 0, '0', 0, ':',
		0, '0', 0, '0', 0, ',', 0, '0', 0, '0', 0, '0',
		0, ' ', 0, '-', 0, '-', 0, '>', 0, ' ',
		0, '0', 0, '0', 0, ':', 0, '0', 0, '0', 0, ':',
		0, '0', 0, '0', 0, ',', 0, '0', 0, '0', 0, '1',
		0, '\r', 0, '\n',
		0, 'T', 0, 'e', 0, 's', 0, 't',
		0, '\r', 0, '\n',
		0, '\r', 0, '\n',
	}
	expected := Subtitle{[]Caption{{
		1,
		makeTime(0, 0, 0, 0),
		makeTime(0, 0, 0, 1),
		[]string{"Test"},
	}}}

	utf8 := ConvertToUTF8(in)

	res, err := NewFromSRT(utf8)
	assert.Equal(t, nil, err)
	assert.Equal(t, expected, res)
}

func TestParseUTF16LESrt(t *testing.T) {
	in := []byte{
		0xff, 0xfe, // UTF16 LE BOM
		'1', 0,
		'\r', 0, '\n', 0,
		'0', 0, '0', 0, ':', 0, '0', 0, '0', 0, ':', 0,
		'0', 0, '0', 0, ',', 0, '0', 0, '0', 0, '0', 0,
		' ', 0, '-', 0, '-', 0, '>', 0, ' ', 0,
		'0', 0, '0', 0, ':', 0, '0', 0, '0', 0, ':', 0,
		'0', 0, '0', 0, ',', 0, '0', 0, '0', 0, '1', 0,
		'\r', 0, '\n', 0,
		'T', 0, 'e', 0, 's', 0, 't', 0,
		'\r', 0, '\n', 0,
		'\r', 0, '\n', 0,
	}
	expected := Subtitle{[]Caption{{
		1,
		makeTime(0, 0, 0, 0),
		makeTime(0, 0, 0, 1),
		[]string{"Test"},
	}}}

	utf8 := ConvertToUTF8(in)

	res, err := NewFromSRT(utf8)
	assert.Equal(t, nil, err)
	assert.Equal(t, expected, res)
}

func TestParseUTF8BomSrt(t *testing.T) {
	in := []byte{
		0xef, 0xbb, 0xbf, // UTF8 BOM
		'1',
		'\r', '\n',
		'0', '0', ':', '0', '0', ':',
		'0', '0', ',', '0', '0', '0',
		' ', '-', '-', '>', ' ',
		'0', '0', ':', '0', '0', ':',
		'0', '0', ',', '0', '0', '1',
		'\r', '\n',
		'T', 'e', 's', 't',
		'\r', '\n',
		'\r', '\n',
	}

	expected := Subtitle{[]Caption{{
		1,
		makeTime(0, 0, 0, 0),
		makeTime(0, 0, 0, 1),
		[]string{"Test"},
	}}}

	utf8 := ConvertToUTF8(in)

	res, err := NewFromSRT(utf8)
	assert.Equal(t, nil, err)
	assert.Equal(t, expected, res)
}