From 96c24e8e5fb24215008a14de1ad0ecf977499686 Mon Sep 17 00:00:00 2001 From: Jannat Patel Date: Thu, 17 Feb 2022 14:56:33 +0530 Subject: [PATCH] test: certificate with expiry --- .../doctype/lms_certification/test_lms_certification.py | 9 +++++---- school/lms/doctype/lms_course/test_lms_course.py | 5 +++-- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/school/lms/doctype/lms_certification/test_lms_certification.py b/school/lms/doctype/lms_certification/test_lms_certification.py index 1c800306..1585c79f 100644 --- a/school/lms/doctype/lms_certification/test_lms_certification.py +++ b/school/lms/doctype/lms_certification/test_lms_certification.py @@ -5,17 +5,18 @@ import frappe import unittest from school.lms.doctype.lms_course.test_lms_course import new_course from school.lms.doctype.lms_certification.lms_certification import create_certificate +from frappe.utils import nowdate, add_years, cint class TestLMSCertification(unittest.TestCase): def test_certificate_creation(self): - course = new_course("Test Certificate", 1) + course = new_course("Test Certificate", 1, 2) certificate = create_certificate(course.name) self.assertEqual(certificate.student, "Administrator") self.assertEqual(certificate.course, course.name) - self.assertEqual(certificate.issue_date, frappe.utils.nowdate()) - self.assertEqual(certificate.expiry_date, None) + self.assertEqual(certificate.issue_date, nowdate()) + self.assertEqual(certificate.expiry_date, add_years(nowdate(), cint(course.expiry))) - frappe.db.delete("LMS Course", course.name) frappe.db.delete("LMS Certification", certificate.name) + frappe.db.delete("LMS Course", course.name) diff --git a/school/lms/doctype/lms_course/test_lms_course.py b/school/lms/doctype/lms_course/test_lms_course.py index 7b592d69..64d27695 100644 --- a/school/lms/doctype/lms_course/test_lms_course.py +++ b/school/lms/doctype/lms_course/test_lms_course.py @@ -59,13 +59,14 @@ def new_user(name, email): doc.insert() return doc -def new_course(title, certificate=0): +def new_course(title, certificate=0, expiry=0): doc = frappe.get_doc({ "doctype": "LMS Course", "title": title, "short_introduction": title, "description": title, - "enable_certificate": certificate + "enable_certificate": certificate, + "expiry": expiry }) doc.insert(ignore_permissions=True) return doc