0]); $this->team = Team::factory()->create(['show_boarding' => false]); }); function twoFactorTeamUser(Team $team, string $role = 'member', bool $twoFactorEnabled = false): User { $user = User::factory()->create([ 'name' => fake()->name(), 'two_factor_confirmed_at' => $twoFactorEnabled ? now() : null, ]); $team->members()->attach($user->id, ['role' => $role]); return $user; } test('a member without 2fa is sent to the setup page when the team requires 2fa', function () { $user = twoFactorTeamUser($this->team); // The session holds the team from before the change. $teamBeforeChange = $this->team->fresh(); $this->team->update(['is_two_factor_required' => true]); $this->actingAs($user)->withSession(['currentTeam' => $teamBeforeChange]); $this->get('/profile')->assertRedirect(route('two-factor.required')); $this->getJson('/profile')->assertForbidden(); }); test('a member without 2fa can open the 2fa setup pages', function () { $user = twoFactorTeamUser($this->team); $this->team->update(['is_two_factor_required' => true]); $this->actingAs($user)->withSession(['currentTeam' => $this->team]); $this->get('/two-factor-required')->assertOk()->assertSee('Configure 2FA'); $this->get('/user/confirm-password')->assertOk(); }); test('a member with 2fa can use a team that requires 2fa', function () { $user = twoFactorTeamUser($this->team, twoFactorEnabled: true); $this->team->update(['is_two_factor_required' => true]); $this->actingAs($user)->withSession(['currentTeam' => $this->team]); $this->get('/profile')->assertOk(); $this->get('/two-factor-required')->assertRedirect(route('dashboard')); }); test('a member without 2fa can use a team that does not require 2fa', function () { $user = twoFactorTeamUser($this->team); $this->actingAs($user)->withSession(['currentTeam' => $this->team]); $this->get('/profile')->assertOk(); }); test('an admin with 2fa can require 2fa for the team', function () { $admin = twoFactorTeamUser($this->team, 'admin', twoFactorEnabled: true); $this->actingAs($admin); session(['currentTeam' => $this->team]); Livewire::test(Index::class) ->set('is_two_factor_required', true) ->call('submit') ->assertDispatched('success'); expect($this->team->fresh()->is_two_factor_required)->toBeTrue(); }); test('an admin without 2fa cannot require 2fa for the team', function () { $admin = twoFactorTeamUser($this->team, 'admin'); $this->actingAs($admin); session(['currentTeam' => $this->team]); Livewire::test(Index::class) ->set('is_two_factor_required', true) ->call('submit') ->assertDispatched('error'); expect($this->team->fresh()->is_two_factor_required)->toBeFalse(); }); test('a member cannot require 2fa for the team', function () { $member = twoFactorTeamUser($this->team, twoFactorEnabled: true); $this->actingAs($member); session(['currentTeam' => $this->team]); Livewire::test(Index::class) ->set('is_two_factor_required', true) ->call('submit'); expect($this->team->fresh()->is_two_factor_required)->toBeFalse(); }); test('the setup page shows the error for a wrong 2fa code', function () { $user = twoFactorTeamUser($this->team); $user->forceFill(['two_factor_secret' => encrypt('JBSWY3DPEHPK3PXP')])->save(); $this->team->update(['is_two_factor_required' => true]); $this->actingAs($user)->withSession(['currentTeam' => $this->team, 'auth.password_confirmed_at' => time()]); $this->from('/two-factor-required') ->post('/user/confirmed-two-factor-authentication', ['code' => '000000']) ->assertRedirect('/two-factor-required'); $this->get('/two-factor-required') ->assertOk() ->assertSee('The provided two factor authentication code was invalid.'); });