Skip to content

Commit 1889d95

Browse files
committed
Added an assertion to simplify the verification of the signed files in the tests
1 parent 5937853 commit 1889d95

File tree

13 files changed

+211
-592
lines changed

13 files changed

+211
-592
lines changed

jsign-ant/pom.xml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,14 @@
2626
<version>1.8.4</version>
2727
<scope>provided</scope>
2828
</dependency>
29+
30+
<dependency>
31+
<groupId>net.jsign</groupId>
32+
<artifactId>jsign-core</artifactId>
33+
<version>${project.version}</version>
34+
<type>test-jar</type>
35+
<scope>test</scope>
36+
</dependency>
2937
</dependencies>
3038

3139
<build>

jsign-ant/src/test/java/net/jsign/JsignTaskTest.java

Lines changed: 11 additions & 72 deletions
Original file line numberDiff line numberDiff line change
@@ -19,21 +19,20 @@
1919
import java.io.File;
2020
import java.io.OutputStream;
2121
import java.io.PrintStream;
22-
import java.util.List;
2322

2423
import org.apache.commons.io.FileUtils;
2524
import org.apache.tools.ant.BuildException;
2625
import org.apache.tools.ant.DefaultLogger;
2726
import org.apache.tools.ant.Project;
2827
import org.apache.tools.ant.ProjectHelper;
29-
import org.bouncycastle.cms.CMSSignedData;
3028
import org.junit.Before;
3129
import org.junit.Test;
3230

3331
import net.jsign.msi.MSIFile;
3432
import net.jsign.pe.PEFile;
3533
import net.jsign.script.PowerShellScript;
3634

35+
import static net.jsign.DigestAlgorithm.*;
3736
import static org.junit.Assert.*;
3837

3938
public class JsignTaskTest {
@@ -177,13 +176,7 @@ public void testSigning() throws Exception {
177176
assertTrue("The file " + targetFile + " wasn't changed", SOURCE_FILE_CRC32 != FileUtils.checksumCRC32(targetFile));
178177

179178
try (PEFile peFile = new PEFile(targetFile)) {
180-
List<CMSSignedData> signatures = peFile.getSignatures();
181-
assertNotNull(signatures);
182-
assertEquals(1, signatures.size());
183-
184-
CMSSignedData signature = signatures.get(0);
185-
186-
assertNotNull(signature);
179+
SignatureAssert.assertSigned(peFile, SHA1);
187180
}
188181
}
189182

@@ -199,22 +192,10 @@ public void testSigningMultipleFiles() throws Exception {
199192
assertTrue("The file " + targetFile2 + " wasn't changed", SOURCE_FILE_CRC32 != FileUtils.checksumCRC32(targetFile2));
200193

201194
try (PEFile peFile = new PEFile(targetFile)) {
202-
List<CMSSignedData> signatures = peFile.getSignatures();
203-
assertNotNull(signatures);
204-
assertEquals(1, signatures.size());
205-
206-
CMSSignedData signature = signatures.get(0);
207-
208-
assertNotNull(signature);
195+
SignatureAssert.assertSigned(peFile, SHA1);
209196
}
210197
try (PEFile peFile = new PEFile(targetFile2)) {
211-
List<CMSSignedData> signatures = peFile.getSignatures();
212-
assertNotNull(signatures);
213-
assertEquals(1, signatures.size());
214-
215-
CMSSignedData signature = signatures.get(0);
216-
217-
assertNotNull(signature);
198+
SignatureAssert.assertSigned(peFile, SHA1);
218199
}
219200
}
220201

@@ -229,13 +210,7 @@ public void testSigningPowerShell() throws Exception {
229210

230211
PowerShellScript script = new PowerShellScript(targetFile);
231212

232-
List<CMSSignedData> signatures = script.getSignatures();
233-
assertNotNull(signatures);
234-
assertEquals(1, signatures.size());
235-
236-
CMSSignedData signature = signatures.get(0);
237-
238-
assertNotNull(signature);
213+
SignatureAssert.assertSigned(script, SHA1);
239214
}
240215

241216
@Test
@@ -248,13 +223,7 @@ public void testSigningMSI() throws Exception {
248223
project.executeTarget("signing-msi");
249224

250225
try (MSIFile file = new MSIFile(targetFile)) {
251-
List<CMSSignedData> signatures = file.getSignatures();
252-
assertNotNull(signatures);
253-
assertEquals(1, signatures.size());
254-
255-
CMSSignedData signature = signatures.get(0);
256-
257-
assertNotNull(signature);
226+
SignatureAssert.assertSigned(file, SHA1);
258227
}
259228
}
260229

@@ -265,13 +234,7 @@ public void testSigningPKCS12() throws Exception {
265234
assertTrue("The file " + targetFile + " wasn't changed", SOURCE_FILE_CRC32 != FileUtils.checksumCRC32(targetFile));
266235

267236
try (PEFile peFile = new PEFile(targetFile)) {
268-
List<CMSSignedData> signatures = peFile.getSignatures();
269-
assertNotNull(signatures);
270-
assertEquals(1, signatures.size());
271-
272-
CMSSignedData signature = signatures.get(0);
273-
274-
assertNotNull(signature);
237+
SignatureAssert.assertSigned(peFile, SHA256);
275238
}
276239
}
277240

@@ -282,13 +245,7 @@ public void testSigningPVKSPC() throws Exception {
282245
assertTrue("The file " + targetFile + " wasn't changed", SOURCE_FILE_CRC32 != FileUtils.checksumCRC32(targetFile));
283246

284247
try (PEFile peFile = new PEFile(targetFile)) {
285-
List<CMSSignedData> signatures = peFile.getSignatures();
286-
assertNotNull(signatures);
287-
assertEquals(1, signatures.size());
288-
289-
CMSSignedData signature = signatures.get(0);
290-
291-
assertNotNull(signature);
248+
SignatureAssert.assertSigned(peFile, SHA256);
292249
}
293250
}
294251

@@ -301,13 +258,7 @@ public void testTimestampingAuthenticode() throws Exception {
301258
assertTrue("The file " + targetFile2 + " wasn't changed", SOURCE_FILE_CRC32 != FileUtils.checksumCRC32(targetFile2));
302259

303260
try (PEFile peFile = new PEFile(targetFile2)) {
304-
List<CMSSignedData> signatures = peFile.getSignatures();
305-
assertNotNull(signatures);
306-
assertEquals(1, signatures.size());
307-
308-
CMSSignedData signature = signatures.get(0);
309-
310-
assertNotNull(signature);
261+
SignatureAssert.assertSigned(peFile, SHA256);
311262
}
312263
}
313264

@@ -320,13 +271,7 @@ public void testTimestampingRFC3161() throws Exception {
320271
assertTrue("The file " + targetFile2 + " wasn't changed", SOURCE_FILE_CRC32 != FileUtils.checksumCRC32(targetFile2));
321272

322273
try (PEFile peFile = new PEFile(targetFile2)) {
323-
List<CMSSignedData> signatures = peFile.getSignatures();
324-
assertNotNull(signatures);
325-
assertEquals(1, signatures.size());
326-
327-
CMSSignedData signature = signatures.get(0);
328-
329-
assertNotNull(signature);
274+
SignatureAssert.assertSigned(peFile, SHA256);
330275
}
331276
}
332277

@@ -339,13 +284,7 @@ public void testReplaceSignature() throws Exception {
339284
assertTrue("The file " + targetFile2 + " wasn't changed", SOURCE_FILE_CRC32 != FileUtils.checksumCRC32(targetFile2));
340285

341286
try (PEFile peFile = new PEFile(targetFile2)) {
342-
List<CMSSignedData> signatures = peFile.getSignatures();
343-
assertNotNull(signatures);
344-
assertEquals(1, signatures.size());
345-
346-
assertNotNull(signatures.get(0));
347-
348-
assertEquals("Digest algorithm", DigestAlgorithm.SHA512.oid, signatures.get(0).getDigestAlgorithmIDs().iterator().next().getAlgorithm());
287+
SignatureAssert.assertSigned(peFile, SHA512);
349288
}
350289
}
351290

jsign-cli/pom.xml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,14 @@
2626
<version>1.4</version>
2727
</dependency>
2828

29+
<dependency>
30+
<groupId>net.jsign</groupId>
31+
<artifactId>jsign-core</artifactId>
32+
<version>${project.version}</version>
33+
<type>test-jar</type>
34+
<scope>test</scope>
35+
</dependency>
36+
2937
<dependency>
3038
<groupId>org.littleshoot</groupId>
3139
<artifactId>littleproxy</artifactId>

0 commit comments

Comments
 (0)