Skip to content

Commit

Permalink
test: type cast date values in onboarding tests (backport #1889) (#1892)
Browse files Browse the repository at this point in the history
(cherry picked from commit 1ae53f0)

Co-authored-by: Rucha Mahabal <ruchamahabal2@gmail.com>
  • Loading branch information
mergify[bot] and ruchamahabal committed Jun 12, 2024
1 parent 9e0f9ce commit ca1c74c
Showing 1 changed file with 11 additions and 10 deletions.
21 changes: 11 additions & 10 deletions hrms/hr/doctype/employee_onboarding/test_employee_onboarding.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,19 +37,15 @@ def test_employee_onboarding_incomplete_task(self):
self.assertEqual(onboarding.boarding_status, "Pending")

# start and end dates
start_date, end_date = frappe.db.get_value(
"Task", onboarding.activities[0].task, ["exp_start_date", "exp_end_date"]
)
self.assertEqual(getdate(start_date), getdate(onboarding.boarding_begins_on))
self.assertEqual(getdate(end_date), add_days(start_date, onboarding.activities[0].duration))
start_date, end_date = get_task_dates(onboarding.activities[0].task)
self.assertEqual(start_date, onboarding.boarding_begins_on)
self.assertEqual(end_date, add_days(start_date, onboarding.activities[0].duration))

start_date, end_date = frappe.db.get_value(
"Task", onboarding.activities[1].task, ["exp_start_date", "exp_end_date"]
)
start_date, end_date = get_task_dates(onboarding.activities[1].task)
self.assertEqual(
getdate(start_date), add_days(onboarding.boarding_begins_on, onboarding.activities[0].duration)
start_date, add_days(onboarding.boarding_begins_on, onboarding.activities[0].duration)
)
self.assertEqual(getdate(end_date), add_days(start_date, onboarding.activities[1].duration))
self.assertEqual(end_date, add_days(start_date, onboarding.activities[1].duration))

# complete the task
project = frappe.get_doc("Project", onboarding.project)
Expand Down Expand Up @@ -154,3 +150,8 @@ def create_employee_onboarding():
onboarding.submit()

return onboarding


def get_task_dates(task: str) -> tuple[str, str]:
start_date, end_date = frappe.db.get_value("Task", task, ["exp_start_date", "exp_end_date"])
return getdate(start_date), getdate(end_date)

0 comments on commit ca1c74c

Please sign in to comment.