def setup(self, driver, config, request: FixtureRequest):
self.driver = driver
self.config = config
self.start_page: StartPage = request.getfixturevalue('start_page')
self.registration_page: RegistrationPage = request.getfixturevalue('registration_page')
self.cabinet_page: CabinetPage = request.getfixturevalue('cabinet_page')
def setup(self, driver, config, request: FixtureRequest):
self.driver = driver
self.config = config
self.base_page: BasePage = request.getfixturevalue('base_page')
self.main_page: MainPage = request.getfixturevalue('main_page')
self.segment_page: SegmentPage = request.getfixturevalue(
'segment_page')
def setup(self, driver, config, request: FixtureRequest):
self.driver = driver
self.config = config
self.base_page: BasePage = request.getfixturevalue('base_page')
self.login_page: LoginPage = request.getfixturevalue('login_page')
self.reg_page: RegPage = request.getfixturevalue('reg_page')
self.main_page: MainPage = request.getfixturevalue('main_page')
def spark_session(request: FixtureRequest) -> SparkSession:
# make sure env variables are set correctly
if "SPARK_HOME" not in os.environ:
os.environ["SPARK_HOME"] = "/usr/local/opt/spark"
clean_spark_dir()
master = "local[2]"
session = (
SparkSession.builder.appName(
f"pytest-pyspark-local-testing-{get_random_string(4)}"
)
.master(master)
.config("spark.ui.showConsoleProgress", "false")
.config("spark.sql.shuffle.partitions", "2")
.config("spark.default.parallelism", "4")
.config("spark.driver.bindAddress", "127.0.0.1")
.config("spark.sql.broadcastTimeout", "2400")
.enableHiveSupport()
.getOrCreate()
)
request.addfinalizer(lambda: clean_close(session))
quiet_py4j()
return session
def spark_session(request: FixtureRequest) -> SparkSession:
# make sure env variables are set correctly
if 'SPARK_HOME' not in os.environ:
os.environ['SPARK_HOME'] = '/usr/local/opt/spark'
clean_spark_dir()
master = "local[2]"
# master: str = "spark://localhost:7077"
if not path.exists("/Applications/Docker.app"):
master = "local[2]"
print(f"++++++ Running on local spark: {master} ++++")
else:
print(f"++++++ Running on docker spark: {master} ++++")
session = SparkSession.builder.appName(f"pytest-pyspark-local-testing-{get_random_string(4)}"). \
master(master). \
config("spark.ui.showConsoleProgress", "false"). \
config("spark.sql.shuffle.partitions", "2"). \
config("spark.default.parallelism", "4"). \
config("spark.sql.broadcastTimeout", "2400"). \
enableHiveSupport(). \
getOrCreate()
request.addfinalizer(lambda: clean_close(session))
quiet_py4j()
return session
def database(app: Flask, request: FixtureRequest) -> SQLAlchemy:
if os.path.exists(TEST_DATABASE_PATH):
os.unlink(TEST_DATABASE_PATH)
db.app = app
db.create_all()
# Bypass pysqlite's broken transactions (see https://bit.ly/2DKiixa).
# pylint:disable=unused-argument
# pylint:disable=unused-variable
@event.listens_for(db.engine, 'connect')
def do_connect(connection: Connection, *args) -> None:
connection.isolation_level = None
# pylint:disable=unused-variable
@event.listens_for(db.engine, 'begin')
def do_begin(connection: Connection) -> None:
connection.execute('BEGIN')
def teardown() -> None:
db.drop_all()
os.unlink(TEST_DATABASE_PATH)
request.addfinalizer(teardown)
return db
def setup(self, driver, config, request: FixtureRequest):
self.driver = driver
self.config = config
self.basepage: BasePage = request.getfixturevalue('base_page')
self.homepage: HomePage = request.getfixturevalue('home_page')
self.segmentpage: SegmentPage = request.getfixturevalue('segment_page')
self.createcompanypage: CreateCompanyPage = request.getfixturevalue('create_company_page')
def initialize_tests(request: FixtureRequest) -> None:
initializer(
["eventual_tortoise.relation"],
db_url="sqlite://:memory:",
app_label="default",
)
request.addfinalizer(finalizer)
def never_sleep(request: FixtureRequest) -> None:
patcher = mock.patch("asyncio.sleep", _nosleep)
patcher.start()
def tearDown() -> None:
patcher.stop()
request.addfinalizer(tearDown)
def stop_std_logging(request: FixtureRequest, capfd: CaptureFixture) -> None:
patcher = mock.patch("bandersnatch.log.setup_logging")
patcher.start()
def tearDown() -> None:
patcher.stop()
request.addfinalizer(tearDown)
def setup(self, driver, request: FixtureRequest):
self.driver = driver
self.auth_page: AuthPage = request.getfixturevalue('auth_page')
self.company_page: CompanyPage = request.getfixturevalue(
'company_page')
self.segment_page: SegmentPage = request.getfixturevalue(
'segment_page')
def setup(self, driver, config, request: FixtureRequest):
self.driver = driver
self.config = config
self.base_page: BasePage = request.getfixturevalue('base_page')
self.authorization_page: AuthorizationPage = request.getfixturevalue(
'authorization_page')(login="******",
passwd="Barsa2016")
def setup(self, driver, config, request: FixtureRequest):
self.driver = driver
self.config = config
self.main_page = request.getfixturevalue('main_page')
self.dashboard = request.getfixturevalue('dashboard_page')
self.campaign = request.getfixturevalue('campaign_page')
self.audience = request.getfixturevalue('audience_page')
def setup(self,driver,config,request:FixtureRequest):
self.driver = driver
self.config = config
self.base_page: BasePage = request.getfixturevalue('base_page')
self.cabinet_page: CabinetPage = request.getfixturevalue('cabinet_page')
self.audience_page : AudiencePage = request.getfixturevalue('audience_page')
self.campaign_page : CampaignPage = request.getfixturevalue('campaign_page')
def setup(self, driver, config, request: FixtureRequest):
self.driver = driver
self.config = config
self.base_page: BasePage = request.getfixturevalue('base_page')
self.main_page: MainPage = request.getfixturevalue('main_page')
self.account_page: AccountPage = request.getfixturevalue(
'account_page')(email="*****@*****.**", password="******")
def setup(self, driver, config, request: FixtureRequest):
self.driver = driver
self.config = config
self.base_page: BasePage = request.getfixturevalue('base_page')
self.main_page: MainPage = request.getfixturevalue('main_page')
self.search_page: SearchPage = request.getfixturevalue('search_page')
self.euro_python_events_page: EuroPythonEventsPage = request.getfixturevalue('euro_python_events_page')
def selenium(request: FixtureRequest) -> webdriver.Remote:
if request.config.getoption('--local-chrome'):
driver = initialize_local_chrome()
request.addfinalizer(lambda: deinitialize_selenium(driver))
return driver
driver = request.getfixturevalue('selenium') # type: webdriver.Remote
yield driver
# https://github.com/SeleniumHQ/docker-selenium/issues/87#issuecomment-286231268
driver.close()
def selenium(request: FixtureRequest) -> webdriver.Remote:
if request.config.getoption('--local-chrome'):
driver = initialize_local_chrome()
request.addfinalizer(lambda: deinitialize_selenium(driver))
return driver
driver = request.getfixturevalue('selenium') # type: webdriver.Remote
yield driver
# https://github.com/SeleniumHQ/docker-selenium/issues/87#issuecomment-286231268
driver.close()
def job_folder(request: FixtureRequest) -> None:
"""Create job folder and add finalizer to remove the folders after running the test."""
if not os.path.isdir(settings.JOB_FOLDER):
os.makedirs(settings.JOB_FOLDER)
def fin() -> None:
shutil.rmtree(settings.JOB_FOLDER)
request.addfinalizer(fin)
def logging_mock(request: FixtureRequest) -> mock.MagicMock:
patcher = mock.patch("logging.config.fileConfig")
logger: mock.MagicMock = patcher.start()
def tearDown() -> None:
patcher.stop()
request.addfinalizer(tearDown)
return logger
def mirror_mock(request: FixtureRequest) -> mock.MagicMock:
patcher = mock.patch("bandersnatch.mirror.BandersnatchMirror")
mirror: mock.MagicMock = patcher.start()
def tearDown() -> None:
patcher.stop()
request.addfinalizer(tearDown)
return mirror
def setup(self, driver, config, request: FixtureRequest):
self.driver = driver
self.config = config
self.base_page: BasePage = request.getfixturevalue('base_page')
self.modal_login: ModalLogin = request.getfixturevalue('modal_login')
self.main_page: MainPage = request.getfixturevalue('main_page')
self.create_campaign_page: CreateCampaignPage = request.getfixturevalue(
'create_campaign_page')
def dag_folder(request: FixtureRequest) -> None:
"""Create dag folder and add finalizer to remove it again after running the test."""
folder = get_dags_folder()
if not os.path.isdir(folder):
os.mkdir(folder)
def fin() -> None:
shutil.rmtree(folder)
request.addfinalizer(fin)
def sync_result(request: FixtureRequest) -> None:
"""Create sync-results and add finalizer to remove the folders after running the test."""
# For sync job test
if not os.path.isdir(settings.SYNC_RESULTS_FOLDER):
os.makedirs(settings.SYNC_RESULTS_FOLDER)
def fin() -> None:
shutil.rmtree(settings.SYNC_RESULTS_FOLDER)
request.addfinalizer(fin)
def head_uc(request: FixtureRequest):
request.instance.driver = uc.Chrome()
def teardown():
request.instance.driver.save_screenshot(FAILED_SCREENSHOT_NAME)
request.instance.driver.quit()
request.addfinalizer(teardown)
return request.instance.driver
def fs(self, request: FixtureRequest,
new_fs: "TestBuildHashEntryPath.FS") -> "TestBuildHashEntryPath.FS":
new_fs.prev_dir = os.getcwd()
def pop_lvl1():
os.chdir(new_fs.prev_dir)
os.chdir(new_fs.level1.realpath())
request.addfinalizer(pop_lvl1)
return new_fs
def start_server_process(request: FixtureRequest):
proc = Process(target=start_app)
proc.daemon = True
proc.start()
time.sleep(2)
def fin():
proc.terminate()
request.addfinalizer(fin)
def setup(self, driver, config, request: FixtureRequest):
self.driver = driver
self.config = config
self.main_page: MainPage = request.getfixturevalue('main_page')
self.cabinet_page: CabinetPage = request.getfixturevalue(
'cabinet_page')
self.campaign_page: CampaignPage = request.getfixturevalue(
'campaign_page')
self.segment_page: SegmentPage = request.getfixturevalue(
'segment_page')
def _setup_fixtures(self):
def func():
pass
self.funcargs = {}
fm = self.session._fixturemanager
self._fixtureinfo = fm.getfixtureinfo(node=self, func=func,
cls=None, funcargs=False)
fixture_request = FixtureRequest(self)
fixture_request._fillfixtures()
return fixture_request
def quilc(request: FixtureRequest) -> None:
dock = docker.from_env()
container = dock.containers.run(
image="rigetti/quilc",
command="-S",
detach=True,
ports={5555: 5555},
remove=True,
)
request.addfinalizer(container.stop)
return None
def setup(self, driver, config, request: FixtureRequest):
self.driver = driver
self.config = config
self.base_page: BasePage = request.getfixturevalue('base_page')
self.login_page: LoginPage = request.getfixturevalue('login_page')
self.creator_page: CreatingCampaign = request.getfixturevalue(
'creator_page')
self.newsegment_page: NewSegment = request.getfixturevalue(
'segment_page_new')
self.segment_page: SegmentPage = request.getfixturevalue(
'segment_page_table')
def _setup_fixtures(doctest_item):
"""
Used by DoctestTextfile and DoctestItem to setup fixture information.
"""
def func():
pass
doctest_item.funcargs = {}
fm = doctest_item.session._fixturemanager
doctest_item._fixtureinfo = fm.getfixtureinfo(node=doctest_item, func=func,
cls=None, funcargs=False)
fixture_request = FixtureRequest(doctest_item)
fixture_request._fillfixtures()
return fixture_request
