bug #150 make entity, related class without namespace (LeJeanbono)

This PR was merged into the 1.0-dev branch. Discussion ---------- make entity, related class without namespace For #148 Commits ------- 5e5e5e6 make entity, related class without namespace
symfony · Apr 11, 2018 · 253cee4 · 253cee4
2 parents afbf1c0 + 5e5e5e6
commit 253cee4
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/src/Str.php b/src/Str.php
@@ -117,7 +117,11 @@ public static function asEventMethod(string $eventName): string
 
     public static function getShortClassName(string $fullClassName): string
     {
-        return substr($fullClassName, strrpos($fullClassName, '\\') + 1);
+        if (!empty(self::getNamespace($fullClassName))) {
+            return substr($fullClassName, strrpos($fullClassName, '\\') + 1);
+        } else {
+            return $fullClassName;
+        }
     }
 
     public static function getNamespace(string $fullClassName): string

diff --git a/tests/StrTest.php b/tests/StrTest.php
@@ -161,4 +161,18 @@ public function getAsCamelCaseTests()
 
         yield ['foo_bar.baz\\pizza', 'FooBarBazPizza'];
     }
+
+    /**
+     * @dataProvider getShortClassNameCaseTests
+     */
+    public function testShortClassName(string $original, string $expected)
+    {
+        $this->assertSame($expected, Str::getShortClassName($original));
+    }
+
+    public function getShortClassNameCaseTests()
+    {
+        yield ['App\\Entity\\Foo', 'Foo'];
+        yield ['Foo', 'Foo'];
+    }
 }