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
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)

View File

@@ -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