def check_total_employee(auth=None, all=None):
auth_count = auth or User.get_auth_user_count()
all_count = all or User.get_all_users_count()
if auth_count > all_count:
return jsonify({
"auth_count": auth_count,
"all_count": all_count,
"msg": "Error! Auth emp count is more than employees"
})
elif auth_count < all_count:
percent = int((auth_count / all_count) * 100)
if percent < 95:
return jsonify({
"auth_count": auth_count,
"all_count": all_count,
"msg": "Percentage of joining is too less"
})
else:
return jsonify({
"auth_count": auth_count,
"all_count": all_count,
"msg": "Percentage of joining is OK"
})
else:
return jsonify({
"auth_count": auth_count,
"all_count": all_count,
"msg": "All good"
})
def test_get_auth_user_count(self, mock_user):
mock_user.query.filter_by(authenticated=1).count.return_value = 9981
count = User.get_auth_user_count()
#print (count)
self.assertEqual(count, 9989)
def get_auth_users():
auth_users = User.get_auth_user_count()
return jsonify({"count": auth_users})
