From 37ef240f39cd808d443275d8746597adeb875ec3 Mon Sep 17 00:00:00 2001
From: Nafies Luthfi <nafiesl@gmail.com>
Date: Thu, 29 Mar 2018 07:41:24 +0800
Subject: [PATCH] Update browserkit base test class

---
 src/stubs/test-browserkit-base-class.stub            |  7 ++++++-
 tests/Generators/FeatureTestGeneratorTest.php        | 14 ++++++++++++--
 tests/Generators/Simple/FeatureTestGeneratorTest.php | 14 ++++++++++++--
 3 files changed, 30 insertions(+), 5 deletions(-)

diff --git a/src/stubs/test-browserkit-base-class.stub b/src/stubs/test-browserkit-base-class.stub
index 919f66b..c0aa4c7 100644
--- a/src/stubs/test-browserkit-base-class.stub
+++ b/src/stubs/test-browserkit-base-class.stub
@@ -19,9 +19,14 @@ abstract class BrowserKitTest extends BaseTestCase
 
     protected function loginAsUser()
     {
-        $user = factory(User::class)->create();
+        $user = $this->createUser();
         $this->actingAs($user);
 
         return $user;
     }
+
+    protected function createUser()
+    {
+        return factory(User::class)->create();
+    }
 }
diff --git a/tests/Generators/FeatureTestGeneratorTest.php b/tests/Generators/FeatureTestGeneratorTest.php
index ef4f22a..a071491 100644
--- a/tests/Generators/FeatureTestGeneratorTest.php
+++ b/tests/Generators/FeatureTestGeneratorTest.php
@@ -33,11 +33,16 @@ abstract class BrowserKitTest extends BaseTestCase
 
     protected function loginAsUser()
     {
-        \$user = factory(User::class)->create();
+        \$user = \$this->createUser();
         \$this->actingAs(\$user);
 
         return \$user;
     }
+
+    protected function createUser()
+    {
+        return factory(User::class)->create();
+    }
 }
 ";
         $this->assertEquals($browserKitTestClassContent, file_get_contents(base_path("tests/BrowserKitTest.php")));
@@ -221,11 +226,16 @@ abstract class {$baseTestClass} extends BaseTestCase
 
     protected function loginAsUser()
     {
-        \$user = factory(User::class)->create();
+        \$user = \$this->createUser();
         \$this->actingAs(\$user);
 
         return \$user;
     }
+
+    protected function createUser()
+    {
+        return factory(User::class)->create();
+    }
 }
 ";
         $this->assertEquals($browserKitTestClassContent, file_get_contents($baseTestPath));
diff --git a/tests/Generators/Simple/FeatureTestGeneratorTest.php b/tests/Generators/Simple/FeatureTestGeneratorTest.php
index 70dfbc2..b198797 100644
--- a/tests/Generators/Simple/FeatureTestGeneratorTest.php
+++ b/tests/Generators/Simple/FeatureTestGeneratorTest.php
@@ -33,11 +33,16 @@ abstract class BrowserKitTest extends BaseTestCase
 
     protected function loginAsUser()
     {
-        \$user = factory(User::class)->create();
+        \$user = \$this->createUser();
         \$this->actingAs(\$user);
 
         return \$user;
     }
+
+    protected function createUser()
+    {
+        return factory(User::class)->create();
+    }
 }
 ";
         $this->assertEquals($browserKitTestClassContent, file_get_contents(base_path("tests/BrowserKitTest.php")));
@@ -174,11 +179,16 @@ abstract class {$baseTestClass} extends BaseTestCase
 
     protected function loginAsUser()
     {
-        \$user = factory(User::class)->create();
+        \$user = \$this->createUser();
         \$this->actingAs(\$user);
 
         return \$user;
     }
+
+    protected function createUser()
+    {
+        return factory(User::class)->create();
+    }
 }
 ";
         $this->assertEquals($browserKitTestClassContent, file_get_contents($baseTestPath));