test: certificate with expiry

This commit is contained in:
Jannat Patel
2022-02-17 14:56:33 +05:30
parent cc0c1cc320
commit 96c24e8e5f
2 changed files with 8 additions and 6 deletions

View File

@@ -5,17 +5,18 @@ import frappe
import unittest import unittest
from school.lms.doctype.lms_course.test_lms_course import new_course from school.lms.doctype.lms_course.test_lms_course import new_course
from school.lms.doctype.lms_certification.lms_certification import create_certificate from school.lms.doctype.lms_certification.lms_certification import create_certificate
from frappe.utils import nowdate, add_years, cint
class TestLMSCertification(unittest.TestCase): class TestLMSCertification(unittest.TestCase):
def test_certificate_creation(self): def test_certificate_creation(self):
course = new_course("Test Certificate", 1) course = new_course("Test Certificate", 1, 2)
certificate = create_certificate(course.name) certificate = create_certificate(course.name)
self.assertEqual(certificate.student, "Administrator") self.assertEqual(certificate.student, "Administrator")
self.assertEqual(certificate.course, course.name) self.assertEqual(certificate.course, course.name)
self.assertEqual(certificate.issue_date, frappe.utils.nowdate()) self.assertEqual(certificate.issue_date, nowdate())
self.assertEqual(certificate.expiry_date, None) 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 Certification", certificate.name)
frappe.db.delete("LMS Course", course.name)

View File

@@ -59,13 +59,14 @@ def new_user(name, email):
doc.insert() doc.insert()
return doc return doc
def new_course(title, certificate=0): def new_course(title, certificate=0, expiry=0):
doc = frappe.get_doc({ doc = frappe.get_doc({
"doctype": "LMS Course", "doctype": "LMS Course",
"title": title, "title": title,
"short_introduction": title, "short_introduction": title,
"description": title, "description": title,
"enable_certificate": certificate "enable_certificate": certificate,
"expiry": expiry
}) })
doc.insert(ignore_permissions=True) doc.insert(ignore_permissions=True)
return doc return doc