From addd375794d2f909aa3824e5ceb46b75c1543256 Mon Sep 17 00:00:00 2001
From: TavoNiievez <ganieves@outlook.com>
Date: Sat, 18 Jan 2025 22:25:38 -0500
Subject: [PATCH] Added Symfony Translation assertions

---
 composer.json                                 |   1 +
 src/Codeception/Module/Symfony.php            |   2 +
 .../Symfony/TranslationAssertionsTrait.php    | 178 ++++++++++++++++++
 3 files changed, 181 insertions(+)
 create mode 100644 src/Codeception/Module/Symfony/TranslationAssertionsTrait.php

diff --git a/composer.json b/composer.json
index 82fb64f3..747a5941 100644
--- a/composer.json
+++ b/composer.json
@@ -52,6 +52,7 @@
         "symfony/security-core": "^5.4 | ^6.4 | ^7.2",
         "symfony/security-csrf": "^5.4 | ^6.4 | ^7.2",
         "symfony/security-http": "^5.4 | ^6.4 | ^7.2",
+        "symfony/translation": "^5.4 | ^6.4 | ^7.2",
         "symfony/twig-bundle": "^5.4 | ^6.4 | ^7.2",
         "symfony/validator": "^5.4 | ^6.4 | ^7.2",
         "symfony/var-exporter": "^5.4 | ^6.4 | ^7.2",
diff --git a/src/Codeception/Module/Symfony.php b/src/Codeception/Module/Symfony.php
index 3f7917bb..ebe6aee6 100644
--- a/src/Codeception/Module/Symfony.php
+++ b/src/Codeception/Module/Symfony.php
@@ -25,6 +25,7 @@
 use Codeception\Module\Symfony\ServicesAssertionsTrait;
 use Codeception\Module\Symfony\SessionAssertionsTrait;
 use Codeception\Module\Symfony\TimeAssertionsTrait;
+use Codeception\Module\Symfony\TranslationAssertionsTrait;
 use Codeception\Module\Symfony\TwigAssertionsTrait;
 use Codeception\Module\Symfony\ValidatorAssertionsTrait;
 use Codeception\TestInterface;
@@ -148,6 +149,7 @@ class Symfony extends Framework implements DoctrineProvider, PartedModule
     use SecurityAssertionsTrait;
     use ServicesAssertionsTrait;
     use SessionAssertionsTrait;
+    use TranslationAssertionsTrait;
     use TimeAssertionsTrait;
     use TwigAssertionsTrait;
     use ValidatorAssertionsTrait;
diff --git a/src/Codeception/Module/Symfony/TranslationAssertionsTrait.php b/src/Codeception/Module/Symfony/TranslationAssertionsTrait.php
new file mode 100644
index 00000000..7c4c385a
--- /dev/null
+++ b/src/Codeception/Module/Symfony/TranslationAssertionsTrait.php
@@ -0,0 +1,178 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Codeception\Module\Symfony;
+
+use Symfony\Component\Translation\DataCollector\TranslationDataCollector;
+use Symfony\Component\VarDumper\Cloner\Data;
+
+trait TranslationAssertionsTrait
+{
+    /**
+     * Asserts that no fallback translations were found.
+     *
+     * ```php
+     * <?php
+     * $I->dontSeeFallbackTranslations();
+     * ```
+     */
+    public function dontSeeFallbackTranslations(): void
+    {
+        $translationCollector = $this->grabTranslationCollector(__FUNCTION__);
+        $fallbacks = $translationCollector->getCountFallbacks();
+
+        $this->assertSame(
+            $fallbacks,
+            0,
+            "Expected no fallback translations, but found {$fallbacks}."
+        );
+    }
+
+    /**
+     * Asserts that no missing translations were found.
+     *
+     * ```php
+     * <?php
+     * $I->dontSeeMissingTranslations();
+     * ```
+     */
+    public function dontSeeMissingTranslations(): void
+    {
+        $translationCollector = $this->grabTranslationCollector(__FUNCTION__);
+        $missings = $translationCollector->getCountMissings();
+
+        $this->assertSame(
+            $missings,
+            0,
+            "Expected no missing translations, but found {$missings}."
+        );
+    }
+
+    /**
+     * Grabs the count of defined translations.
+     *
+     * ```php
+     * <?php
+     * $count = $I->grabDefinedTranslations();
+     * ```
+     *
+     * @return int The count of defined translations.
+     */
+    public function grabDefinedTranslationsCount(): int
+    {
+        $translationCollector = $this->grabTranslationCollector(__FUNCTION__);
+        return $translationCollector->getCountDefines();
+    }
+
+    /**
+     * Asserts that there are no missing translations and no fallback translations.
+     *
+     * ```php
+     * <?php
+     * $I->seeAllTranslationsDefined();
+     * ```
+     */
+    public function seeAllTranslationsDefined(): void
+    {
+        $this->dontSeeMissingTranslations();
+        $this->dontSeeFallbackTranslations();
+    }
+
+    /**
+     * Asserts that the default locale is the expected one.
+     *
+     * ```php
+     * <?php
+     * $I->seeDefaultLocaleIs('en');
+     * ```
+     *
+     * @param string $expectedLocale The expected default locale
+     */
+    public function seeDefaultLocaleIs(string $expectedLocale): void
+    {
+        $translationCollector = $this->grabTranslationCollector(__FUNCTION__);
+        $locale = $translationCollector->getLocale();
+
+        $this->assertSame(
+            $expectedLocale,
+            $locale,
+            "Expected default locale '{$expectedLocale}', but found '{$locale}'."
+        );
+    }
+
+    /**
+     * Asserts that the fallback locales match the expected ones.
+     *
+     * ```php
+     * <?php
+     * $I->seeFallbackLocalesAre(['es', 'fr']);
+     * ```
+     *
+     * @param array $expectedLocales The expected fallback locales
+     */
+    public function seeFallbackLocalesAre(array $expectedLocales): void
+    {
+        $translationCollector = $this->grabTranslationCollector(__FUNCTION__);
+        $fallbackLocales = $translationCollector->getFallbackLocales();
+
+        if ($fallbackLocales instanceof Data) {
+            $fallbackLocales = $fallbackLocales->getValue(true);
+        }
+
+        $this->assertSame(
+            $expectedLocales,
+            $fallbackLocales,
+            "Fallback locales do not match expected."
+        );
+    }
+
+    /**
+     * Asserts that the count of fallback translations is less than the given limit.
+     *
+     * ```php
+     * <?php
+     * $I->seeFallbackTranslationsCountLessThan(10);
+     * ```
+     *
+     * @param int $limit Maximum count of fallback translations
+     */
+    public function seeFallbackTranslationsCountLessThan(int $limit): void
+    {
+        $translationCollector = $this->grabTranslationCollector(__FUNCTION__);
+        $fallbacks = $translationCollector->getCountFallbacks();
+
+        $this->assertLessThan(
+            $limit,
+            $fallbacks,
+            "Expected fewer than {$limit} fallback translations, but found {$fallbacks}."
+        );
+    }
+
+    /**
+     * Asserts that the count of missing translations is less than the given limit.
+     *
+     * ```php
+     * <?php
+     * $I->seeMissingTranslationsCountLessThan(5);
+     * ```
+     *
+     * @param int $limit Maximum count of missing translations
+     */
+    public function seeMissingTranslationsCountLessThan(int $limit): void
+    {
+        $translationCollector = $this->grabTranslationCollector(__FUNCTION__);
+        $missings = $translationCollector->getCountMissings();
+
+        $this->assertLessThan(
+            $limit,
+            $missings,
+            "Expected fewer than {$limit} missing translations, but found {$missings}."
+        );
+    }
+
+    protected function grabTranslationCollector(string $function): TranslationDataCollector
+    {
+        return $this->grabCollector('translation', $function);
+    }
+}