1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
# The MIT License (MIT)
# Copyright (c) Microsoft Corporation. All rights reserved.
import test_config
from azure.cosmos import CosmosClient as CosmosSyncClient
cosmos_sync_client = CosmosSyncClient(test_config.TestConfig.host, test_config.TestConfig.masterKey)
def pytest_configure(config):
"""
Allows plugins and conftest files to perform initial configuration.
This hook is called for every plugin and initial conftest
file after command line options have been parsed.
"""
def pytest_sessionstart(session):
"""
Called after the Session object has been created and
before performing collection and entering the run test loop.
"""
config = test_config.TestConfig
config.create_database_if_not_exist(cosmos_sync_client)
config.create_single_partition_container_if_not_exist(cosmos_sync_client)
config.create_multi_partition_container_if_not_exist(cosmos_sync_client)
config.create_single_partition_prefix_pk_container_if_not_exist(cosmos_sync_client)
config.create_multi_partition_prefix_pk_container_if_not_exist(cosmos_sync_client)
def pytest_sessionfinish(session, exitstatus):
"""
Called after whole test run finished, right before
returning the exit status to the system.
"""
config = test_config.TestConfig
config.try_delete_database(cosmos_sync_client)
def pytest_unconfigure(config):
"""
called before test process is exited.
"""
|
