Пример #1
0
 def _createCpio(self, rpmName = None):
     if rpmName is None:
         rpmName = 'perl-Archive-Tar-1.46-68.fc11.x86_64.rpm'
     cpioPath = os.path.join(self.workDir, 'archive.cpio')
     rpmFile = file(os.path.join(self.archiveDir, rpmName))
     rpmhelper.extractRpmPayload(rpmFile, file(cpioPath, "w"))
     return cpioPath
Пример #2
0
    def testExtractRpmPayload(self):
        archives = [
            ('popt-1.5-4x.i386.rpm', 78252),
            ('rpm-with-bzip-5.0.29-1.i386.rpm', 248),
        ]
        # We cannot unpack xz-compressed contents with /usr/bin/lzma
        if os.path.exists('/usr/bin/xz'):
            archives.extend([
                ('popt-1.13-6.fc12.i686.rpm', 85544),
            ])
        for rpmName, expectedSize in archives:
            fileobj = file(os.path.join(self.archiveDir, rpmName))
            sio = StringIO.StringIO()
            rpmhelper.extractRpmPayload(fileobj, sio)
            sio.seek(0, 2)
            self.assertEqual(sio.tell(), expectedSize)

        # Test exception codepaths
        header = {rpmhelper.PAYLOADFORMAT: "super-payload format"}
        self.mock(rpmhelper, "readHeader", lambda x: header)
        e = self.assertRaises(rpmhelper.UnknownPayloadFormat,
                              rpmhelper.extractRpmPayload, file("/dev/null"),
                              None)
        self.assertEqual(e.args[0], "super-payload format")

        # No payload format, it should assume CPIO. Test compression
        header = {rpmhelper.PAYLOADCOMPRESSOR: "lame"}
        e = self.assertRaises(rpmhelper.UnknownCompressionType,
                              rpmhelper.extractRpmPayload, file("/dev/null"),
                              None)
        self.assertEqual(e.args[0], "lame")

        # No payload compressor defined we should detect gzip or
        # assume uncompressed
        header = {rpmhelper.PAYLOADFORMAT: "cpio"}
        fileIn = StringIO.StringIO()
        fileIn.write('\x1f\x8bThis should be some gziped data')
        fileIn.seek(0)

        def gzipFunc(*args, **kw):
            return 'Gzip File'

        self.mock(util, "GzipFile", gzipFunc)
        self.assertEqual(rpmhelper.UncompressedRpmPayload(fileIn), 'Gzip File')
        fileIn = StringIO.StringIO()
        fileIn.write('This is just some uncompressed data')
        fileIn.seek(0)
        self.assertEqual(rpmhelper.UncompressedRpmPayload(fileIn), fileIn)
Пример #3
0
    def testExtractRpmPayload(self):
        archives = [("popt-1.5-4x.i386.rpm", 78252), ("rpm-with-bzip-5.0.29-1.i386.rpm", 248)]
        # We cannot unpack xz-compressed contents with /usr/bin/lzma
        if os.path.exists("/usr/bin/xz"):
            archives.extend([("popt-1.13-6.fc12.i686.rpm", 85544)])
        for rpmName, expectedSize in archives:
            fileobj = file(os.path.join(self.archiveDir, rpmName))
            sio = StringIO.StringIO()
            rpmhelper.extractRpmPayload(fileobj, sio)
            sio.seek(0, 2)
            self.assertEqual(sio.tell(), expectedSize)

        # Test exception codepaths
        header = {rpmhelper.PAYLOADFORMAT: "super-payload format"}
        self.mock(rpmhelper, "readHeader", lambda x: header)
        e = self.assertRaises(rpmhelper.UnknownPayloadFormat, rpmhelper.extractRpmPayload, file("/dev/null"), None)
        self.assertEqual(e.args[0], "super-payload format")

        # No payload format, it should assume CPIO. Test compression
        header = {rpmhelper.PAYLOADCOMPRESSOR: "lame"}
        e = self.assertRaises(rpmhelper.UnknownCompressionType, rpmhelper.extractRpmPayload, file("/dev/null"), None)
        self.assertEqual(e.args[0], "lame")

        # No payload compressor defined we should detect gzip or
        # assume uncompressed
        header = {rpmhelper.PAYLOADFORMAT: "cpio"}
        fileIn = StringIO.StringIO()
        fileIn.write("\x1f\x8bThis should be some gziped data")
        fileIn.seek(0)

        def gzipFunc(*args, **kw):
            return "Gzip File"

        self.mock(util, "GzipFile", gzipFunc)
        self.assertEqual(rpmhelper.UncompressedRpmPayload(fileIn), "Gzip File")
        fileIn = StringIO.StringIO()
        fileIn.write("This is just some uncompressed data")
        fileIn.seek(0)
        self.assertEqual(rpmhelper.UncompressedRpmPayload(fileIn), fileIn)