View Javadoc
1   /*
2    * This file is part of dependency-check-core.
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    *     http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   *
16   * Copyright (c) 2012 Jeremy Long. All Rights Reserved.
17   */
18  package org.owasp.dependencycheck.utils;
19  
20  import java.io.File;
21  import java.io.IOException;
22  import java.security.NoSuchAlgorithmException;
23  
24  import org.junit.Assert;
25  import org.junit.Rule;
26  import org.junit.Test;
27  import org.junit.rules.ExpectedException;
28  import org.owasp.dependencycheck.utils.Checksum;
29  import org.owasp.dependencycheck.utils.Checksum;
30  
31  /**
32   *
33   * @author Jeremy Long <jeremy.long@owasp.org>
34   */
35  public class ChecksumTest {
36  
37      @Rule
38      public ExpectedException expectedException = ExpectedException.none();
39  
40      /**
41       * Test of getChecksum method, of class Checksum.
42       *
43       * @throws Exception thrown when an exception occurs.
44       */
45      @Test
46      public void testGetChecksum() throws Exception {
47          String algorithm = "MD5";
48          File file = new File(this.getClass().getClassLoader().getResource("checkSumTest.file").getPath());
49          byte[] expResult = {-16, -111, 92, 95, 70, -72, -49, -94, -125, -27, -83, 103, -96, -101, 55, -109};
50          byte[] result = Checksum.getChecksum(algorithm, file);
51          boolean arraysAreEqual = true;
52          if (expResult.length == result.length) {
53              for (int i = 0; arraysAreEqual && i < result.length; i++) {
54                  arraysAreEqual = result[i] == expResult[i];
55              }
56          } else {
57              Assert.fail("Checksum results do not match expected results.");
58          }
59          Assert.assertTrue(arraysAreEqual);
60      }
61  
62      /**
63       * Test of getChecksum method, of class Checksum. This checks that an exception is thrown when an invalid path is
64       * specified.
65       *
66       * @throws Exception is thrown when an exception occurs.
67       */
68      @Test
69      public void testGetChecksum_FileNotFound() throws Exception {
70          String algorithm = "MD5";
71          File file = new File("not a valid file");
72  
73          expectedException.expect(IOException.class);
74          Checksum.getChecksum(algorithm, file);
75      }
76  
77      /**
78       * Test of getChecksum method, of class Checksum. This checks that an exception is thrown when an invalid algorithm
79       * is specified.
80       *
81       * @throws Exception is thrown when an exception occurs.
82       */
83      @Test
84      public void testGetChecksum_NoSuchAlgorithm() throws Exception {
85          String algorithm = "some unknown algorithm";
86          File file = new File(this.getClass().getClassLoader().getResource("checkSumTest.file").getPath());
87  
88          expectedException.expect(NoSuchAlgorithmException.class);
89          Checksum.getChecksum(algorithm, file);
90      }
91  
92      /**
93       * Test of getMD5Checksum method, of class Checksum.
94       *
95       * @throws Exception is thrown when an exception occurs.
96       */
97      @Test
98      public void testGetMD5Checksum() throws Exception {
99          File file = new File(this.getClass().getClassLoader().getResource("checkSumTest.file").getPath());
100         String expResult = "F0915C5F46B8CFA283E5AD67A09B3793";
101         String result = Checksum.getMD5Checksum(file);
102         Assert.assertEquals(expResult, result);
103     }
104 
105     /**
106      * Test of getSHA1Checksum method, of class Checksum.
107      *
108      * @throws Exception is thrown when an exception occurs.
109      */
110     @Test
111     public void testGetSHA1Checksum() throws Exception {
112         File file = new File(this.getClass().getClassLoader().getResource("checkSumTest.file").getPath());
113         String expResult = "B8A9FF28B21BCB1D0B50E24A5243D8B51766851A";
114         String result = Checksum.getSHA1Checksum(file);
115         Assert.assertEquals(expResult, result);
116     }
117 
118     /**
119      * Test of getHex method, of class Checksum.
120      */
121     @Test
122     public void testGetHex() {
123         byte[] raw = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16};
124         String expResult = "000102030405060708090A0B0C0D0E0F10";
125         String result = Checksum.getHex(raw);
126         Assert.assertEquals(expResult, result);
127     }
128 }