vassili/seitime-frappe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix(test): unsubscribe doctypes in tearDown

Browse source
This commit is contained in:
Rucha Mahabal 2020-08-14 13:48:18 +05:30 • committed by Rushabh Mehta
parent 8d09b37308
commit bb0e8af0fb
1 changed files with 13 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
frappe/event_streaming/doctype/event_producer/test_event_producer.py
View file

@ -23,6 +23,8 @@ class TestEventProducer(unittest.TestCase):
def setUp(self):
create_event_producer(producer_url)
def tearDown(self):
unsubscribe_doctypes(producer_url)
def test_insert(self):
producer = get_remote_site()
@ -290,7 +292,12 @@ def get_mapping(mapping_name, local, remote, field_map):
def create_event_producer(producer_url):
if frappe.db.exists('Event Producer', producer_url):
event_producer = frappe.get_doc('Event Producer', producer_url)
for entry in event_producer.producer_doctypes:
entry.unsubscribe = 0
event_producer.save()
return
event_producer = frappe.new_doc('Event Producer')
event_producer.producer_doctypes = []
event_producer.producer_url = producer_url
@ -329,3 +336,9 @@ def get_remote_site():
frappe_authorization_source='Event Consumer'
)
return producer_site
def unsubscribe_doctypes(producer_url):
event_producer = frappe.get_doc('Event Producer', producer_url)
for entry in event_producer.producer_doctypes:
entry.unsubscribe = 1
event_producer.save()