centralcloud/oncall-engine
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix tests

Browse source
This commit is contained in:
Julia 2022-08-18 18:27:34 +03:00
parent febe4cc9e0
commit 32daaf74ce
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
engine/apps/api/tests/test_oncall_shift.py
View file

@ -455,11 +455,17 @@ def test_update_old_on_call_shift_with_future_version(
response = client.put(url, data=data_to_update, format="json", **make_user_auth_headers(user1, token))
next_shift_start_date = timezone.datetime.combine(next_rotation_start_date.date(), start_date.time()).astimezone(
timezone.pytz.UTC
)
expected_payload = data_to_update | {
"id": new_on_call_shift.public_primary_key,
"type": CustomOnCallShift.TYPE_ROLLING_USERS_EVENT,
"schedule": schedule.public_primary_key,
"updated_shift": None,
"shift_start": next_shift_start_date.strftime("%Y-%m-%dT%H:%M:%SZ"),
"shift_end": (next_shift_start_date + updated_duration).strftime("%Y-%m-%dT%H:%M:%SZ"),
}
assert response.status_code == status.HTTP_200_OK

2
engine/apps/schedules/models/custom_on_call_shift.py
View file

@ -558,7 +558,7 @@ class CustomOnCallShift(models.Model):
instance_data["start"] = timezone.datetime.combine(
instance_data["rotation_start"].date(),
instance_data["start"].time(),
)
).astimezone(pytz.UTC)
if self.last_updated_shift is None or self.last_updated_shift.event_is_started:
# create new shift