def test_run_sendinblue_only(mock_import_contacts):
"""
Test that two chunks of users are used and therefore two requests are sent.
"""
UserFactory.create_batch(5)
run(4, synchronize_batch=False)
assert len(mock_import_contacts.call_args_list) == 2
def test_run_batch_only():
"""
Test that two chunks of users are used and therefore two requests are sent.
"""
UserFactory.create_batch(5)
run(4, synchronize_sendinblue=False)
assert len(push_testing.requests) == 2
def test_run(mock_import_contacts):
"""
Test that two chunks of users are used and therefore two requests are sent.
"""
UserFactory.create_batch(5)
run(4)
assert len(push_testing.requests) == 2
assert len(mock_import_contacts.call_args_list) == 2
def test_get_users_chunks():
"""
Test that the correct number of chunks have been fetched, that each one
does not exceed the maximum chunk size, and that all of the users have been
fetched.
"""
users = UserFactory.create_batch(15)
chunks = list(get_users_chunks(6))
assert len(chunks) == 3
assert [len(chunk) for chunk in chunks] == [6, 6, 3]
expected_ids = {user.id for user in users}
found_ids = {user.id for chunk in chunks for user in chunk}
assert found_ids == expected_ids
