tests: fix tests

This commit is contained in:
Himanshu Warekar 2019-09-20 14:22:40 +05:30
parent cc8953ad66
commit a0a57f4a5d

View file

@ -7,7 +7,7 @@ import frappe
import unittest
import json
from frappe.website.doctype.personal_data_download_request.personal_data_download_request import get_user_data
from frappe.contacts.doctype.contact.contact import get_contact_name
class TestRequestPersonalData(unittest.TestCase):
def setUp(self):
@ -18,7 +18,8 @@ class TestRequestPersonalData(unittest.TestCase):
def test_user_data_creation(self):
user_data = json.loads(get_user_data('test_privacy@example.com'))
expected_data = {'Contact': frappe.get_all('Contact', {'email_id':'test_privacy@example.com'}, ["*"])}
contact_name = get_contact_name('test_privacy@example.com')
expected_data = {'Contact': frappe.get_all('Contact', contact_name, ["*"])}
expected_data = json.loads(json.dumps(expected_data, default=str))
self.assertEqual({'Contact': user_data['Contact']}, expected_data)
@ -48,8 +49,7 @@ class TestRequestPersonalData(unittest.TestCase):
frappe.db.sql("delete from `tabEmail Queue`")
def create_user_if_not_exists(email, first_name = None):
if frappe.db.exists("User", email):
return
frappe.delete_doc_if_exists("User", email)
frappe.get_doc({
"doctype": "User",