From 7cba0a33cee244379cb2078b7c286c4de5c4f341 Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Mon, 17 Oct 2011 20:25:59 +0100 Subject: Add functions for creating and unpacking strata. --- morphlib/bins_tests.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) (limited to 'morphlib/bins_tests.py') diff --git a/morphlib/bins_tests.py b/morphlib/bins_tests.py index b3c13a64..30076120 100644 --- a/morphlib/bins_tests.py +++ b/morphlib/bins_tests.py @@ -76,3 +76,27 @@ class ChunkTests(unittest.TestCase): self.assertEqual(recursive_lstat(self.instdir), recursive_lstat(self.unpacked)) + +class StratumTests(unittest.TestCase): + + def setUp(self): + self.tempdir = tempfile.mkdtemp() + self.instdir = os.path.join(self.tempdir, 'inst') + self.stratum_file = os.path.join(self.tempdir, 'stratum') + self.unpacked = os.path.join(self.tempdir, 'unpacked') + + def tearDown(self): + shutil.rmtree(self.tempdir) + + def populate_instdir(self): + os.mkdir(self.instdir) + os.mkdir(os.path.join(self.instdir, 'bin')) + + def test_creates_and_unpacks_stratum_exactly(self): + self.populate_instdir() + morphlib.bins.create_stratum(self.instdir, self.stratum_file) + os.mkdir(self.unpacked) + morphlib.bins.unpack_stratum(self.stratum_file, self.unpacked) + self.assertEqual(recursive_lstat(self.instdir), + recursive_lstat(self.unpacked)) + -- cgit v1.2.1