Explicitly setting currentTeam relation on switchTeam to avoid refresh. (#331)

Author: telkins

src/HasTeams.php

@@ -45,6 +45,8 @@ public function switchTeam($team)
             'current_team_id' => $team->id,
         ])->save();
 
+        $this->setRelation('currentTeam', $team);
+
         return true;
     }
 

tests/TeamBehaviorTest.php

@@ -129,6 +129,31 @@ public function test_has_team_permission_checks_token_permissions()
         $this->assertTrue($john->hasTeamPermission($team, 'foo'));
     }
 
+    public function test_user_does_not_need_to_refresh_after_switching_teams()
+    {
+        $this->migrate();
+
+        $action = new CreateTeam;
+
+        $user = User::forceCreate([
+            'name' => 'Taylor Otwell',
+            'email' => '[email protected]',
+            'password' => 'secret',
+        ]);
+
+        $personalTeam = $action->create($user, ['name' => 'Personal Team']);
+
+        $personalTeam->forceFill(['personal_team' => true])->save();
+
+        $anotherTeam = $action->create($user, ['name' => 'Test Team']);
+
+        $this->assertTrue($user->isCurrentTeam($personalTeam));
+
+        $user->switchTeam($anotherTeam);
+
+        $this->assertTrue($user->isCurrentTeam($anotherTeam));
+    }
+
     protected function migrate()
     {
         $this->artisan('migrate', ['--database' => 'testbench'])->run();