94 lines
3.1 KiB
Java
94 lines
3.1 KiB
Java
/*
|
|
* Copyright (C) 2011 The Guava Authors
|
|
*
|
|
* 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.google.common.hash;
|
|
|
|
import static org.mockito.Mockito.mock;
|
|
import static org.mockito.Mockito.verify;
|
|
import static org.mockito.Mockito.verifyNoMoreInteractions;
|
|
import static org.mockito.Mockito.when;
|
|
|
|
import com.google.common.testing.NullPointerTester;
|
|
import java.io.ByteArrayOutputStream;
|
|
import junit.framework.TestCase;
|
|
|
|
/**
|
|
* Tests for {@link HashingOutputStream}.
|
|
*
|
|
* @author Nick Piepmeier
|
|
*/
|
|
public class HashingOutputStreamTest extends TestCase {
|
|
private Hasher hasher;
|
|
private HashFunction hashFunction;
|
|
private final ByteArrayOutputStream buffer = new ByteArrayOutputStream();
|
|
|
|
@SuppressWarnings("DoNotMock")
|
|
@Override
|
|
protected void setUp() throws Exception {
|
|
super.setUp();
|
|
hasher = mock(Hasher.class);
|
|
hashFunction = mock(HashFunction.class);
|
|
|
|
when(hashFunction.newHasher()).thenReturn(hasher);
|
|
}
|
|
|
|
public void testWrite_putSingleByte() throws Exception {
|
|
int b = 'q';
|
|
HashingOutputStream out = new HashingOutputStream(hashFunction, buffer);
|
|
out.write(b);
|
|
|
|
verify(hashFunction).newHasher();
|
|
verify(hasher).putByte((byte) b);
|
|
verifyNoMoreInteractions(hashFunction, hasher);
|
|
}
|
|
|
|
public void testWrite_putByteArray() throws Exception {
|
|
byte[] buf = new byte[] {'y', 'a', 'm', 's'};
|
|
HashingOutputStream out = new HashingOutputStream(hashFunction, buffer);
|
|
out.write(buf);
|
|
|
|
verify(hashFunction).newHasher();
|
|
verify(hasher).putBytes(buf, 0, buf.length);
|
|
verifyNoMoreInteractions(hashFunction, hasher);
|
|
}
|
|
|
|
public void testWrite_putByteArrayAtPos() throws Exception {
|
|
byte[] buf = new byte[] {'y', 'a', 'm', 's'};
|
|
HashingOutputStream out = new HashingOutputStream(hashFunction, buffer);
|
|
out.write(buf, 0, 3);
|
|
|
|
verify(hashFunction).newHasher();
|
|
verify(hasher).putBytes(buf, 0, 3);
|
|
verifyNoMoreInteractions(hashFunction, hasher);
|
|
}
|
|
|
|
public void testHash_hashesCorrectly() throws Exception {
|
|
byte[] buf = new byte[] {'y', 'a', 'm', 's'};
|
|
HashCode expectedHash = Hashing.md5().hashBytes(buf);
|
|
HashingOutputStream out = new HashingOutputStream(Hashing.md5(), buffer);
|
|
|
|
out.write(buf);
|
|
|
|
assertEquals(expectedHash, out.hash());
|
|
}
|
|
|
|
public void testChecksForNull() throws Exception {
|
|
NullPointerTester tester = new NullPointerTester();
|
|
tester.testAllPublicInstanceMethods(
|
|
new HashingOutputStream(Hashing.md5(), new ByteArrayOutputStream()));
|
|
tester.testAllPublicStaticMethods(HashingOutputStream.class);
|
|
tester.testAllPublicConstructors(HashingOutputStream.class);
|
|
}
|
|
}
|