Skip to content

Commit 276b25f

Browse files
committed
Automatically disable the CSP for demo_mode or MasterQA
1 parent 2d13833 commit 276b25f

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

seleniumbase/fixtures/base_case.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2458,6 +2458,10 @@ def get_new_driver(self, browser=None, headless=None,
24582458
user_agent = agent
24592459
if user_agent is None:
24602460
user_agent = self.user_agent
2461+
if disable_csp is None:
2462+
disable_csp = self.disable_csp
2463+
if self.demo_mode or self.masterqa_mode:
2464+
disable_csp = True
24612465
if cap_file is None:
24622466
cap_file = self.cap_file
24632467
valid_browsers = constants.ValidBrowsers.valid_browsers
@@ -2799,12 +2803,13 @@ def __highlight_with_jquery_2(self, message, selector, o_bs):
27992803

28002804
############
28012805

2802-
def setUp(self):
2806+
def setUp(self, masterqa_mode=False):
28032807
"""
28042808
Be careful if a subclass of BaseCase overrides setUp()
28052809
You'll need to add the following line to the subclass setUp() method:
28062810
super(SubClassOfBaseCase, self).setUp()
28072811
"""
2812+
self.masterqa_mode = masterqa_mode
28082813
self.is_pytest = None
28092814
try:
28102815
# This raises an exception if the test is not coming from pytest

seleniumbase/masterqa/master_qa.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -392,7 +392,7 @@ class MasterQA(__MasterQATestCase__):
392392
def setUp(self):
393393
self.check_count = 0
394394
self.auto_close_results_page = False
395-
super(__MasterQATestCase__, self).setUp()
395+
super(__MasterQATestCase__, self).setUp(masterqa_mode=True)
396396
self.manual_check_setup()
397397
if self.headless:
398398
self.auto_close_results_page = True

0 commit comments

Comments
 (0)