package org.infinispan.protostream.test;
import org.infinispan.protostream.Configuration;
import org.infinispan.protostream.ConfigurationBuilder;
import org.infinispan.protostream.DescriptorParserException;
import org.infinispan.protostream.ProtobufUtil;
import org.infinispan.protostream.SerializationContext;
import org.infinispan.protostream.domain.marshallers.MarshallerRegistration;
import java.io.IOException;
/**
* @author anistor@redhat.com
* @since 1.0
*/
public abstract class AbstractProtoStreamTest {
protected SerializationContext createContext() throws IOException, DescriptorParserException {
return createContext(new ConfigurationBuilder().build());
}
protected SerializationContext createContext(Configuration cfg) throws IOException, DescriptorParserException {
SerializationContext ctx = ProtobufUtil.newSerializationContext(cfg);
MarshallerRegistration.registerMarshallers(ctx);
return ctx;
}
}