X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fdbtest_contents.py;h=90fe49666484a6e35ee19ac688fe0a440f7dfc4b;hb=c757aaa7d63b4508ad353a2339507ed837a8e0a3;hp=622e4aaf80d857fa16247c2e779253a794ec01a7;hpb=e8df34db4cc473d53136905027307484400fdf0d;p=dak.git

diff --git a/tests/dbtest_contents.py b/tests/dbtest_contents.py
index 622e4aaf..90fe4966 100755
--- a/tests/dbtest_contents.py
+++ b/tests/dbtest_contents.py
@@ -1,12 +1,13 @@
 #!/usr/bin/env python
 
-from db_test import DBDakTestCase
+from db_test import DBDakTestCase, fixture
 
 from daklib.dbconn import *
-from daklib.contents import ContentsWriter, ContentsScanner
+from daklib.contents import ContentsWriter, ContentsScanner, UnpackedSource
 
 from os.path import normpath
 from sqlalchemy.exc import FlushError, IntegrityError
+from subprocess import CalledProcessError
 import unittest
 
 class ContentsTestCase(DBDakTestCase):
@@ -166,12 +167,27 @@ class ContentsTestCase(DBDakTestCase):
         self.assertEqual(['usr/bin/hello', 'usr/share/doc/hello/copyright'],
             filelist)
         self.session.commit()
-        ContentsScanner(self.binary['hello_2.2-1_i386']).scan()
-        bin_contents_list = self.binary['hello_2.2-1_i386'].contents.all()
+        ContentsScanner(self.binary['hello_2.2-1_i386'].binary_id).scan()
+        bin_contents_list = self.binary['hello_2.2-1_i386'].contents.order_by('file').all()
         self.assertEqual(2, len(bin_contents_list))
         self.assertEqual('usr/bin/hello', bin_contents_list[0].file)
         self.assertEqual('usr/share/doc/hello/copyright', bin_contents_list[1].file)
 
+    def test_unpack(self):
+        '''
+        Tests the UnpackedSource class.
+        '''
+        self.setup_poolfiles()
+        dscfilename = fixture('ftp/pool/' + self.file['hello_2.2-1.dsc'].filename)
+        unpacked = UnpackedSource(dscfilename)
+        self.assertTrue(len(unpacked.get_root_directory()) > 0)
+        self.assertEqual('hello (2.2-1) unstable; urgency=low\n',
+            unpacked.get_changelog_file().readline())
+        all_filenames = set(unpacked.get_all_filenames())
+        self.assertEqual(8, len(all_filenames))
+        self.assertTrue('debian/rules' in all_filenames)
+        self.assertRaises(CalledProcessError, lambda: UnpackedSource('invalidname'))
+
     def classes_to_clean(self):
         return [Override, Suite, BinContents, DBBinary, DBSource, Architecture, Section, \
             OverrideType, Maintainer, Component, Priority, PoolFile]