def test_bulk_create(self):
"""
Test that bulk_create creates and saves objects, and creates an associated
historical record for the creation.
"""
licenses = [License(subscription_plan=self.subscription_plan) for _ in range(3)]
License.bulk_create(licenses)
for user_license in licenses:
user_license.refresh_from_db()
assert UNASSIGNED == user_license.status
license_history = user_license.history.all()
assert 1 == len(license_history)
assert self.CREATE_HISTORY_TYPE == user_license.history.earliest().history_type
def test_bulk_create(self, mock_track_license_changes):
"""
Test that bulk_create creates and saves objects, and creates an associated
historical record for the creation, and calls the create track_event.
"""
licenses = [License(subscription_plan=self.subscription_plan) for _ in range(3)]
License.bulk_create(licenses)
for user_license in licenses:
user_license.refresh_from_db()
assert UNASSIGNED == user_license.status
license_history = user_license.history.all()
assert 1 == len(license_history)
assert self.CREATE_HISTORY_TYPE == user_license.history.earliest().history_type
mock_track_license_changes.assert_called_with(
licenses,
SegmentEvents.LICENSE_CREATED
)
def test_bulk_update(self):
"""
Test that bulk_update saves objects, and creates an associated
historical record for the update action
"""
licenses = [License(subscription_plan=self.subscription_plan) for _ in range(3)]
License.bulk_create(licenses)
for user_license in licenses:
user_license.status = REVOKED
License.bulk_update(licenses, ['status'])
for user_license in licenses:
user_license.refresh_from_db()
assert REVOKED == user_license.status
license_history = user_license.history.all()
assert 2 == len(license_history)
assert self.CREATE_HISTORY_TYPE == user_license.history.earliest().history_type
assert self.UPDATE_HISTORY_TYPE == user_license.history.first().history_type