X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/c09300c06faa1ccedc6a89e7c26fd7f1bd3511bf..refs/pull/5280/head:/tests/Commands/ClearActivityCommandTest.php

diff --git a/tests/Commands/ClearActivityCommandTest.php b/tests/Commands/ClearActivityCommandTest.php
index 60f75675d..a3d7456b6 100644
--- a/tests/Commands/ClearActivityCommandTest.php
+++ b/tests/Commands/ClearActivityCommandTest.php
@@ -1,7 +1,11 @@
-<?php namespace Tests\Commands;
+<?php
 
-use BookStack\Actions\ActivityType;
-use BookStack\Entities\Models\Page;
+namespace Tests\Commands;
+
+use BookStack\Activity\ActivityType;
+use BookStack\Facades\Activity;
+use Illuminate\Support\Facades\Artisan;
+use Illuminate\Support\Facades\DB;
 use Tests\TestCase;
 
 class ClearActivityCommandTest extends TestCase
@@ -9,21 +13,22 @@ class ClearActivityCommandTest extends TestCase
     public function test_clear_activity_command()
     {
         $this->asEditor();
-        $page = Page::first();
-        \Activity::addForEntity($page, ActivityType::PAGE_UPDATE);
+        $page = $this->entities->page();
+        Activity::add(ActivityType::PAGE_UPDATE, $page);
 
         $this->assertDatabaseHas('activities', [
-            'type' => 'page_update',
-            'entity_id' => $page->id,
-            'user_id' => $this->getEditor()->id
+            'type'      => 'page_update',
+            'loggable_id' => $page->id,
+            'user_id'   => $this->users->editor()->id,
         ]);
 
-        $exitCode = \Artisan::call('bookstack:clear-activity');
+        DB::rollBack();
+        $exitCode = Artisan::call('bookstack:clear-activity');
+        DB::beginTransaction();
         $this->assertTrue($exitCode === 0, 'Command executed successfully');
 
-
         $this->assertDatabaseMissing('activities', [
-            'type' => 'page_update'
+            'type' => 'page_update',
         ]);
     }
-}
\ No newline at end of file
+}