def test_all_with_multiple_rows():
nipsa_user1 = models.NipsaUser("test_id1")
db.Session.add(nipsa_user1)
nipsa_user2 = models.NipsaUser("test_id2")
db.Session.add(nipsa_user2)
nipsa_user3 = models.NipsaUser("test_id3")
db.Session.add(nipsa_user3)
assert models.NipsaUser.all() == [nipsa_user1, nipsa_user2, nipsa_user3]
def flag(self, userid):
"""
Add a NIPSA flag for a userid.
Add the given user's ID to the list of NIPSA'd user IDs. If the user
is already NIPSA'd then nothing will happen (but an "add_nipsa"
message for the user will still be published to the queue).
"""
nipsa_user = self._user_query(userid).one_or_none()
if nipsa_user is None:
nipsa_user = models.NipsaUser(userid)
self.session.add(nipsa_user)
worker.add_nipsa.delay(userid)
def add_nipsa(request, userid):
"""NIPSA a user.
Add the given user's ID to the list of NIPSA'd user IDs.
If the user is already NIPSA'd then nothing will happen (but an "add_nipsa"
message for the user will still be published to the queue).
"""
nipsa_user = models.NipsaUser.get_by_userid(userid)
if not nipsa_user:
nipsa_user = models.NipsaUser(userid)
request.db.add(nipsa_user)
worker.add_nipsa.delay(userid)
def add_nipsa(request, userid):
"""NIPSA a user.
Add the given user's ID to the list of NIPSA'd user IDs.
If the user is already NIPSA'd then nothing will happen (but an "add_nipsa"
message for the user will still be published to the queue).
"""
nipsa_user = models.NipsaUser.get_by_userid(userid)
if not nipsa_user:
nipsa_user = models.NipsaUser(userid)
request.db.add(nipsa_user)
_publish(request, json.dumps({"action": "add_nipsa", "userid": userid}))
def test_init():
nipsa_user = models.NipsaUser("test_id")
assert nipsa_user.userid == "test_id"
def test_null_id(db_session):
with pytest.raises(exc.IntegrityError):
db_session.add(models.NipsaUser(None))
db_session.flush()
def test_two_rows_with_same_id(db_session):
db_session.add(models.NipsaUser("test_id"))
with pytest.raises(exc.IntegrityError):
db_session.add(models.NipsaUser("test_id"))
db_session.flush()
def test_all_with_one_row():
nipsa_user = models.NipsaUser("test_id")
db.Session.add(nipsa_user)
assert models.NipsaUser.all() == [nipsa_user]
def test_get_by_userid_with_matching_user():
nipsa_user = models.NipsaUser("test_id")
db.Session.add(nipsa_user)
assert models.NipsaUser.get_by_userid("test_id") == nipsa_user