mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 10:53:28 -05:00
Reintroduce some tests. A bit of cleanup of mocks.
User object fields made a bit clearer and simple test to verify user added.
This commit is contained in:
@@ -73,29 +73,14 @@ class UserApiClient(BaseAPIClient):
|
||||
|
||||
class User(object):
|
||||
def __init__(self, fields, max_failed_login_count=3):
|
||||
self.fields = fields
|
||||
self.max_failed_login_count = max_failed_login_count
|
||||
self._id = fields.get('id')
|
||||
self._name = fields.get('name')
|
||||
self._email_address = fields.get('email_address')
|
||||
self._mobile_number = fields.get('mobile_number')
|
||||
self._password_changed_at = fields.get('password_changed_at')
|
||||
self._failed_login_count = 0
|
||||
|
||||
@property
|
||||
def id(self):
|
||||
return self.fields.get('id')
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
return self.fields.get('name')
|
||||
|
||||
@property
|
||||
def email_address(self):
|
||||
return self.fields.get('email_address')
|
||||
|
||||
@property
|
||||
def mobile_number(self):
|
||||
return self.fields.get('mobile_number')
|
||||
|
||||
@property
|
||||
def password_changed_at(self):
|
||||
return self.fields.get('password_changed_at')
|
||||
self._state = fields.get('state')
|
||||
self.max_failed_login_count = max_failed_login_count
|
||||
|
||||
def get_id(self):
|
||||
return self.id
|
||||
@@ -106,13 +91,53 @@ class User(object):
|
||||
def is_active(self):
|
||||
return self.state == 'active'
|
||||
|
||||
@property
|
||||
def id(self):
|
||||
return self._id
|
||||
|
||||
@id.setter
|
||||
def id(self, id):
|
||||
self._id = id
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
return self._name
|
||||
|
||||
@name.setter
|
||||
def name(self, name):
|
||||
self._name = name
|
||||
|
||||
@property
|
||||
def email_address(self):
|
||||
return self._email_address
|
||||
|
||||
@email_address.setter
|
||||
def email_address(self, email_address):
|
||||
self._email_address = email_address
|
||||
|
||||
@property
|
||||
def mobile_number(self):
|
||||
return self._mobile_number
|
||||
|
||||
@mobile_number.setter
|
||||
def mobile_number(self, mobile_number):
|
||||
self._mobile_number = mobile_number
|
||||
|
||||
@property
|
||||
def password_changed_at(self):
|
||||
return self._password_changed_at
|
||||
|
||||
@password_changed_at.setter
|
||||
def password_changed_at(self, password_changed_at):
|
||||
self._password_changed_at = password_changed_at
|
||||
|
||||
@property
|
||||
def state(self):
|
||||
return self.fields['state']
|
||||
return self._state
|
||||
|
||||
@state.setter
|
||||
def state(self, state):
|
||||
self.fields['state'] = state
|
||||
self._state = state
|
||||
|
||||
@property
|
||||
def failed_login_count(self):
|
||||
@@ -126,7 +151,15 @@ class User(object):
|
||||
return False
|
||||
|
||||
def is_locked(self):
|
||||
return self.failed_login_count > self.max_failed_login_count
|
||||
return self.failed_login_count >= self.max_failed_login_count
|
||||
|
||||
def serialize(self):
|
||||
return self.fields
|
||||
return {"id": self.id,
|
||||
"name": self.name,
|
||||
"email_address": self.email_address,
|
||||
"mobile_number": self.mobile_number,
|
||||
"password_changed_at": self.password_changed_at,
|
||||
"state": self.state,
|
||||
"failed_login_count": self.failed_login_count,
|
||||
"is_locked": self.is_locked()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user