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, *
13 * software distributed under the License is distributed on an *
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY *
15 * KIND, either express or implied. See the License for the *
16 * specific language governing permissions and limitations *
17 * under the License. *
18 ****************************************************************/
19
20 package org.apache.james.jdkim.canon;
21
22 import java.io.ByteArrayOutputStream;
23 import java.io.IOException;
24 import java.security.NoSuchAlgorithmException;
25
26 public class RelaxedBodyCanonicalizerTest extends AbstractOutputStreamTestCase {
27
28 private byte[] testData;
29 private byte[] expectedData;
30
31 protected void setUp() throws Exception {
32 testData = "this is a \r\n canonicalization \ttest\r\n\r\n\r\n"
33 .getBytes();
34 expectedData = "this is a\r\n canonicalization test\r\n".getBytes();
35 }
36
37 public void testSingleBytes() throws NoSuchAlgorithmException, IOException {
38 ByteArrayOutputStream bos = new ByteArrayOutputStream();
39 RelaxedBodyCanonicalizer os = new RelaxedBodyCanonicalizer(bos);
40 for (int i = 0; i < testData.length; i++) {
41 os.write(testData[i]);
42 }
43 os.close();
44 assertArrayEquals(expectedData, bos.toByteArray());
45 }
46
47 public void testChunks() throws NoSuchAlgorithmException, IOException {
48 ByteArrayOutputStream bos = new ByteArrayOutputStream();
49 RelaxedBodyCanonicalizer os = new RelaxedBodyCanonicalizer(bos);
50 chunker(testData, os);
51 assertArrayEquals(expectedData, bos.toByteArray());
52 }
53 }