from os import sys, mkdir, path, system, chdir, remove, rename
from subprocess import check_call
from sys import exit, stdout, stderr
import wget
from filecmp import cmp
def test(name, ext, lenient=False):
print('{}.{}'.format(name, ext))
if path.exists('{}.{}.yaml'.format(name, ext)):
remove('{}.{}.yaml'.format(name, ext))
if path.exists('{}.test.{}'.format(name, ext)):
remove('{}.test.{}'.format(name, ext))
if path.exists('{}.test.{}.yaml'.format(name, ext)):
remove('{}.test.{}.yaml'.format(name, ext))
if lenient:
check_call(['cargo', 'run', '--', '-p', 'ru', '-kvdf', '{}.{}'.format(name, ext)], stdout=stdout, stderr=stderr)
rename('{}.{}.yaml'.format(name, ext), '{}.fit.{}.yaml'.format(name, ext))
check_call(['cargo', 'run', '--', '-p', 'ru', '-vf', '{}.fit.{}.yaml'.format(name, ext)], stdout=stdout, stderr=stderr)
source = '{}.fit.{}'.format(name, ext)
else:
source = '{}.{}'.format(name, ext)
check_call(['cargo', 'run', '--', '-p', 'ru', '-kvd', source], stdout=stdout, stderr=stderr)
rename('{}.yaml'.format(source), '{}.test.{}.yaml'.format(name, ext))
check_call(['cargo', 'run', '--', '-p', 'ru', '-kv', '{}.test.{}.yaml'.format(name, ext)], stdout=stdout, stderr=stderr)
if not cmp(source, '{}.test.{}'.format(name, ext)):
exit(1)
cd = path.dirname(path.realpath(__file__))
chdir(cd)
if not path.exists('test_data'):
mkdir('test_data')
chdir("test_data")
if not path.exists('Data Files/Aleanne Armor and Clothes 1+2.esp'):
wget.download('http://www.fullrest.ru/files/ale-clothing-v1-1c/files?fid=2379', 'ale-clothing-v1-1c.7z')
system('7za x ale-clothing-v1-1c.7z')
if not path.exists('Data Files/TravelingMerchants-1.2_1C.esp'):
wget.download('http://www.fullrest.ru/uploads/files/TravelingMerchants-1.2_1C.rar', 'TravelingMerchants-1.2_1C.rar')
system('unrar x -y TravelingMerchants-1.2_1C.rar')
if not path.exists('Data Files/Animal Behaviour.esp'):
wget.download('https://www.fullrest.ru/uploads/files/AnimalRealism.rar', 'AnimalRealism.rar')
system('unrar x -y AnimalRealism.rar')
if not path.exists('Data Files/Morrowind.esm'):
print("Put Morrowind.esm into test_data/Data Files")
exit(1)
if not path.exists('Data Files/Tribunal.esm'):
print("Put Tribunal.esm into test_data/Data Files")
exit(1)
if not path.exists('Data Files/Bloodmoon.esm'):
print("Put Bloodmoon.esm into test_data/Data Files")
exit(1)
test('Saves/Alchemy0000', 'ess')
test('Saves/F0000', 'ess')
test('Data Files/Animal Behaviour', 'esp', lenient=True)
test('Data Files/Aleanne Armor and Clothes 1+2', 'esp')
test('Data Files/TravelingMerchants-1.2_1C', 'esp')
test('Data Files/Morrowind', 'esm')
test('Data Files/Tribunal', 'esm')
test('Data Files/Bloodmoon', 'esm')
print("All tests passed.")