Add tests for python.to_lorry

2 files changed, 73 insertions, 1 deletions

diff --git a/baserockimport/exts/python.to_lorry b/baserockimport/exts/python.to_lorry
index e39c8d1..b7341ca 100755
--- a/baserockimport/exts/python.to_lorry
+++ b/baserockimport/exts/python.to_lorry
@@ -31,7 +31,7 @@ import select
 
 import pkg_resources
 
-from importer_pip_common import *
+from importer_python_common import *
 
 def fetch_package_metadata(package_name):
     try:
diff --git a/baserockimport/exts/python_lorry_tests.py b/baserockimport/exts/python_lorry_tests.py
new file mode 100755
index 0000000..12ef564
--- /dev/null
+++ b/baserockimport/exts/python_lorry_tests.py
@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+# Copyright (C) 2014  Codethink Limited
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; version 2 of the License.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 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
+
+class Tests(unittest.TestCase):
+
+    def test_make_tarball_lorry(self):
+        gzip, bzip, lzma = 'gzip', 'bzip2', 'lzma'
+
+        valid_extensions = {'tar.gz': gzip, 'tgz': gzip, 'tar.Z': gzip,
+                            'tar.bz2': bzip, 'tbz2': bzip,
+                            'tar.lzma': lzma, 'tar.xz': lzma,
+                            'tlz': lzma, 'txz': lzma}
+
+        def make_url(extension):
+            return 'http://foobar/baz.%s' % extension
+
+        def get_tarball_lorry_url(name, lorry_json):
+            return json.loads(lorry_json)['python-packages/'
+                                            + name + '-tarball']['url']
+
+        def get_tarball_lorry_compression(name, lorry_json):
+            return json.loads(lorry_json)['python-packages/'
+                                            + name + '-tarball']['compression']
+
+        fake_package_name = 'name'
+        urls = [(make_url(ext), ext) for ext in valid_extensions]
+
+        for (url, ext) in urls:
+            lorry_json = python_lorry.make_tarball_lorry('name', url)
+            print lorry_json
+
+            tarball_url = get_tarball_lorry_url(fake_package_name, lorry_json)
+            print 'Tarball url: %s' % tarball_url
+
+            self.assertEqual(tarball_url, url)
+
+            tarball_compression = get_tarball_lorry_compression(
+                                        fake_package_name, lorry_json)
+
+            print 'Tarball compression: %s' % tarball_compression
+            self.assertEqual(tarball_compression, valid_extensions[ext])
+
+        url = 'http://foobar/baz.tar'
+        lorry_json = python_lorry.make_tarball_lorry('name', url)
+        self.assertEqual(get_tarball_lorry_url(fake_package_name,
+                                               lorry_json), url)
+        self.assertTrue('compression' not in lorry_json)
+
+
+if __name__ == '__main__':
+    suite = unittest.TestLoader().loadTestsFromTestCase(Tests)
+    unittest.TextTestRunner(verbosity=2).run(suite)