def test_preload(self): s = MockDest() with JarWriter(fileobj=s) as jar: jar.add('foo', 'foo') jar.add('bar', 'abcdefghijklmnopqrstuvwxyz') jar.add('baz/qux', 'aaaaaaaaaaaaanopqrstuvwxyz') jar = JarReader(fileobj=s) self.assertEqual(jar.last_preloaded, None) with JarWriter(fileobj=s) as jar: jar.add('foo', 'foo') jar.add('bar', 'abcdefghijklmnopqrstuvwxyz') jar.add('baz/qux', 'aaaaaaaaaaaaanopqrstuvwxyz') jar.preload(['baz/qux', 'bar']) jar = JarReader(fileobj=s) self.assertEqual(jar.last_preloaded, 'bar') files = [j for j in jar] self.assertEqual(files[0].filename, 'baz/qux') self.assertEqual(files[1].filename, 'bar') self.assertEqual(files[2].filename, 'foo')
def test_preload(self): s = MockDest() with JarWriter(fileobj=s) as jar: jar.add("foo", b"foo") jar.add("bar", b"abcdefghijklmnopqrstuvwxyz") jar.add("baz/qux", b"aaaaaaaaaaaaanopqrstuvwxyz") jar = JarReader(fileobj=s) self.assertEqual(jar.last_preloaded, None) with JarWriter(fileobj=s) as jar: jar.add("foo", b"foo") jar.add("bar", b"abcdefghijklmnopqrstuvwxyz") jar.add("baz/qux", b"aaaaaaaaaaaaanopqrstuvwxyz") jar.preload(["baz/qux", "bar"]) jar = JarReader(fileobj=s) self.assertEqual(jar.last_preloaded, "bar") files = [j for j in jar] self.assertEqual(files[0].filename, "baz/qux") self.assertEqual(files[1].filename, "bar") self.assertEqual(files[2].filename, "foo")