Switch to pathlib

This commit is contained in:
2017-06-23 20:05:10 -07:00
parent 8d8c3b84fd
commit 4b7feeebb1
2 changed files with 29 additions and 27 deletions

View File

@@ -1,21 +1,22 @@
#!/usr/bin/env python3
import unittest
import os.path
from pathlib import Path
import os
import json
import blenderpack
class test_blenderpack_make_repo(unittest.TestCase):
helper_path = os.path.join('tests', 'test_helpers')
addon_path = os.path.join(helper_path, 'addons')
helper_path = Path('tests', 'test_helpers')
addon_path = Path(helper_path, 'addons')
def test_extract_blinfo_from_nonexistent(self):
test_file = 'file_that_doesnt_exist'
self.assertRaises(
FileNotFoundError,
blenderpack.extract_blinfo,
os.path.join(self.addon_path, test_file)
Path(self.addon_path, test_file)
)
def test_extract_blinfo_from_nonaddon(self):
@@ -23,22 +24,22 @@ class test_blenderpack_make_repo(unittest.TestCase):
self.assertRaises(
blenderpack.BadAddon,
blenderpack.extract_blinfo,
os.path.join(self.addon_path, test_file)
Path(self.addon_path, test_file)
)
def test_make_repo_valid(self):
blenderpack.make_repo(os.path.join(self.helper_path, 'addons'))
repojson = os.path.join(self.helper_path, 'addons', 'repo.json')
blenderpack.make_repo(Path(self.helper_path, 'addons'))
repojson = Path(self.helper_path, 'addons', 'repo.json')
self.assertTrue(os.path.exists(repojson))
with open(repojson, 'r') as f:
self.assertTrue(repojson.is_file())
with repojson.open('r') as f:
json.loads(f.read())
os.remove(repojson)
self.fail('unfinished test')
def test_make_repo_from_nonexistent(self):
blenderpack.make_repo(os.path.join(self.helper_path, 'addons'))
blenderpack.make_repo(Path(self.helper_path, 'addons'))
self.fail('unfinished test')
@@ -51,8 +52,8 @@ bl_info_tests = {
def generate_test(test_file):
def test(self):
reality = str(blenderpack.extract_blinfo(os.path.join(self.addon_path, test_file)))
with open(os.path.join(self.helper_path, test_file + '_output'), 'r') as f:
reality = str(blenderpack.extract_blinfo(Path(self.addon_path, test_file)))
with (self.helper_path / (test_file + '_output')).open() as f:
expectation = f.read()
self.assertEqual(expectation, reality)
return test