package com.linkedin.databus.client.netty;
/*
*
* Copyright 2013 LinkedIn Corp. All rights reserved
*
* 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.
*
*/
import java.io.IOException;
import java.io.OutputStream;
import java.io.StringWriter;
import java.net.SocketAddress;
import java.nio.channels.WritableByteChannel;
import java.util.HashMap;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import junit.framework.Assert;
import org.codehaus.jackson.JsonGenerationException;
import org.codehaus.jackson.map.JsonMappingException;
import org.codehaus.jackson.map.ObjectMapper;
import org.jboss.netty.buffer.ChannelBuffer;
import org.jboss.netty.buffer.ChannelBufferOutputStream;
import org.jboss.netty.buffer.ChannelBuffers;
import org.jboss.netty.handler.codec.http.HttpChunk;
import org.jboss.netty.handler.codec.http.HttpRequest;
import org.jboss.netty.handler.codec.http.HttpResponse;
import com.linkedin.databus.core.Checkpoint;
import com.linkedin.databus.core.DbusEventBuffer;
import com.linkedin.databus.core.DbusEventV1Factory;
import com.linkedin.databus.core.Encoding;
import com.linkedin.databus.core.OffsetNotFoundException;
import com.linkedin.databus.core.ScnNotFoundException;
import com.linkedin.databus.core.StreamEventsArgs;
import com.linkedin.databus.core.monitoring.mbean.DbusEventsStatisticsCollector;
import com.linkedin.databus2.core.container.request.RegisterResponseEntry;
import com.linkedin.databus2.core.filter.AllowAllDbusFilter;
import com.linkedin.databus2.test.container.SimpleObjectCaptureHandler;
import com.linkedin.databus2.test.container.SimpleTestServerConnection;
public class NettyTestUtils
{
public static void sendServerResponses(SimpleTestServerConnection srv,
SocketAddress clientAddr,
HttpResponse sourcesResp,
HttpChunk body,
long timeout)
{
srv.sendServerResponse(clientAddr, sourcesResp, timeout);
srv.sendServerResponse(clientAddr, body, timeout);
srv.sendServerResponse(clientAddr, HttpChunk.LAST_CHUNK, timeout);
}
public static void sendServerResponses(SimpleTestServerConnection srv,
SocketAddress clientAddr,
HttpResponse sourcesResp,
HttpChunk body)
{
sendServerResponses(srv, clientAddr, sourcesResp, body, 1000);
}
public static String generateRegisterResponse(RegisterResponseEntry... entries)
throws JsonGenerationException, JsonMappingException, IOException
{
ObjectMapper mapper = new ObjectMapper();
StringWriter w = new StringWriter();
mapper.writeValue(w, entries);
w.close();
return w.toString();
}
public static String generateRegisterResponseV4(HashMap<String, List<Object>> entries)
throws JsonGenerationException, JsonMappingException, IOException
{
ObjectMapper mapper = new ObjectMapper();
StringWriter w = new StringWriter();
mapper.writeValue(w, entries);
w.close();
return w.toString();
}
public static ChannelBuffer streamToChannelBuffer(DbusEventBuffer buf, Checkpoint cp,
int maxSize,
DbusEventsStatisticsCollector stats)
throws ScnNotFoundException, OffsetNotFoundException, IOException
{
ChannelBuffer tmpBuf = ChannelBuffers.buffer(new DbusEventV1Factory().getByteOrder(), maxSize);
OutputStream tmpOS = new ChannelBufferOutputStream(tmpBuf);
WritableByteChannel tmpChannel = java.nio.channels.Channels.newChannel(tmpOS);
StreamEventsArgs args = new StreamEventsArgs(maxSize).setStatsCollector(stats);
buf.streamEvents(cp, tmpChannel, args);
tmpChannel.close();
return tmpBuf;
}
public static Matcher waitForHttpRequest(SimpleObjectCaptureHandler objCapture,
String regex,
long timeout)
{
Pattern pattern = Pattern.compile(regex);
Assert.assertTrue(objCapture.waitForMessage(timeout, 0));
Object msgObj = objCapture.getMessages().get(0);
Assert.assertTrue(msgObj instanceof HttpRequest);
HttpRequest msgReq = (HttpRequest)msgObj;
Matcher result = pattern.matcher(msgReq.getUri());
Assert.assertTrue(result.matches());
return result;
}
}