def put(self):
password = self.get_argument("password")
current_username = self.get_secure_cookie("username")
UserMeta.change_password(username=current_username, password=password)
self.write({"msg": "success"})
def test_change_password(self, _session):
sample_user = mock.Mock()
with mock.patch.object(UserMeta, "query", _session.query_property()) as _query:
_query.filter_by.return_value.first.return_value = sample_user
UserMeta.change_password("Bob", "changedpassword")
pbkdf2_sha256.verify("changedpassword", sample_user.password)
_session.add.assert_called_once_with(sample_user)
_session.commit.assert_called_once_with()
_session.close.assert_called_once_with()
def post(self):
username, password = (
self.get_argument("username"), self.get_argument("password")
)
if username and password and UserMeta.check_password(username=username,
password=password):
self.set_secure_cookie("username", username)
query_data = UserMeta.query.filter_by(username=username).first()
session.close()
self.write({
"id": query_data.id,
"username": username,
"type": query_data.user_type
})
else:
self.write({
"msg": "Wrong username or password"
})
def test_check_password_if_not_has_user(self, session):
with mock.patch.object(UserMeta, "query", session.query_property()) as _query:
self.assertFalse(UserMeta.check_password("Bob", "chkpasswd"))
def test_check_password_if_has_user(self, session):
_insert_data = UserMeta(username="******", user_type=2, password="******", public_key="/path/key2.pub")
session.add(_insert_data)
with mock.patch.object(UserMeta, "query", session.query_property()) as _query:
self.assertTrue(UserMeta.check_password("Bob", "chkpasswd"))
