diff options
Diffstat (limited to 'baserockimport/exts/python_lorry_tests.py')
-rwxr-xr-x | baserockimport/exts/python_lorry_tests.py | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/baserockimport/exts/python_lorry_tests.py b/baserockimport/exts/python_lorry_tests.py index f57be6d..298f9b0 100755 --- a/baserockimport/exts/python_lorry_tests.py +++ b/baserockimport/exts/python_lorry_tests.py @@ -17,18 +17,17 @@ # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. -import imp -python_lorry = imp.load_source('python_lorry', 'python.to_lorry') - import json - import unittest +import utils + LORRY_PREFIX = 'python' +EXTNAME = 'python' class Tests(unittest.TestCase): - def test_make_tarball_lorry(self): + def test_str_tarball_lorry(self): gzip, bzip, lzma = 'gzip', 'bzip2', 'lzma' valid_extensions = {'tar.gz': gzip, 'tgz': gzip, 'tar.Z': gzip, @@ -51,8 +50,8 @@ class Tests(unittest.TestCase): urls = [(make_url(ext), ext) for ext in valid_extensions] for (url, ext) in urls: - lorry_json = python_lorry.make_tarball_lorry(LORRY_PREFIX, - 'name', url) + lorry_json = utils.str_tarball_lorry(EXTNAME, LORRY_PREFIX, + 'name', url) print lorry_json tarball_url = get_tarball_lorry_url(fake_package_name, lorry_json) @@ -67,7 +66,8 @@ class Tests(unittest.TestCase): self.assertEqual(tarball_compression, valid_extensions[ext]) url = 'http://foobar/baz.tar' - lorry_json = python_lorry.make_tarball_lorry(LORRY_PREFIX, 'name', url) + lorry_json = utils.str_tarball_lorry(EXTNAME, LORRY_PREFIX, + 'name', url) self.assertEqual(get_tarball_lorry_url(fake_package_name, lorry_json), url) self.assertTrue('compression' not in lorry_json) |