test: fix test cases

This commit is contained in:
Himanshu Warekar 2019-09-15 20:38:17 +05:30
parent d017610677
commit cc74ab39a0
2 changed files with 23 additions and 27 deletions

View file

@ -9,31 +9,6 @@ from frappe.exceptions import ValidationError
class TestContact(unittest.TestCase):
def create_contact(self, name, salutation, emails=None, phones=None, save=True):
doc = frappe.get_doc({
"doctype": "Contact",
"first_name": name,
"status": "Open",
"salutation": salutation
})
if emails:
for d in emails:
doc.add_email(d.get("email"), d.get("is_primary"))
if phones:
for d in phones:
doc.add_phone(d.get("phone"), d.get("is_primary_phone"), d.get("is_primary_mobile_no"))
if save:
doc.insert()
return doc
def tearDown(self):
frappe.db.sql("delete from `tabContact`")
frappe.db.sql("delete from `tabContact Phone`")
frappe.db.sql("delete from `tabContact Email`")
def test_check_default_email(self):
emails = [
{"email": "test1@example.com", "is_primary": 0},
@ -87,4 +62,25 @@ class TestContact(unittest.TestCase):
self.assertRaises(ValidationError, contact_email.save)
# No default set for phones if many phones are passed as params
self.assertRaises(ValidationError, contact_phone.save)
self.assertRaises(ValidationError, contact_phone.save)
def create_contact(name, salutation, emails=None, phones=None, save=True):
doc = frappe.get_doc({
"doctype": "Contact",
"first_name": name,
"status": "Open",
"salutation": salutation
})
if emails:
for d in emails:
doc.add_email(d.get("email"), d.get("is_primary"))
if phones:
for d in phones:
doc.add_phone(d.get("phone"), d.get("is_primary_phone"), d.get("is_primary_mobile_no"))
if save:
doc.insert()
return doc

View file

@ -83,7 +83,7 @@ def create_linked_contact(link_list, address):
"status": "Open"
})
contact.add_email("test_contact@example.com")
contact.add_phone("+91 0000000000")
contact.add_phone("+91 0000000000", is_primary_phone=True)
for name in link_list:
contact.append("links",{