From 1a9ad3727ed8c5da6712f7708420d9c615f09d90 Mon Sep 17 00:00:00 2001
From: freek <freek@spatie.be>
Date: Sun, 20 Nov 2016 10:05:10 -0500
Subject: [PATCH] add tests

---
 tests/ActivityloggerTest.php | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/tests/ActivityloggerTest.php b/tests/ActivityloggerTest.php
index 8b5b537a..b1b1c506 100644
--- a/tests/ActivityloggerTest.php
+++ b/tests/ActivityloggerTest.php
@@ -203,10 +203,24 @@ public function it_will_not_replace_non_placeholders()
         $this->assertEquals($description, $this->getLastActivity()->description);
     }
 
-    public function it_returns_an_instance_of_the_activity_log_after_logging()
+    public function it_returns_an_instance_of_the_activity_after_logging()
     {
         $activityModel = activity()->log('test');
 
         $this->assertInstanceOf(Activity::class, $activityModel);
     }
+
+    /** @test */
+    public function it_returns_an_instance_of_the_activity_log_after_logging_when_using_a_custom_model()
+    {
+        $activityClass = new class extends Activity {};
+
+        $activityClassName = get_class($activityClass);
+
+        $this->app['config']->set('laravel-activitylog.activity_model', $activityClassName);
+
+        $activityModel = activity()->log('test');
+
+        $this->assertInstanceOf($activityClassName, $activityModel);
+    }
 }