Added MessageStore strategy and SimpleMessageStore implementation for in-memory storage with a BoundedHashMap.

This commit is contained in:
Mark Fisher
2008-03-27 19:11:54 +00:00
parent 71c2b80f7a
commit ee5e61da44
5 changed files with 233 additions and 2 deletions

View File

@@ -0,0 +1,82 @@
/*
* Copyright 2002-2008 the original author or 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 org.springframework.integration.message;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import org.junit.Test;
/**
* @author Mark Fisher
*/
public class SimpleMessageStoreTests {
@Test
public void testPut() {
SimpleMessageStore store = new SimpleMessageStore();
Message<?> message1 = new StringMessage("message-1");
Message<?> previous = store.put(1, message1);
assertNull(previous);
assertNotNull(store.get(1));
assertEquals(message1, store.get(1));
}
@Test
public void testReplace() {
SimpleMessageStore store = new SimpleMessageStore();
Message<?> messageA = new StringMessage("message-a");
Message<?> messageB = new StringMessage("message-b");
store.put(1, messageA);
Message<?> previous = store.put(1, messageB);
assertEquals(messageA, previous);
assertEquals(messageB, store.get(1));
}
@Test
public void testRemove() {
SimpleMessageStore store = new SimpleMessageStore();
Message<?> message = new StringMessage("message");
assertNull(store.remove(1));
store.put(1, message);
assertEquals(message, store.remove(1));
}
@Test
public void testCapacityEnforced() {
SimpleMessageStore store = new SimpleMessageStore(3);
Message<?> message1 = new StringMessage("message-1");
Message<?> message2 = new StringMessage("message-2");
Message<?> message3 = new StringMessage("message-3");
Message<?> message4 = new StringMessage("message-4");
store.put(1, message1);
store.put(2, message2);
store.put(3, message3);
assertEquals(3, store.size());
assertEquals(message1, store.get(1));
assertEquals(message2, store.get(2));
assertEquals(message3, store.get(3));
store.put(4, message4);
assertEquals(3, store.size());
assertNull(store.get(1));
assertEquals(message2, store.get(2));
assertEquals(message3, store.get(3));
assertEquals(message4, store.get(4));
}
}

View File

@@ -0,0 +1,48 @@
/*
* Copyright 2002-2008 the original author or 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 org.springframework.integration.util;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import org.junit.Test;
/**
* @author Mark Fisher
*/
public class BoundedHashMapTests {
@Test
public void testCapacityEnforced() {
BoundedHashMap<String, Integer> map = new BoundedHashMap<String, Integer>(3);
map.put("A", 1);
map.put("B", 2);
map.put("C", 3);
assertEquals(3, map.size());
assertTrue(map.containsKey("A"));
assertTrue(map.containsKey("B"));
assertTrue(map.containsKey("C"));
map.put("D", 4);
assertEquals(3, map.size());
assertFalse(map.containsKey("A"));
assertTrue(map.containsKey("B"));
assertTrue(map.containsKey("C"));
assertTrue(map.containsKey("D"));
}
}