class LocalePluralFormatTest

class LocalePluralFormatTest extends BrowserTestBase {
  
  /**
   * An admin user.
   *
   * @var \Drupal\user\Entity\User
   */
  protected $adminUser;
  
  /**
   * Modules to enable.
   *
   * @var array
   */
  public static $modules = [
    'locale',
  ];
  
  /**
   * {@inheritdoc}
   */
  protected $defaultTheme = 'stark';
  
  /**
   * {@inheritdoc}
   */
  protected function setUp() {
    parent::setUp();
    $this->adminUser = $this->drupalCreateUser([
      'administer languages',
      'translate interface',
      'access administration pages',
    ]);
    $this->drupalLogin($this->adminUser);
  }
  
  /**
   * Tests locale_get_plural() and \Drupal::translation()->formatPlural()
   * functionality.
   */
  public function testGetPluralFormat() {
    // Import some .po files with formulas to set up the environment.
    // These will also add the languages to the system.
    $this->importPoFile($this->getPoFileWithSimplePlural(), [
      'langcode' => 'fr',
    ]);
    $this->importPoFile($this->getPoFileWithComplexPlural(), [
      'langcode' => 'hr',
    ]);
    // Attempt to import some broken .po files as well to prove that these
    // will not overwrite the proper plural formula imported above.
    $this->importPoFile($this->getPoFileWithMissingPlural(), [
      'langcode' => 'fr',
      'overwrite_options[not_customized]' => TRUE,
    ]);
    $this->importPoFile($this->getPoFileWithBrokenPlural(), [
      'langcode' => 'hr',
      'overwrite_options[not_customized]' => TRUE,
    ]);
    // Reset static caches from locale_get_plural() to ensure we get fresh data.
    drupal_static_reset('locale_get_plural');
    drupal_static_reset('locale_get_plural:plurals');
    drupal_static_reset('locale');
    // Expected plural translation strings for each plural index.
    $plural_strings = [
      // English is not imported in this case, so we assume built-in text
      // and formulas.
'en' => [
        0 => '1 hour',
        1 => '@count hours',
      ],
      'fr' => [
        0 => '@count heure',
        1 => '@count heures',
      ],
      'hr' => [
        0 => '@count sat',
        1 => '@count sata',
        2 => '@count sati',
      ],
      // Hungarian is not imported, so it should assume the same text as
      // English, but it will always pick the plural form as per the built-in
      // logic, so only index -1 is relevant with the plural value.
'hu' => [
        0 => '1 hour',
        -1 => '@count hours',
      ],
    ];
    // Expected plural indexes precomputed base on the plural formulas with
    // given $count value.
    $plural_tests = [
      'en' => [
        1 => 0,
        0 => 1,
        5 => 1,
        123 => 1,
        235 => 1,
      ],
      'fr' => [
        1 => 0,
        0 => 0,
        5 => 1,
        123 => 1,
        235 => 1,
      ],
      'hr' => [
        1 => 0,
        21 => 0,
        0 => 2,
        2 => 1,
        8 => 2,
        123 => 1,
        235 => 2,
      ],
      'hu' => [
        1 => -1,
        21 => -1,
        0 => -1,
      ],
    ];
    foreach ($plural_tests as $langcode => $tests) {
      foreach ($tests as $count => $expected_plural_index) {
        // Assert that the we get the right plural index.
        $this->assertIdentical(locale_get_plural($count, $langcode), $expected_plural_index, 'Computed plural index for ' . $langcode . ' for count ' . $count . ' is ' . $expected_plural_index);
        // Assert that the we get the right translation for that. Change the
        // expected index as per the logic for translation lookups.
        $expected_plural_index = $count == 1 ? 0 : $expected_plural_index;
        $expected_plural_string = str_replace('@count', $count, $plural_strings[$langcode][$expected_plural_index]);
        $this->assertIdentical(\Drupal::translation()->formatPlural($count, '1 hour', '@count hours', [], [
          'langcode' => $langcode,
        ])
          ->render(), $expected_plural_string, 'Plural translation of 1 hours / @count hours for count ' . $count . ' in ' . $langcode . ' is ' . $expected_plural_string);
        // DO NOT use translation to pass translated strings into
        // PluralTranslatableMarkup::createFromTranslatedString() this way. It
        // is designed to be used with *already* translated text like settings
        // from configuration. We use PHP translation here just because we have
        // the expected result data in that format.
        $translated_string = \Drupal::translation()->translate('1 hour' . PoItem::DELIMITER . '@count hours', [], [
          'langcode' => $langcode,
        ]);
        $plural = PluralTranslatableMarkup::createFromTranslatedString($count, $translated_string, [], [
          'langcode' => $langcode,
        ]);
        $this->assertIdentical($plural->render(), $expected_plural_string);
      }
    }
  }
  
  /**
   * Tests plural editing of DateFormatter strings
   */
  public function testPluralEditDateFormatter() {
    // Import some .po files with formulas to set up the environment.
    // These will also add the languages to the system.
    $this->importPoFile($this->getPoFileWithSimplePlural(), [
      'langcode' => 'fr',
    ]);
    // Set French as the site default language.
    $this->config('system.site')
      ->set('default_langcode', 'fr')
      ->save();
    // Visit User Info page before updating translation strings. Change the
    // created time to ensure that the we're dealing in seconds and it can't be
    // exactly 1 minute.
    $this->adminUser
      ->set('created', time() - 1)
      ->save();
    $this->drupalGet('user');
    // Member for time should be translated.
    $this->assertText("seconde", "'Member for' text is translated.");
    $path = 'admin/config/regional/translate/';
    $search = [
      'langcode' => 'fr',
      // Limit to only translated strings to ensure that database ordering does
      // not break the test.
'translation' => 'translated',
    ];
    $this->drupalPostForm($path, $search, t('Filter'));
    // Plural values for the langcode fr.
    $this->assertText('@count seconde');
    $this->assertText('@count secondes');
    // Inject a plural source string to the database. We need to use a specific
    // langcode here because the language will be English by default and will
    // not save our source string for performance optimization if we do not ask
    // specifically for a language.
    \Drupal::translation()->formatPlural(1, '1 second', '@count seconds', [], [
      'langcode' => 'fr',
    ])
      ->render();
    $lid = Database::getConnection()->select('locales_source', 'ls')
      ->fields('ls', [
      'lid',
    ])
      ->condition('source', "1 second" . PoItem::DELIMITER . "@count seconds")
      ->condition('context', '')
      ->execute()
      ->fetchField();
    // Look up editing page for this plural string and check fields.
    $search = [
      'string' => '1 second',
      'langcode' => 'fr',
    ];
    $this->drupalPostForm('admin/config/regional/translate', $search, t('Filter'));
    // Save complete translations for the string in langcode fr.
    $edit = [
      "strings[{$lid}][translations][0]" => '1 seconde updated',
      "strings[{$lid}][translations][1]" => '@count secondes updated',
    ];
    $this->drupalPostForm($path, $edit, t('Save translations'));
    // User interface input for translating seconds should not be duplicated
    $this->assertUniqueText('@count seconds', 'Interface translation input for @count seconds only appears once.');
    // Member for time should be translated. Change the created time to ensure
    // that the we're dealing in multiple seconds and it can't be exactly 1
    // second or minute.
    $this->adminUser
      ->set('created', time() - 2)
      ->save();
    $this->drupalGet('user');
    $this->assertText("secondes updated", "'Member for' text is translated.");
  }
  
  /**
   * Tests plural editing and export functionality.
   */
  public function testPluralEditExport() {
    // Import some .po files with formulas to set up the environment.
    // These will also add the languages to the system.
    $this->importPoFile($this->getPoFileWithSimplePlural(), [
      'langcode' => 'fr',
    ]);
    $this->importPoFile($this->getPoFileWithComplexPlural(), [
      'langcode' => 'hr',
    ]);
    // Get the French translations.
    $this->drupalPostForm('admin/config/regional/translate/export', [
      'langcode' => 'fr',
    ], t('Export'));
    // Ensure we have a translation file.
    $this->assertRaw('# French translation of Drupal', 'Exported French translation file.');
    // Ensure our imported translations exist in the file.
    $this->assertRaw("msgid \"Monday\"\nmsgstr \"lundi\"", 'French translations present in exported file.');
    // Check for plural export specifically.
    $this->assertRaw("msgid \"1 hour\"\nmsgid_plural \"@count hours\"\nmsgstr[0] \"@count heure\"\nmsgstr[1] \"@count heures\"", 'Plural translations exported properly.');
    // Get the Croatian translations.
    $this->drupalPostForm('admin/config/regional/translate/export', [
      'langcode' => 'hr',
    ], t('Export'));
    // Ensure we have a translation file.
    $this->assertRaw('# Croatian translation of Drupal', 'Exported Croatian translation file.');
    // Ensure our imported translations exist in the file.
    $this->assertRaw("msgid \"Monday\"\nmsgstr \"Ponedjeljak\"", 'Croatian translations present in exported file.');
    // Check for plural export specifically.
    $this->assertRaw("msgid \"1 hour\"\nmsgid_plural \"@count hours\"\nmsgstr[0] \"@count sat\"\nmsgstr[1] \"@count sata\"\nmsgstr[2] \"@count sati\"", 'Plural translations exported properly.');
    // Check if the source appears on the translation page.
    $this->drupalGet('admin/config/regional/translate');
    $this->assertText("1 hour");
    $this->assertText("@count hours");
    // Look up editing page for this plural string and check fields.
    $path = 'admin/config/regional/translate/';
    $search = [
      'langcode' => 'hr',
    ];
    $this->drupalPostForm($path, $search, t('Filter'));
    // Labels for plural editing elements.
    $this->assertText('Singular form');
    $this->assertText('First plural form');
    $this->assertText('2. plural form');
    $this->assertNoText('3. plural form');
    // Plural values for langcode hr.
    $this->assertText('@count sat');
    $this->assertText('@count sata');
    $this->assertText('@count sati');
    $connection = Database::getConnection();
    // Edit langcode hr translations and see if that took effect.
    $lid = $connection->select('locales_source', 'ls')
      ->fields('ls', [
      'lid',
    ])
      ->condition('source', "1 hour" . PoItem::DELIMITER . "@count hours")
      ->condition('context', '')
      ->execute()
      ->fetchField();
    $edit = [
      "strings[{$lid}][translations][1]" => '@count sata edited',
    ];
    $this->drupalPostForm($path, $edit, t('Save translations'));
    $search = [
      'langcode' => 'fr',
    ];
    $this->drupalPostForm('admin/config/regional/translate', $search, t('Filter'));
    // Plural values for the langcode fr.
    $this->assertText('@count heure');
    $this->assertText('@count heures');
    $this->assertNoText('2. plural form');
    // Edit langcode fr translations and see if that took effect.
    $edit = [
      "strings[{$lid}][translations][0]" => '@count heure edited',
    ];
    $this->drupalPostForm($path, $edit, t('Save translations'));
    // Inject a plural source string to the database. We need to use a specific
    // langcode here because the language will be English by default and will
    // not save our source string for performance optimization if we do not ask
    // specifically for a language.
    \Drupal::translation()->formatPlural(1, '1 day', '@count days', [], [
      'langcode' => 'fr',
    ])
      ->render();
    $lid = $connection->select('locales_source', 'ls')
      ->fields('ls', [
      'lid',
    ])
      ->condition('source', "1 day" . PoItem::DELIMITER . "@count days")
      ->condition('context', '')
      ->execute()
      ->fetchField();
    // Look up editing page for this plural string and check fields.
    $search = [
      'string' => '1 day',
      'langcode' => 'fr',
    ];
    $this->drupalPostForm('admin/config/regional/translate', $search, t('Filter'));
    // Save complete translations for the string in langcode fr.
    $edit = [
      "strings[{$lid}][translations][0]" => '1 jour',
      "strings[{$lid}][translations][1]" => '@count jours',
    ];
    $this->drupalPostForm($path, $edit, t('Save translations'));
    // Save complete translations for the string in langcode hr.
    $search = [
      'string' => '1 day',
      'langcode' => 'hr',
    ];
    $this->drupalPostForm('admin/config/regional/translate', $search, t('Filter'));
    $edit = [
      "strings[{$lid}][translations][0]" => '@count dan',
      "strings[{$lid}][translations][1]" => '@count dana',
      "strings[{$lid}][translations][2]" => '@count dana',
    ];
    $this->drupalPostForm($path, $edit, t('Save translations'));
    // Get the French translations.
    $this->drupalPostForm('admin/config/regional/translate/export', [
      'langcode' => 'fr',
    ], t('Export'));
    // Check for plural export specifically.
    $this->assertRaw("msgid \"1 hour\"\nmsgid_plural \"@count hours\"\nmsgstr[0] \"@count heure edited\"\nmsgstr[1] \"@count heures\"", 'Edited French plural translations for hours exported properly.');
    $this->assertRaw("msgid \"1 day\"\nmsgid_plural \"@count days\"\nmsgstr[0] \"1 jour\"\nmsgstr[1] \"@count jours\"", 'Added French plural translations for days exported properly.');
    // Get the Croatian translations.
    $this->drupalPostForm('admin/config/regional/translate/export', [
      'langcode' => 'hr',
    ], t('Export'));
    // Check for plural export specifically.
    $this->assertRaw("msgid \"1 hour\"\nmsgid_plural \"@count hours\"\nmsgstr[0] \"@count sat\"\nmsgstr[1] \"@count sata edited\"\nmsgstr[2] \"@count sati\"", 'Edited Croatian plural translations exported properly.');
    $this->assertRaw("msgid \"1 day\"\nmsgid_plural \"@count days\"\nmsgstr[0] \"@count dan\"\nmsgstr[1] \"@count dana\"\nmsgstr[2] \"@count dana\"", 'Added Croatian plural translations exported properly.');
  }
  
  /**
   * Imports a standalone .po file in a given language.
   *
   * @param string $contents
   *   Contents of the .po file to import.
   * @param array $options
   *   Additional options to pass to the translation import form.
   */
  public function importPoFile($contents, array $options = []) {
    $file_system = \Drupal::service('file_system');
    $name = $file_system->tempnam('temporary://', "po_") . '.po';
    file_put_contents($name, $contents);
    $options['files[file]'] = $name;
    $this->drupalPostForm('admin/config/regional/translate/import', $options, t('Import'));
    $file_system->unlink($name);
  }
  
  /**
   * Returns a .po file with a simple plural formula.
   */
  public function getPoFileWithSimplePlural() {
    return <<<EOF
    msgid ""
    msgstr ""
    "Project-Id-Version: Drupal 8\\n"
    "MIME-Version: 1.0\\n"
    "Content-Type: text/plain; charset=UTF-8\\n"
    "Content-Transfer-Encoding: 8bit\\n"
    "Plural-Forms: nplurals=2; plural=(n > 1);\\n"
    
    msgid "1 hour"
    msgid_plural "@count hours"
    msgstr[0] "@count heure"
    msgstr[1] "@count heures"
    
    msgid "1 second"
    msgid_plural "@count seconds"
    msgstr[0] "@count seconde"
    msgstr[1] "@count secondes"
    
    msgid "Monday"
    msgstr "lundi"
    EOF;
  }
  
  /**
   * Returns a .po file with a complex plural formula.
   */
  public function getPoFileWithComplexPlural() {
    return <<<EOF
    msgid ""
    msgstr ""
    "Project-Id-Version: Drupal 8\\n"
    "MIME-Version: 1.0\\n"
    "Content-Type: text/plain; charset=UTF-8\\n"
    "Content-Transfer-Encoding: 8bit\\n"
    "Plural-Forms: nplurals=3; plural=n%10==1 && n%100!=11 ? 0 : n%10>=2 && n%10<=4 && (n%100<10 || n%100>=20) ? 1 : 2;\\n"
    
    msgid "1 hour"
    msgid_plural "@count hours"
    msgstr[0] "@count sat"
    msgstr[1] "@count sata"
    msgstr[2] "@count sati"
    
    msgid "Monday"
    msgstr "Ponedjeljak"
    EOF;
  }
  
  /**
   * Returns a .po file with a missing plural formula.
   */
  public function getPoFileWithMissingPlural() {
    return <<<EOF
    msgid ""
    msgstr ""
    "Project-Id-Version: Drupal 8\\n"
    "MIME-Version: 1.0\\n"
    "Content-Type: text/plain; charset=UTF-8\\n"
    "Content-Transfer-Encoding: 8bit\\n"
    
    msgid "Monday"
    msgstr "lundi"
    EOF;
  }
  
  /**
   * Returns a .po file with a broken plural formula.
   */
  public function getPoFileWithBrokenPlural() {
    return <<<EOF
    msgid ""
    msgstr ""
    "Project-Id-Version: Drupal 8\\n"
    "MIME-Version: 1.0\\n"
    "Content-Type: text/plain; charset=UTF-8\\n"
    "Content-Transfer-Encoding: 8bit\\n"
    "Plural-Forms: broken, will not parse\\n"
    
    msgid "Monday"
    msgstr "Ponedjeljak"
    EOF;
  }

}