From c41bb168232eea3c9d37c4b7841794ecec4c70f2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kamil=20Krzy=C5=9Bk=C3=B3w?= <kamilzary@gmail.com>
Date: Fri, 20 Oct 2023 19:04:29 +0100
Subject: [PATCH] test(general): Add tests for build_only_locale

---
 tests/test_languages_option.py | 65 +++++++++++++++++++++++++++++++++-
 1 file changed, 64 insertions(+), 1 deletion(-)

diff --git a/tests/test_languages_option.py b/tests/test_languages_option.py
index 59c7ff0..e572695 100644
--- a/tests/test_languages_option.py
+++ b/tests/test_languages_option.py
@@ -244,4 +244,67 @@ def test_plugin_languages_dual_lang_with_null():
             "site_description": None,
             "site_url": None,
         },
-    ]
\ No newline at end of file
+    ]
+
+
+def test_plugin_build_only_locale():
+    mkdocs_config = load_config(
+        "tests/mkdocs.yml",
+        theme={"name": "material"},
+        docs_dir="docs_suffix_structure_two_languages/",
+        plugins={
+            "i18n": {
+                "build_only_locale": "fr",
+                "languages": [
+                    {
+                        "locale": "en",
+                        "name": "english",
+                        "default": True,
+                        "build": True,
+                    },
+                    {
+                        "locale": "fr", 
+                        "name": "français",
+                        "default": False,
+                        "build": False, 
+                    },
+                ],
+            }
+        },
+    )
+    
+    english = mkdocs_config["plugins"]["i18n"].config.languages[0]
+    french = mkdocs_config["plugins"]["i18n"].config.languages[1]
+
+    assert english["default"] == False
+    assert english["build"] == False
+    assert french["default"] == True
+    assert french["build"] == True
+
+
+def test_plugin_build_only_locale_abort():
+    with pytest.raises(Abort):
+        load_config(
+            "tests/mkdocs.yml",
+            theme={"name": "material"},
+            docs_dir="docs_suffix_structure_two_languages/",
+            plugins={
+                "i18n": {
+                    "build_only_locale": "zh",
+                    "languages": [
+                        {
+                            "locale": "en",
+                            "name": "english",
+                            "default": True,
+                            "build": True,
+                        },
+                        {
+                            "locale": "fr", 
+                            "name": "français",
+                            "default": False,
+                            "build": False, 
+                        },
+                    ],
+                }
+            },
+        )