96 lines
1.8 KiB
PHP
96 lines
1.8 KiB
PHP
<?php
|
|
|
|
namespace Tests\Setup;
|
|
|
|
use Vanguard\Role;
|
|
use Vanguard\Support\Enum\UserStatus;
|
|
use Vanguard\User;
|
|
use \Facades\Tests\Setup\RoleFactory;
|
|
|
|
class UserFactory
|
|
{
|
|
protected $params = [];
|
|
protected $role;
|
|
|
|
public function withCredentials($username, $password)
|
|
{
|
|
$this->params['username'] = $username;
|
|
$this->params['password'] = $password;
|
|
|
|
return $this;
|
|
}
|
|
|
|
public function banned()
|
|
{
|
|
$this->params['status'] = UserStatus::BANNED;
|
|
|
|
return $this;
|
|
}
|
|
|
|
public function unconfirmed()
|
|
{
|
|
$this->params['status'] = UserStatus::UNCONFIRMED;
|
|
|
|
return $this;
|
|
}
|
|
|
|
public function unverified()
|
|
{
|
|
$this->params['email_verified_at'] = null;
|
|
|
|
return $this;
|
|
}
|
|
|
|
public function role(Role $role)
|
|
{
|
|
$this->role = $role;
|
|
|
|
return $this;
|
|
}
|
|
|
|
public function admin()
|
|
{
|
|
return $this->role(Role::where('name', 'Admin')->first());
|
|
}
|
|
|
|
public function user()
|
|
{
|
|
return $this->role(Role::where('name', 'User')->first());
|
|
}
|
|
|
|
public function email($email)
|
|
{
|
|
$this->params['email'] = $email;
|
|
|
|
return $this;
|
|
}
|
|
|
|
public function twoFactor($countryCode, $phone)
|
|
{
|
|
$this->params['two_factor_country_code'] = $countryCode;
|
|
$this->params['two_factor_phone'] = $phone;
|
|
|
|
return $this;
|
|
}
|
|
|
|
public function withPermissions($permissions)
|
|
{
|
|
$permissions = func_get_args();
|
|
|
|
$this->role(
|
|
RoleFactory::withPermissions($permissions)->create()
|
|
);
|
|
|
|
return $this;
|
|
}
|
|
|
|
public function create(array $overrides = [])
|
|
{
|
|
if ($this->role) {
|
|
$this->params['role_id'] = $this->role;
|
|
}
|
|
|
|
return User::factory()->create(array_merge($this->params, $overrides));
|
|
}
|
|
}
|