1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
/*
* 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.utils;
import static org.junit.Assert.assertTrue;
import com.android.apkzlib.zip.ZFile;
import com.android.testutils.TestResources;
import com.google.common.base.Preconditions;
import com.google.common.io.ByteSource;
import com.google.common.io.Resources;
import java.io.ByteArrayInputStream;
import java.io.EOFException;
import java.io.File;
import java.io.IOException;
import java.io.RandomAccessFile;
import javax.annotation.Nonnull;
/**
* Utility functions for tests.
*/
public final class ApkZFileTestUtils {
/**
* Reads a portion of a file to memory.
*
* @param file the file to read data from
* @param start the offset in the file to start reading
* @param length the number of bytes to read
* @return the bytes read
* @throws Exception failed to read the file
*/
@Nonnull
public static byte[] readSegment(@Nonnull File file, long start, int length) throws Exception {
Preconditions.checkArgument(start >= 0, "start < 0");
Preconditions.checkArgument(length >= 0, "length < 0");
byte data[];
try (RandomAccessFile raf = new RandomAccessFile(file, "r")) {
raf.seek(start);
data = new byte[length];
int tot = 0;
while (tot < length) {
int r = raf.read(data, tot, length - tot);
if (r < 0) {
throw new EOFException();
}
tot += r;
}
}
return data;
}
/**
* Obtains the test resource with the given path.
*
* @param path the path
* @return the test resource
*/
@Nonnull
public static File getResource(@Nonnull String path) {
File resource = TestResources.getFile(ApkZFileTestUtils.class, path);
assertTrue(resource.exists());
return resource;
}
/**
* Obtains the test resource with the given path.
*
* @param path the path
* @return the test resource
*/
@Nonnull
public static ByteSource getResourceBytes(@Nonnull String path) {
return Resources.asByteSource(Resources.getResource(ApkZFileTestUtils.class, path));
}
/**
* Sleeps the current thread for enough time to ensure that the local file system had enough
* time to notice a "tick". This method is usually called in tests when it is necessary to
* ensure filesystem writes are detected through timestamp modification.
*
* @param currentTimestamp last timestamp read from disk
* @throws InterruptedException waiting interrupted
* @throws IOException issues creating a temporary file
*/
public static void waitForFileSystemTick(long currentTimestamp)
throws InterruptedException, IOException {
while (getFreshTimestamp() <= currentTimestamp) {
Thread.sleep(100);
}
}
/*
* Adds a basic compiled AndroidManifest to the given ZFile containing minSdkVersion equal 15
* and targetSdkVersion equal 25.
*/
public static void addAndroidManifest(ZFile zf) throws IOException {
zf.add("AndroidManifest.xml", new ByteArrayInputStream(getAndroidManifest()));
}
/*
* Provides a basic compiled AndroidManifest containing minSdkVersion equal 15 and
* targetSdkVersion equal 25.
*/
public static byte[] getAndroidManifest() throws IOException {
return ApkZFileTestUtils.getResourceBytes("/testData/packaging/AndroidManifest.xml").read();
}
/**
* Obtains the timestamp of a newly-created file.
*
* @return the timestamp
* @throws IOException the I/O Exception
*/
private static long getFreshTimestamp() throws IOException {
File notUsed = File.createTempFile(ApkZFileTestUtils.class.getName(), "waitForFSTick");
long freshTimestamp = notUsed.lastModified();
assertTrue(notUsed.delete());
return freshTimestamp;
}
}
|
