1 | /** |
---|
2 | * Licensed to the Apache Software Foundation (ASF) under one |
---|
3 | * or more contributor license agreements. See the NOTICE file |
---|
4 | * distributed with this work for additional information |
---|
5 | * regarding copyright ownership. The ASF licenses this file |
---|
6 | * to you under the Apache License, Version 2.0 (the |
---|
7 | * "License"); you may not use this file except in compliance |
---|
8 | * with the License. You may obtain a copy of the License at |
---|
9 | * |
---|
10 | * http://www.apache.org/licenses/LICENSE-2.0 |
---|
11 | * |
---|
12 | * Unless required by applicable law or agreed to in writing, software |
---|
13 | * distributed under the License is distributed on an "AS IS" BASIS, |
---|
14 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
---|
15 | * See the License for the specific language governing permissions and |
---|
16 | * limitations under the License. |
---|
17 | */ |
---|
18 | |
---|
19 | package org.apache.hadoop.io; |
---|
20 | |
---|
21 | import org.apache.hadoop.io.TestWritable; |
---|
22 | import junit.framework.TestCase; |
---|
23 | import java.security.MessageDigest; |
---|
24 | import java.util.Random; |
---|
25 | |
---|
26 | /** Unit tests for MD5Hash. */ |
---|
27 | public class TestMD5Hash extends TestCase { |
---|
28 | public TestMD5Hash(String name) { super(name); } |
---|
29 | |
---|
30 | private static final Random RANDOM = new Random(); |
---|
31 | |
---|
32 | public static MD5Hash getTestHash() throws Exception { |
---|
33 | MessageDigest digest = MessageDigest.getInstance("MD5"); |
---|
34 | byte[] buffer = new byte[1024]; |
---|
35 | RANDOM.nextBytes(buffer); |
---|
36 | digest.update(buffer); |
---|
37 | return new MD5Hash(digest.digest()); |
---|
38 | } |
---|
39 | |
---|
40 | protected static byte[] D00 = new byte[] {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}; |
---|
41 | protected static byte[] DFF = new byte[] {-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1}; |
---|
42 | |
---|
43 | public void testMD5Hash() throws Exception { |
---|
44 | MD5Hash md5Hash = getTestHash(); |
---|
45 | |
---|
46 | final MD5Hash md5Hash00 |
---|
47 | = new MD5Hash(D00); |
---|
48 | |
---|
49 | final MD5Hash md5HashFF |
---|
50 | = new MD5Hash(DFF); |
---|
51 | |
---|
52 | MD5Hash orderedHash = new MD5Hash(new byte[]{1,2,3,4,5,6,7,8,9,10,11,12, |
---|
53 | 13,14,15,16}); |
---|
54 | MD5Hash backwardHash = new MD5Hash(new byte[]{-1,-2,-3,-4,-5,-6,-7,-8, |
---|
55 | -9,-10,-11,-12, -13, -14, |
---|
56 | -15,-16}); |
---|
57 | MD5Hash closeHash1 = new MD5Hash(new byte[]{-1,0,0,0,0,0,0,0, |
---|
58 | 0,0,0,0,0,0,0,0}); |
---|
59 | MD5Hash closeHash2 = new MD5Hash(new byte[]{-1,1,0,0,0,0,0,0, |
---|
60 | 0,0,0,0,0,0,0,0}); |
---|
61 | |
---|
62 | // test i/o |
---|
63 | TestWritable.testWritable(md5Hash); |
---|
64 | TestWritable.testWritable(md5Hash00); |
---|
65 | TestWritable.testWritable(md5HashFF); |
---|
66 | |
---|
67 | // test equals() |
---|
68 | assertEquals(md5Hash, md5Hash); |
---|
69 | assertEquals(md5Hash00, md5Hash00); |
---|
70 | assertEquals(md5HashFF, md5HashFF); |
---|
71 | |
---|
72 | // test compareTo() |
---|
73 | assertTrue(md5Hash.compareTo(md5Hash) == 0); |
---|
74 | assertTrue(md5Hash00.compareTo(md5Hash) < 0); |
---|
75 | assertTrue(md5HashFF.compareTo(md5Hash) > 0); |
---|
76 | |
---|
77 | // test toString and string ctor |
---|
78 | assertEquals(md5Hash, new MD5Hash(md5Hash.toString())); |
---|
79 | assertEquals(md5Hash00, new MD5Hash(md5Hash00.toString())); |
---|
80 | assertEquals(md5HashFF, new MD5Hash(md5HashFF.toString())); |
---|
81 | |
---|
82 | assertEquals(0x01020304, orderedHash.quarterDigest()); |
---|
83 | assertEquals(0xfffefdfc, backwardHash.quarterDigest()); |
---|
84 | |
---|
85 | assertEquals(0x0102030405060708L, orderedHash.halfDigest()); |
---|
86 | assertEquals(0xfffefdfcfbfaf9f8L, backwardHash.halfDigest()); |
---|
87 | assertTrue("hash collision", |
---|
88 | closeHash1.hashCode() != closeHash2.hashCode()); |
---|
89 | |
---|
90 | Thread t1 = new Thread() { |
---|
91 | public void run() { |
---|
92 | for (int i = 0; i < 100; i++) { |
---|
93 | MD5Hash hash = new MD5Hash(DFF); |
---|
94 | assertEquals(hash, md5HashFF); |
---|
95 | } |
---|
96 | } |
---|
97 | }; |
---|
98 | |
---|
99 | Thread t2 = new Thread() { |
---|
100 | public void run() { |
---|
101 | for (int i = 0; i < 100; i++) { |
---|
102 | MD5Hash hash = new MD5Hash(D00); |
---|
103 | assertEquals(hash, md5Hash00); |
---|
104 | } |
---|
105 | } |
---|
106 | }; |
---|
107 | |
---|
108 | t1.start(); |
---|
109 | t2.start(); |
---|
110 | t1.join(); |
---|
111 | t2.join(); |
---|
112 | |
---|
113 | } |
---|
114 | |
---|
115 | } |
---|