Fix test + lint

This commit is contained in:
zandercymatics 2024-04-22 11:08:59 -06:00
parent 68e44eb98e
commit 2a7eb6db81
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
3 changed files with 7 additions and 6 deletions

View file

@ -6,7 +6,7 @@ from registrar.models import (
User, User,
UserGroup, UserGroup,
) )
from registrar.models.verified_by_staff import VerifiedByStaff
fake = Faker() fake = Faker()
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)

View file

@ -29,6 +29,7 @@ class User(AbstractUser):
Users achieve access to our system in a few different ways. Users achieve access to our system in a few different ways.
These choices reflect those pathways. These choices reflect those pathways.
""" """
GRANDFATHERED = "grandfathered", "Legacy user" GRANDFATHERED = "grandfathered", "Legacy user"
VERIFIED_BY_STAFF = "verified_by_staff", "Verified by staff" VERIFIED_BY_STAFF = "verified_by_staff", "Verified by staff"
REGULAR = "regular", "Verified by Login.gov" REGULAR = "regular", "Verified by Login.gov"

View file

@ -2820,7 +2820,7 @@ class MyUserAdminTest(TestCase):
request.user = create_user() request.user = create_user()
fieldsets = self.admin.get_fieldsets(request) fieldsets = self.admin.get_fieldsets(request)
expected_fieldsets = ( expected_fieldsets = (
(None, {"fields": ("password", "status")}), (None, {"fields": ("password", "status", "verification_type")}),
("Personal Info", {"fields": ("first_name", "last_name", "email")}), ("Personal Info", {"fields": ("first_name", "last_name", "email")}),
("Permissions", {"fields": ("is_active", "groups")}), ("Permissions", {"fields": ("is_active", "groups")}),
("Important dates", {"fields": ("last_login", "date_joined")}), ("Important dates", {"fields": ("last_login", "date_joined")}),