def test_strategies(test_data):
# Check each case
base_dir = os.path.join(os.path.realpath(os.path.dirname(__file__)),
'cases')
print('Base directory: %s' % base_dir)
for item in os.listdir(base_dir):
if os.path.isdir(os.path.join(base_dir, item)):
# Enter a directory
print("Entering directory '%s'..." % item)
for conf_file in os.listdir(os.path.join(base_dir, item)):
conf_path = os.path.join(base_dir, item, conf_file)
if os.path.isfile(conf_path):
# Load file
print('Loading file: %s' % conf_file)
with open(conf_path, encoding='utf-8') as f:
conf = yaml.safe_load(f)
# Make strategy and run
stgy = Strategy(conf_file, conf['test'])
stgy.execute(test_data)
# Check result
if 'remain' in conf and set([x.name for x in stgy.remain_list]) \
!= set(conf['remain'] if conf['remain'] is not None else []) \
or \
'remove' in conf and set([x.name for x in stgy.remove_list]) \
!= set(conf['remove'] if conf['remove'] is not None else []):
raise AssertionError()
# Leave the directory
print("Leaving directory '%s'..." % item)
def test_strategies(mocker, test_data, test_env):
# Logger
lg = logger.Logger.register(__name__)
# Exceptions mapping
exception_map = {
IllegalCharacter: 'IllegalCharacter',
ConditionSyntaxError: 'ConditionSyntaxError',
NoSuchCondition: 'NoSuchCondition'
}
# Check each case
base_dir = os.path.join(os.path.realpath(os.path.dirname(__file__)),
'cases')
lg.info('Base directory: %s' % base_dir)
for item in os.listdir(base_dir):
if os.path.isdir(os.path.join(base_dir, item)):
# Enter a directory
lg.info("Entering directory '%s'..." % item)
for conf_file in os.listdir(os.path.join(base_dir, item)):
conf_path = os.path.join(base_dir, item, conf_file)
if os.path.isfile(conf_path):
# Load file
lg.info('Loading file: %s' % conf_file)
with _open(conf_path, encoding='utf-8') as f:
conf = yaml.safe_load(f)
try:
# Make strategy and run
mocker.patch('time.time',
return_value=test_env['time.time'])
mocker.patch(
'shutil.disk_usage',
return_value=test_env['shutil.disk_usage'])
stgy = Strategy(conf_file, conf['test'])
stgy.execute(test_data)
# Check result
if 'remain' in conf:
assert set([
x.name for x in stgy.remain_list
]) == set(conf['remain']
if conf['remain'] is not None else [])
if 'remove' in conf:
assert set([
x.name for x in stgy.remove_list
]) == set(conf['remove']
if conf['remove'] is not None else [])
except Exception as e:
if 'exceptions' in conf and exception_map[
sys.exc_info()[0]] in conf['exceptions']:
pass
else:
raise e
# Leave the directory
lg.info("Leaving directory '%s'..." % item)