def test_workbasket_errored_updates_tracked_models(
upload,
new_workbasket,
valid_user,
settings,
):
settings.TRANSACTION_SCHEMA = "workbaskets.models.SEED_FIRST"
original_footnote = factories.FootnoteFactory.create()
new_footnote = original_footnote.new_version(
workbasket=new_workbasket,
update_type=UpdateType.UPDATE,
)
assert_workbasket_valid(new_workbasket)
new_workbasket.submit_for_approval()
new_footnote.refresh_from_db()
assert new_footnote.version_group.current_version.pk == original_footnote.pk
new_workbasket.approve(valid_user.pk, "SEED_FIRST")
new_footnote.refresh_from_db()
assert new_footnote.version_group.current_version.pk == new_footnote.pk
new_workbasket.export_to_cds()
new_footnote.refresh_from_db()
assert new_footnote.version_group.current_version.pk == new_footnote.pk
new_workbasket.cds_error()
new_footnote.refresh_from_db()
assert new_footnote.version_group.current_version.pk == original_footnote.pk
def test_edit_after_submit(
upload,
valid_user_client,
date_ranges,
other_statuses,
should_reuse,
):
# submit a workbasket containing a newly created footnote
workbasket = factories.WorkBasketFactory.create()
with workbasket.new_transaction():
footnote = factories.FootnoteFactory.create(
update_type=UpdateType.CREATE,
)
assert footnote.transaction.workbasket == workbasket
assert_workbasket_valid(workbasket)
# create workbaskets in different unapproved states
# to check that the system doesn't select these
other_baskets = [
factories.WorkBasketFactory.create(status=other_status)
for other_status in other_statuses
]
response = valid_user_client.get(
reverse(
"workbaskets:workbasket-ui-submit",
kwargs={"pk": workbasket.pk},
),
)
assert response.status_code == 302
# edit the footnote description start date, to avoid FO4 violation
description = footnote.descriptions.first()
description.validity_start = date_ranges.later.lower
description.save(force_write=True)
# edit the footnote
response = valid_user_client.post(
footnote.get_url("edit"),
validity_period_post_data(
date_ranges.later.lower,
date_ranges.later.upper,
),
)
assert response.status_code == 302
# check that the session workbasket has been replaced by a new one
session_workbasket = WorkBasket.load_from_session(valid_user_client.session)
assert session_workbasket.id != workbasket.id
assert session_workbasket.status == WorkflowStatus.EDITING
assert (session_workbasket in other_baskets) == should_reuse
# check that the footnote edit is in the new session workbasket
assert session_workbasket.transactions.count() == 1
tx = session_workbasket.transactions.first()
assert tx.tracked_models.count() == 1
new_footnote_version = tx.tracked_models.first()
assert new_footnote_version.pk != footnote.pk
assert new_footnote_version.version_group == footnote.version_group
def test_submit_workbasket(
mock_upload,
approved_transaction,
unapproved_transaction,
valid_user,
client,
):
workbasket = unapproved_transaction.workbasket
assert_workbasket_valid(workbasket)
url = reverse(
"workbaskets:workbasket-ui-submit",
kwargs={"pk": workbasket.pk},
)
client.force_login(valid_user)
response = client.get(url)
assert response.status_code == 302
assert response.url == reverse("index")
workbasket.refresh_from_db()
assert workbasket.approver is not None
assert "workbasket" not in client.session
mock_upload.delay.assert_called_once_with()
def test_workbasket_accepted_updates_current_tracked_models(
upload,
new_workbasket,
valid_user,
):
original_footnote = factories.FootnoteFactory.create()
new_footnote = original_footnote.new_version(
workbasket=new_workbasket,
update_type=UpdateType.UPDATE,
)
assert new_footnote.version_group.current_version.pk == original_footnote.pk
assert_workbasket_valid(new_workbasket)
new_workbasket.submit_for_approval()
new_footnote.refresh_from_db()
assert new_footnote.version_group.current_version.pk == original_footnote.pk
new_workbasket.approve(valid_user.pk, "SEED_FIRST")
new_footnote.refresh_from_db()
assert new_footnote.version_group.current_version.pk == new_footnote.pk