def test_crc32(self): filename = make_random_temp_file() self.tempfiles.append(filename) consumer = Crc32Consumer() producer = FileProducer(filename) yield producer.beginProducing(consumer) self.assertEquals(consumer.crc32(), producer.crc32())
def create_zipstream_from_tempfiles(self, num_files, min_size=0, max_size=10 * (1024 ** 2)): consumer = TestConsumer() zipstream = ZipStream(consumer) deferreds = [] producers = {} for i in range(num_files): producersize = random.randint(min_size, max_size) filename = make_random_temp_file(producersize) producer = FileProducer(filename) deferreds.append(zipstream.addProducer(producer)) producers.update({filename: producer}) self.tempfiles.append(filename) deferredlist = DeferredList(deferreds) deferredlist.addCallback(lambda _: zipstream.centralDirectory()) deferredlist.addCallback(lambda _: consumer.seek(0)) yield deferredlist returnValue((zipstream, consumer, producers))