change compiler to support *.yaml files (#20)

Author: nutama

src/DependencyInjection/EnumTranslationCompilerPass.php

@@ -61,7 +61,7 @@ protected function registerEnums(ContainerBuilder $container, array $enum_files)
         foreach ($enum_files as $file) {
             list(, $locale, $format) = explode('.', basename($file), 3);
 
-            if (strtolower($format) !== 'yml') {
+            if (!\in_array(strtolower($format), ['yml', 'yaml'], true)) {
                 throw new \RuntimeException(sprintf("Unsupported enum translation file format '%s'.", $format));
             }
 

test/DependencyInjection/EnumTranslationCompilerPassTest.php

@@ -10,6 +10,34 @@
  */
 class EnumTranslationCompilerPassTest extends \PHPUnit_Framework_TestCase
 {
+    /**
+     * @dataProvider processLoadYamlFilesProvider
+     */
+    public function testProcessLoadYamlFiles($resource_path)
+    {
+        $resources  = realpath($resource_path);
+        $container  = new ContainerBuilder();
+        $translator = new Definition();
+        $translator->setArguments([[], [], [], [
+            'resource_files' => ['en' => [$resources]]
+        ]]);
+        $container->setDefinition('translator.default', $translator);
+
+        $pass = new EnumTranslationCompilerPass();
+        $pass->process($container);
+
+        $calls = $translator->getMethodCalls();
+
+        $this->assertEquals(1, count($calls));
+        $this->assertEquals(['addResource', ['enum', $resources, 'en', MockEnum::class]], $calls[0]);
+    }
+
+    public function processLoadYamlFilesProvider()
+    {
+        yield [__DIR__ . '/../Mock/Resources/translations/enum.en.yml'];
+        yield [__DIR__ . '/../Mock/Resources/translations/enum.en.yaml'];
+    }
+
     public function testProcess()
     {
         $resources  = realpath(__DIR__ . '/../Mock/Resources/translations/enum.en.yml');

test/Mock/Resources/translations/enum.en.yaml

@@ -0,0 +1,8 @@
+hostnet:
+    bundle:
+        entity_translation_bundle:
+            mock:
+                # Hostnet\Bundle\EntityTranslationBundle\Mock\MockEnum enum translations
+                mock_enum:
+                    foo : "Foo"
+                    bar : "Not so Bar"