Absorb go-xgettext/main_test.go

1 files changed, 509 insertions, 0 deletions

diff --git a/tests/gotext.go b/tests/gotext.go
index 36d1327..e679bda 100644
--- a/tests/gotext.go
+++ b/tests/gotext.go
@@ -1,4 +1,513 @@
 package gotext
 
+import (
+	"bytes"
+	"fmt"
+	"os"
+
+	g "gobang"
+)
+
+
+
+func src(payload string) string {
+	f := g.Must(os.CreateTemp("", "gotext-temp-*.go"))
+	g.Must(f.WriteString(payload))
+	g.TErrorIf(f.Close())
+	return f.Name()
+}
+
+
+
+func test_formatComment() {
+	g.TestStart("formatComment()")
+
+	g.Testing("formatting comments FIXME", func() {
+		table := []struct{
+			in  string
+			out string
+		}{
+			{"// foo ",          "#. foo\n"},
+			{"/* foo */",        "#. foo\n"},
+			{"/* foo\n */",      "#. foo\n"},
+			{"/* foo\nbar   */", "#. foo\n#. bar\n"},
+		}
+
+		for _, row := range table {
+			g.TAssertEqual(formatComment(row.in), row.out)
+		}
+	})
+}
+
+func test_processFiles() {
+	g.TestStart("processFiles()")
+
+	g.Testing("simple commented file", func() {
+		fname := src(g.Heredoc(`
+			package main
+
+			func main() {
+				// TRANSLATORS: a comment
+				i18n.G("foo")
+			}
+		`))
+		defer os.Remove(fname)
+		g.TErrorIf(processFiles([]string{fname}))
+
+		g.TAssertEqual(msgIDs, map[string][]msgID{
+			"foo": []msgID{
+				msgID{
+					comment: "#. TRANSLATORS: a comment\n",
+					fname:   fname,
+					line:    5,
+				},
+			},
+		})
+	})
+
+	g.Testing("process multiple entries", func() {
+		fname := src(g.Heredoc(`
+			package main
+
+			func main() {
+				// TRANSLATORS: comment 1
+				i18n.G("foo")
+
+				// TRANSLATORS: comment 2
+				i18n.G("foo")
+			}
+		`))
+		defer os.Remove(fname)
+		g.TErrorIf(processFiles([]string{fname}))
+
+		g.TAssertEqual(msgIDs, map[string][]msgID{
+			"foo": []msgID{
+				{
+					comment: "#. TRANSLATORS: comment 1\n",
+					fname:   fname,
+					line:    5,
+				},
+				{
+					comment: "#. TRANSLATORS: comment 2\n",
+					fname:   fname,
+					line:    8,
+				},
+			},
+		})
+	})
+
+	g.Testing("process files with concat", func() {
+		msgIDs = nil
+		opts.NoLocation = true
+		fname := src(g.Heredoc(`
+			package main
+
+			func main() {
+				// TRANSLATORS: a comment
+				i18n.G("foo\n" + "bar\n" + "baz")
+			}
+		`))
+		g.TErrorIf(processFiles([]string{fname}))
+		opts.NoLocation = false
+
+		g.TAssertEqual(msgIDs, map[string][]msgID{
+			"foo\\nbar\\nbaz": []msgID{
+				{
+					comment: "#. TRANSLATORS: a comment\n",
+					fname:   fname,
+					line:    5,
+				},
+			},
+		})
+	})
+
+	g.Testing("file with quote", func() {
+		fname := src(fmt.Sprintf(g.Heredoc(`
+			package main
+
+			func main() {
+				i18n.G(%[1]s foo "bar"%[1]s)
+			}
+		`), "`"))
+		defer os.Remove(fname)
+		g.TErrorIf(processFiles([]string{fname}))
+
+		out := bytes.NewBuffer([]byte(""))
+		writePotFile(out)
+
+		expected := fmt.Sprintf(g.Heredoc(`
+			#, fuzzy
+			msgid ""
+			msgstr ""
+			"Language: \n"
+			"MIME-Version: 1.0\n"
+			"Content-Type: text/plain; charset=UTF-8\n"
+			"Content-Transfer-Encoding: 8bit\n"
+
+			#: %s:4
+			msgid " foo \"bar\""
+			msgstr ""
+
+		`), fname)
+		g.TAssertEqual(out.String(), expected)
+	})
+}
+
+func test_writePotFile() {
+	g.TestStart("writePotFile()")
+
+	g.Testing("write simple file", func() {
+		msgIDs = map[string][]msgID{
+			"foo": []msgID{
+				{
+					fname:   "fname",
+					line:    2,
+					comment: "#. foo\n",
+				},
+			},
+		}
+		out := bytes.NewBuffer([]byte(""))
+		writePotFile(out)
+
+		expected := g.Heredoc(`
+			#, fuzzy
+			msgid ""
+			msgstr ""
+			"Language: \n"
+			"MIME-Version: 1.0\n"
+			"Content-Type: text/plain; charset=UTF-8\n"
+			"Content-Transfer-Encoding: 8bit\n"
+
+			#. foo
+			#: fname:2
+			msgid "foo"
+			msgstr ""
+
+		`)
+		g.TAssertEqual(out.String(), expected)
+	})
+
+	g.Testing("write template with 2 entries", func() {
+		msgIDs = map[string][]msgID{
+			"foo": []msgID{
+				{
+					fname:   "fname",
+					line:    2,
+					comment: "#. comment1\n",
+				},
+				{
+					fname:   "fname",
+					line:    4,
+					comment: "#. comment2\n",
+				},
+			},
+		}
+		out := bytes.NewBuffer([]byte(""))
+		writePotFile(out)
+
+		expected := g.Heredoc(`
+			#, fuzzy
+			msgid ""
+			msgstr ""
+			"Language: \n"
+			"MIME-Version: 1.0\n"
+			"Content-Type: text/plain; charset=UTF-8\n"
+			"Content-Transfer-Encoding: 8bit\n"
+
+			#. comment1
+			#. comment2
+			#: fname:2 fname:4
+			msgid "foo"
+			msgstr ""
+
+		`)
+		g.TAssertEqual(out.String(), expected)
+	})
+
+	g.Testing("template without comment", func() {
+		msgIDs = map[string][]msgID{
+			"foo": []msgID{
+				{
+					fname: "fname",
+					line:  2,
+				},
+			},
+		}
+		out := bytes.NewBuffer([]byte(""))
+		writePotFile(out)
+
+		expected := g.Heredoc(`
+			#, fuzzy
+			msgid ""
+			msgstr ""
+			"Language: \n"
+			"MIME-Version: 1.0\n"
+			"Content-Type: text/plain; charset=UTF-8\n"
+			"Content-Transfer-Encoding: 8bit\n"
+
+			#: fname:2
+			msgid "foo"
+			msgstr ""
+
+		`)
+		g.TAssertEqual(out.String(), expected)
+	})
+
+	g.Testing("output without location", func() {
+		msgIDs = map[string][]msgID{
+			"foo": []msgID{
+				{
+					fname: "fname",
+					line:  2,
+				},
+			},
+		}
+
+		opts.NoLocation = true
+		out := bytes.NewBuffer([]byte(""))
+		writePotFile(out)
+		opts.NoLocation = false
+
+		expected := g.Heredoc(`
+			#, fuzzy
+			msgid ""
+			msgstr ""
+			"Language: \n"
+			"MIME-Version: 1.0\n"
+			"Content-Type: text/plain; charset=UTF-8\n"
+			"Content-Transfer-Encoding: 8bit\n"
+
+			msgid "foo"
+			msgstr ""
+
+		`)
+		g.TAssertEqual(out.String(), expected)
+	})
+
+	g.Testing("output with hint", func() {
+		msgIDs = map[string][]msgID{
+			"foo": []msgID{
+				{
+					fname:      "fname",
+					line:       2,
+					formatHint: "c-format",
+				},
+			},
+		}
+
+		out := bytes.NewBuffer([]byte(""))
+		writePotFile(out)
+
+		expected := g.Heredoc(`
+			#, fuzzy
+			msgid ""
+			msgstr ""
+			"Language: \n"
+			"MIME-Version: 1.0\n"
+			"Content-Type: text/plain; charset=UTF-8\n"
+			"Content-Transfer-Encoding: 8bit\n"
+
+			#: fname:2
+			#, c-format
+			msgid "foo"
+			msgstr ""
+
+		`)
+		g.TAssertEqual(out.String(), expected)
+	})
+
+	g.Testing("output with plural", func() {
+		msgIDs = map[string][]msgID{
+			"foo": []msgID{
+				{
+					msgidPlural: "plural",
+					fname:       "fname",
+					line:        2,
+				},
+			},
+		}
+
+		out := bytes.NewBuffer([]byte(""))
+		writePotFile(out)
+
+		expected := g.Heredoc(`
+			#, fuzzy
+			msgid ""
+			msgstr ""
+			"Language: \n"
+			"MIME-Version: 1.0\n"
+			"Content-Type: text/plain; charset=UTF-8\n"
+			"Content-Transfer-Encoding: 8bit\n"
+
+			#: fname:2
+			msgid "foo"
+			msgid_plural "plural"
+			msgstr[0] ""
+			msgstr[1] ""
+
+		`)
+		g.TAssertEqual(out.String(), expected)
+	})
+
+	g.Testing("sorted output", func() {
+		msgIDs = map[string][]msgID{
+			"aaa": []msgID{
+				{
+					fname: "fname",
+					line:  2,
+				},
+			},
+			"zzz": []msgID{
+				{
+					fname: "fname",
+					line:  2,
+				},
+			},
+		}
+
+		opts.SortOutput = true
+		// we need to run this a bunch of times as the ordering might
+		// be right by pure chance
+		// FIXME
+		for i := 0; i < 10; i++ {
+			out := bytes.NewBuffer([]byte(""))
+			writePotFile(out)
+
+			expected := g.Heredoc(`
+				#, fuzzy
+				msgid ""
+				msgstr ""
+				"Language: \n"
+				"MIME-Version: 1.0\n"
+				"Content-Type: text/plain; charset=UTF-8\n"
+				"Content-Transfer-Encoding: 8bit\n"
+
+				#: fname:2
+				msgid "aaa"
+				msgstr ""
+
+				#: fname:2
+				msgid "zzz"
+				msgstr ""
+
+			`)
+			g.TAssertEqual(out.String(), expected)
+		}
+		opts.SortOutput = false
+	})
+
+	g.Testing("multiline output", func() {
+		msgIDs = map[string][]msgID{
+			"foo\\nbar\\nbaz": []msgID{
+				{
+					fname:   "fname",
+					line:    2,
+					comment: "#. foo\n",
+				},
+			},
+		}
+		out := bytes.NewBuffer([]byte(""))
+		writePotFile(out)
+		expected := g.Heredoc(`
+			#, fuzzy
+			msgid ""
+			msgstr ""
+			"Language: \n"
+			"MIME-Version: 1.0\n"
+			"Content-Type: text/plain; charset=UTF-8\n"
+			"Content-Transfer-Encoding: 8bit\n"
+
+			#. foo
+			#: fname:2
+			msgid ""
+			"foo\n"
+			"bar\n"
+			"baz"
+			msgstr ""
+
+		`)
+		g.TAssertEqual(out.String(), expected)
+	})
+
+	g.Testing("output tidy output", func() {
+		msgIDs = map[string][]msgID{
+			"foo\\nbar\\nbaz": []msgID{
+				{
+					fname: "fname",
+					line:  2,
+				},
+			},
+			"zzz\\n": []msgID{
+				{
+					fname: "fname",
+					line:  4,
+				},
+			},
+		}
+		out := bytes.NewBuffer([]byte(""))
+		writePotFile(out)
+		expected := g.Heredoc(`
+			#, fuzzy
+			msgid ""
+			msgstr ""
+			"Language: \n"
+			"MIME-Version: 1.0\n"
+			"Content-Type: text/plain; charset=UTF-8\n"
+			"Content-Transfer-Encoding: 8bit\n"
+
+			#: fname:2
+			msgid ""
+			"foo\n"
+			"bar\n"
+			"baz"
+			msgstr ""
+
+			#: fname:4
+			msgid "zzz\n"
+			msgstr ""
+
+		`)
+		g.TAssertEqual(out.String(), expected)
+	})
+
+	g.Testing("source with double quotes", func() {
+		fname := src(g.Heredoc(`
+			package main
+
+			func main() {
+				i18n.G("foo \"bar\"")
+			}
+		`))
+		defer os.Remove(fname)
+		g.TErrorIf(processFiles([]string{fname}))
+
+		out := bytes.NewBuffer([]byte(""))
+		writePotFile(out)
+
+		expected := fmt.Sprintf(g.Heredoc(`
+			#, fuzzy
+			msgid ""
+			msgstr ""
+			"Language: \n"
+			"MIME-Version: 1.0\n"
+			"Content-Type: text/plain; charset=UTF-8\n"
+			"Content-Transfer-Encoding: 8bit\n"
+
+			#: %[1]s:4
+			msgid "foo \"bar\""
+			msgstr ""
+
+		`), fname)
+		g.TAssertEqual(out.String(), expected)
+	})
+}
+
+
+
 func MainTest() {
+	g.Init()
+
+	test_formatComment()
+	test_processFiles()
+	test_writePotFile()
 }