From a0a57f4a5d4712e348b5cc97daaa44bfa4e45435 Mon Sep 17 00:00:00 2001 From: Himanshu Warekar Date: Fri, 20 Sep 2019 14:22:40 +0530 Subject: [PATCH] tests: fix tests --- .../test_personal_data_download_request.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/frappe/website/doctype/personal_data_download_request/test_personal_data_download_request.py b/frappe/website/doctype/personal_data_download_request/test_personal_data_download_request.py index 1965999042..50a144b4e0 100644 --- a/frappe/website/doctype/personal_data_download_request/test_personal_data_download_request.py +++ b/frappe/website/doctype/personal_data_download_request/test_personal_data_download_request.py @@ -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",