From 2cd052f8891fdb0cc4782d1d0576e3f38a72492a Mon Sep 17 00:00:00 2001
From: O01eg <o01eg@yandex.ru>
Date: Wed, 29 Sep 2021 10:38:55 +0300
Subject: [PATCH] Add tests for Translations

---
 .gitignore               |  3 +++
 tests/test_translation.h | 31 +++++++++++++++++++++++++++++++
 2 files changed, 34 insertions(+)

diff --git a/.gitignore b/.gitignore
index 00d87b6bf61..5b3414fe7e3 100644
--- a/.gitignore
+++ b/.gitignore
@@ -390,3 +390,6 @@ gcov.css
 # https://clangd.llvm.org/ cache folder
 .clangd/
 .cache/
+
+# Generated by unit tests files
+tests/data/*.translation
diff --git a/tests/test_translation.h b/tests/test_translation.h
index 52ff49bf9b9..93c53bbbc95 100644
--- a/tests/test_translation.h
+++ b/tests/test_translation.h
@@ -35,6 +35,10 @@
 #include "core/string/translation.h"
 #include "core/string/translation_po.h"
 
+#ifdef TOOLS_ENABLED
+#include "editor/import/resource_importer_csv_translation.h"
+#endif
+
 #include "thirdparty/doctest/doctest.h"
 
 namespace TestTranslation {
@@ -145,6 +149,33 @@ TEST_CASE("[OptimizedTranslation] Generate from Translation and read messages")
 	CHECK(messages.size() == 0);
 }
 
+#ifdef TOOLS_ENABLED
+TEST_CASE("[Translation] CSV import") {
+	Ref<ResourceImporterCSVTranslation> import_csv_translation = memnew(ResourceImporterCSVTranslation);
+
+	Map<StringName, Variant> options;
+	options["compress"] = false;
+	options["delimiter"] = 0;
+
+	List<String> gen_files;
+
+	Error result = import_csv_translation->import(TestUtils::get_data_path("translations.csv"),
+			"", options, nullptr, &gen_files);
+	CHECK(result == OK);
+	CHECK(gen_files.size() == 2);
+
+	for (const String &file : gen_files) {
+		Ref<Translation> translation = ResourceLoader::load(file);
+		CHECK(translation.is_valid());
+		TranslationServer::get_singleton()->add_translation(translation);
+	}
+
+	TranslationServer::get_singleton()->set_locale("de");
+
+	CHECK(Object().tr("GOOD_MORNING", "") == "Guten Morgen");
+}
+#endif // TOOLS_ENABLED
+
 } // namespace TestTranslation
 
 #endif // TEST_TRANSLATION_H