From 623c926fc7cfbb6c34543a75fff4b90bfd670a71 Mon Sep 17 00:00:00 2001
From: Dennis Eichhorn <spl1nes.com@googlemail.com>
Date: Sat, 23 Jul 2016 23:34:52 +0200
Subject: [PATCH] Implement use

---
 testGenerator.php | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/testGenerator.php b/testGenerator.php
index ff879ae..f6c0929 100644
--- a/testGenerator.php
+++ b/testGenerator.php
@@ -41,13 +41,16 @@ foreach ($files as $file) {
         && stripos($subdir, 'Enum') === false
         && stripos($subdir, 'Null') === false
         && stripos($subdir, 'Interface') === false
+        && stripos($subdir, 'Trait') === false
         && stripos($subdir, 'Abstract') === false
     ) {
         if (!file_exists($testPath)) {
-            $namespace = str_replace('/', '\\', $split[0]);
-            $namespace = explode('\\', $namespace);
-            $classname = $namespace[count($namespace) - 1] . 'Test';
+            $namespace    = str_replace('/', '\\', $split[0]);
+            $namespace    = explode('\\', $namespace);
+            $classnameSrc = $namespace[count($namespace) - 1];
+            $classname    = $classnameSrc . 'Test';
             unset($namespace[count($namespace) - 1]);
+            $use        = trim('phpOMS\\' . trim(implode('\\', $namespace), '\\') . '\\' . $classnameSrc, '\\');
             $namespace  = trim('Tests\PHPUnit\Framework\\' . trim(implode('\\', $namespace), '\\'), '\\');
             $autoloader = str_repeat('/..', count(explode('\\', $namespace)));
 
@@ -73,6 +76,8 @@ foreach ($files as $file) {
                 . '' . PHP_EOL
                 . 'require_once __DIR__ . \'' . $autoloader . '/phpOMS/Autoloader.php\';' . PHP_EOL
                 . '' . PHP_EOL
+                . 'use ' . $use . ';' . PHP_EOL
+                . '' . PHP_EOL
                 . 'class ' . $classname . ' extends \PHPUnit_Framework_TestCase' . PHP_EOL
                 . '{' . PHP_EOL
                 . '}' . PHP_EOL
@@ -98,6 +103,7 @@ foreach ($files as $file) {
         && stripos($subdir, 'Enum') === false
         && stripos($subdir, 'Null') === false
         && stripos($subdir, 'Interface') === false
+        && stripos($subdir, 'Trait') === false
         && stripos($subdir, 'Abstract') === false
     ) {
         if (!file_exists($testPath)) {