From a974fff07d9c3c5ff08d3f120f629abe618e2ab6 Mon Sep 17 00:00:00 2001 From: Xiaoyan Lin Date: Mon, 2 May 2016 21:58:47 -0700 Subject: [PATCH] Add regression tests for FullMemcacheMessageRequest and FullMemcacheMessageResponse Motivation: It's better to have some tests for FullMemcacheMessageRequest and FullMemcacheMessageResponse to avoid regression like #5197. Modifications: Add tests for FullMemcacheMessageRequest and FullMemcacheMessageResponse. Result: There are some basic tests for FullMemcacheMessageRequest and FullMemcacheMessageResponse. --- .../FullMemcacheMessageRequestTest.java | 62 ++++++++++++++++++ .../FullMemcacheMessageResponseTest.java | 64 +++++++++++++++++++ 2 files changed, 126 insertions(+) create mode 100644 codec-memcache/src/test/java/io/netty/handler/codec/memcache/binary/FullMemcacheMessageRequestTest.java create mode 100644 codec-memcache/src/test/java/io/netty/handler/codec/memcache/binary/FullMemcacheMessageResponseTest.java diff --git a/codec-memcache/src/test/java/io/netty/handler/codec/memcache/binary/FullMemcacheMessageRequestTest.java b/codec-memcache/src/test/java/io/netty/handler/codec/memcache/binary/FullMemcacheMessageRequestTest.java new file mode 100644 index 0000000000..ad41f49ae1 --- /dev/null +++ b/codec-memcache/src/test/java/io/netty/handler/codec/memcache/binary/FullMemcacheMessageRequestTest.java @@ -0,0 +1,62 @@ +/* + * Copyright 2016 The Netty Project + * + * The Netty Project licenses this file to you 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 io.netty.handler.codec.memcache.binary; + +import io.netty.buffer.ByteBuf; +import io.netty.buffer.Unpooled; +import io.netty.channel.embedded.EmbeddedChannel; +import io.netty.util.CharsetUtil; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; + +import static org.junit.Assert.*; + +public class FullMemcacheMessageRequestTest { + + private EmbeddedChannel channel; + + @Before + public void setup() throws Exception { + channel = new EmbeddedChannel( + new BinaryMemcacheRequestEncoder(), + new BinaryMemcacheRequestDecoder(), + new BinaryMemcacheObjectAggregator(1024)); + } + + @After + public void teardown() throws Exception { + assertFalse(channel.finish()); + } + + @Test + public void testEncodeDecode() throws Exception { + ByteBuf key = Unpooled.wrappedBuffer("key".getBytes(CharsetUtil.UTF_8)); + ByteBuf content = Unpooled.wrappedBuffer("content".getBytes(CharsetUtil.UTF_8)); + ByteBuf extras = Unpooled.wrappedBuffer("extras".getBytes(CharsetUtil.UTF_8)); + FullBinaryMemcacheRequest req = new DefaultFullBinaryMemcacheRequest(key, extras, content); + assertTrue(channel.writeOutbound(req)); + // header + content + assertEquals(2, channel.outboundMessages().size()); + assertTrue(channel.writeInbound(channel.readOutbound(), channel.readOutbound())); + + FullBinaryMemcacheRequest read = channel.readInbound(); + assertEquals("key", read.key().toString(CharsetUtil.UTF_8)); + assertEquals("content", read.content().toString(CharsetUtil.UTF_8)); + assertEquals("extras", read.extras().toString(CharsetUtil.UTF_8)); + read.release(); + } +} diff --git a/codec-memcache/src/test/java/io/netty/handler/codec/memcache/binary/FullMemcacheMessageResponseTest.java b/codec-memcache/src/test/java/io/netty/handler/codec/memcache/binary/FullMemcacheMessageResponseTest.java new file mode 100644 index 0000000000..f8bf7e4a00 --- /dev/null +++ b/codec-memcache/src/test/java/io/netty/handler/codec/memcache/binary/FullMemcacheMessageResponseTest.java @@ -0,0 +1,64 @@ +/* + * Copyright 2016 The Netty Project + * + * The Netty Project licenses this file to you 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 io.netty.handler.codec.memcache.binary; + +import io.netty.buffer.ByteBuf; +import io.netty.buffer.Unpooled; +import io.netty.channel.embedded.EmbeddedChannel; +import io.netty.util.CharsetUtil; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertFalse; +import static org.junit.Assert.assertTrue; + +public class FullMemcacheMessageResponseTest { + + private EmbeddedChannel channel; + + @Before + public void setup() throws Exception { + channel = new EmbeddedChannel( + new BinaryMemcacheResponseEncoder(), + new BinaryMemcacheResponseDecoder(), + new BinaryMemcacheObjectAggregator(1024)); + } + + @After + public void teardown() throws Exception { + assertFalse(channel.finish()); + } + + @Test + public void testEncodeDecode() throws Exception { + ByteBuf key = Unpooled.wrappedBuffer("key".getBytes(CharsetUtil.UTF_8)); + ByteBuf content = Unpooled.wrappedBuffer("content".getBytes(CharsetUtil.UTF_8)); + ByteBuf extras = Unpooled.wrappedBuffer("extras".getBytes(CharsetUtil.UTF_8)); + FullBinaryMemcacheResponse resp = new DefaultFullBinaryMemcacheResponse(key, extras, content); + assertTrue(channel.writeOutbound(resp)); + // header + content + assertEquals(2, channel.outboundMessages().size()); + assertTrue(channel.writeInbound(channel.readOutbound(), channel.readOutbound())); + + FullBinaryMemcacheResponse read = channel.readInbound(); + assertEquals("key", read.key().toString(CharsetUtil.UTF_8)); + assertEquals("content", read.content().toString(CharsetUtil.UTF_8)); + assertEquals("extras", read.extras().toString(CharsetUtil.UTF_8)); + read.release(); + } +}