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 *
10
* http://www.apache.org/licenses/LICENSE-2.0 *
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
****************************************************************/
20
package org.apache.james.mime4j.storage;
22
import java.io.ByteArrayInputStream;
23
import java.io.ByteArrayOutputStream;
24
import java.io.IOException;
26
import org.apache.james.mime4j.codec.CodecUtil;
28
import junit.framework.TestCase;
30
public class StorageProviderTest extends TestCase {
32
public void testMemoryStorageProvider() throws Exception {
33
StorageProvider provider = new MemoryStorageProvider();
35
testReadWrite(provider, 0);
36
testReadWrite(provider, 1);
37
testReadWrite(provider, 1024);
38
testReadWrite(provider, 20000);
43
public void testTempFileStorageProvider() throws Exception {
44
StorageProvider provider = new TempFileStorageProvider();
46
testReadWrite(provider, 0);
47
testReadWrite(provider, 1);
48
testReadWrite(provider, 1024);
49
testReadWrite(provider, 20000);
54
public void testThresholdStorageProvider() throws Exception {
55
final int threshold = 5000;
56
StorageProvider backend = new TempFileStorageProvider();
57
StorageProvider provider = new ThresholdStorageProvider(backend,
60
testReadWrite(provider, 0);
61
testReadWrite(provider, 1);
62
testReadWrite(provider, threshold - 1);
63
testReadWrite(provider, threshold);
64
testReadWrite(provider, threshold + 1);
65
testReadWrite(provider, 2 * threshold);
66
testReadWrite(provider, 10 * threshold);
71
public void testCipherStorageProvider() throws Exception {
72
StorageProvider backend = new TempFileStorageProvider();
73
StorageProvider provider = new CipherStorageProvider(backend);
75
testReadWrite(provider, 0);
76
testReadWrite(provider, 1);
77
testReadWrite(provider, 1024);
78
testReadWrite(provider, 20000);
83
private void testReadWrite(StorageProvider provider, int size)
85
testStore(provider, size);
86
testCreateStorageOutputStream(provider, size);
89
private void testStore(StorageProvider provider, int size)
91
byte[] data = createData(size);
92
assertEquals(size, data.length);
94
Storage storage = provider.store(new ByteArrayInputStream(data));
96
ByteArrayOutputStream baos = new ByteArrayOutputStream();
97
CodecUtil.copy(storage.getInputStream(), baos);
98
verifyData(data, baos.toByteArray());
101
private void testCreateStorageOutputStream(StorageProvider provider,
102
int size) throws IOException {
103
byte[] data = createData(size);
104
assertEquals(size, data.length);
106
StorageOutputStream out = provider.createStorageOutputStream();
107
CodecUtil.copy(new ByteArrayInputStream(data), out);
108
Storage storage = out.toStorage();
110
ByteArrayOutputStream baos = new ByteArrayOutputStream();
111
CodecUtil.copy(storage.getInputStream(), baos);
112
verifyData(data, baos.toByteArray());
115
private void verifyData(byte[] expected, byte[] actual) {
116
assertEquals(expected.length, actual.length);
117
for (int i = 0; i < expected.length; i++) {
118
assertEquals(expected[i], actual[i]);
122
private byte[] createData(int size) {
123
byte[] data = new byte[size];
124
for (int i = 0; i < size; i++) {
130
private void testDelete(StorageProvider provider) throws IOException {
131
Storage storage = provider.store(new ByteArrayInputStream(
136
// getInputStream has to throw an IllegalStateException
138
storage.getInputStream();
140
} catch (IllegalStateException expected) {
143
// invoking delete a second time should not have any effect