1 files changed, 180 insertions, 0 deletions
diff --git a/src/test/java/com/android/apkzlib/sign/ManifestGenerationTest.java b/src/test/java/com/android/apkzlib/sign/ManifestGenerationTest.java
new file mode 100644
index 0000000..746617b
--- /dev/null
+++ b/src/test/java/com/android/apkzlib/sign/ManifestGenerationTest.java
@@ -0,0 +1,180 @@
+/*
+ * Copyright (C) 2016 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.android.apkzlib.sign;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
+import com.android.apkzlib.zip.StoredEntry;
+import com.android.apkzlib.zip.ZFile;
+import com.android.apkzlib.utils.ApkZFileTestUtils;
+import com.google.common.base.Charsets;
+import com.google.common.io.Closer;
+import java.io.ByteArrayInputStream;
+import java.io.File;
+import java.io.InputStream;
+import java.util.Set;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.TemporaryFolder;
+import org.mockito.internal.util.collections.Sets;
+
+public class ManifestGenerationTest {
+
+    private static final String WIKI_PATH = "/testData/packaging/text-files/wikipedia.html";
+
+    @Rule
+    public TemporaryFolder mTemporaryFolder = new TemporaryFolder();
+
+    @Test
+    public void elementaryManifestGeneration() throws Exception {
+        File zip = new File(mTemporaryFolder.getRoot(), "f.zip");
+
+        try (ZFile zf = new ZFile(zip)) {
+            zf.add("abc", new ByteArrayInputStream(new byte[]{1}));
+            zf.add("x/", new ByteArrayInputStream(new byte[0]));
+            zf.add("x/abc", new ByteArrayInputStream(new byte[]{2}));
+
+            ManifestGenerationExtension extension =
+                    new ManifestGenerationExtension("Me, of course", "Myself");
+            extension.register(zf);
+
+            zf.update();
+
+            StoredEntry se = zf.get("META-INF/MANIFEST.MF");
+            assertNotNull(se);
+
+            String text = new String(se.read(), Charsets.US_ASCII);
+            text = text.trim();
+            String lines[] = text.split(System.getProperty("line.separator"));
+            assertEquals(3, lines.length);
+
+            assertEquals("Manifest-Version: 1.0", lines[0].trim());
+
+            Set<String> linesSet = Sets.newSet();
+            for (String l : lines) {
+                linesSet.add(l.trim());
+            }
+
+            assertTrue(linesSet.contains("Built-By: Me, of course"));
+            assertTrue(linesSet.contains("Created-By: Myself"));
+        }
+    }
+
+    @Test
+    public void manifestGenerationOnHalfWrittenFile() throws Exception {
+        File zip = new File(mTemporaryFolder.getRoot(), "f.zip");
+        try (Closer closer = Closer.create()) {
+            ZFile zf = closer.register(new ZFile(zip));
+
+            try (InputStream wiki = getClass().getResourceAsStream(WIKI_PATH)) {
+                zf.add("wiki", wiki);
+            }
+
+            ManifestGenerationExtension extension =
+                    new ManifestGenerationExtension("Me, of course", "Myself");
+            extension.register(zf);
+
+            zf.close();
+
+            StoredEntry se = zf.get("META-INF/MANIFEST.MF");
+            assertNotNull(se);
+
+            String text = new String(se.read(), Charsets.US_ASCII);
+            text = text.trim();
+            String lines[] = text.split(System.getProperty("line.separator"));
+            assertEquals(3, lines.length);
+
+            assertEquals("Manifest-Version: 1.0", lines[0].trim());
+
+            Set<String> linesSet = Sets.newSet();
+            for (String l : lines) {
+                linesSet.add(l.trim());
+            }
+
+            assertTrue(linesSet.contains("Built-By: Me, of course"));
+            assertTrue(linesSet.contains("Created-By: Myself"));
+        }
+    }
+
+    @Test
+    public void manifestGenerationOnExistingFile() throws Exception {
+        File zip = new File(mTemporaryFolder.getRoot(), "f.zip");
+        try (Closer closer = Closer.create()) {
+            ZFile zf = closer.register(new ZFile(zip));
+
+            try (InputStream wiki = getClass().getResourceAsStream(WIKI_PATH)) {
+                zf.add("wiki", wiki);
+            }
+
+            zf.close();
+
+            ManifestGenerationExtension extension =
+                    new ManifestGenerationExtension("Me, of course", "Myself");
+            extension.register(zf);
+
+            zf.close();
+
+            StoredEntry se = zf.get("META-INF/MANIFEST.MF");
+            assertNotNull(se);
+
+            String text = new String(se.read(), Charsets.US_ASCII);
+            text = text.trim();
+            String lines[] = text.split(System.getProperty("line.separator"));
+            assertEquals(3, lines.length);
+
+            assertEquals("Manifest-Version: 1.0", lines[0].trim());
+
+            Set<String> linesSet = Sets.newSet();
+            for (String l : lines) {
+                linesSet.add(l.trim());
+            }
+
+            assertTrue(linesSet.contains("Built-By: Me, of course"));
+            assertTrue(linesSet.contains("Created-By: Myself"));
+        }
+    }
+
+    @Test
+    public void manifestGenerationOnIncrementalNoChanges() throws Exception {
+        File zip = new File(mTemporaryFolder.getRoot(), "f.zip");
+        try (Closer closer = Closer.create()) {
+            ZFile zf = closer.register(new ZFile(zip));
+
+            ManifestGenerationExtension extension =
+                    new ManifestGenerationExtension("Me, of course", "Myself");
+            extension.register(zf);
+
+            try (InputStream wiki = getClass().getResourceAsStream(WIKI_PATH)) {
+                zf.add("wiki", wiki);
+            }
+
+            zf.close();
+
+            long timeOfWriting = zip.lastModified();
+
+            ApkZFileTestUtils.waitForFileSystemTick(timeOfWriting);
+
+            zf = closer.register(new ZFile(zip));
+            zf.close();
+
+            long secondTimeOfWriting = zip.lastModified();
+            assertEquals(timeOfWriting, secondTimeOfWriting);
+        }
+    }
+}