Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -242,7 +242,8 @@ public ReplicationGroupAdmin getReplicationGroupAdmin() {
LOG.info("addresses is empty");
return null;
}
return new ReplicationGroupAdmin(PALO_JOURNAL_GROUP, addresses);
return new ReplicationGroupAdmin(PALO_JOURNAL_GROUP, addresses,
replicationConfig.getRepNetConfig());
}

// Return a handle to the epochDB
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,12 @@
import com.sleepycat.je.OperationStatus;
import com.sleepycat.je.rep.InsufficientAcksException;
import com.sleepycat.je.rep.ReplicatedEnvironment;
import com.sleepycat.je.rep.ReplicationBasicConfig;
import com.sleepycat.je.rep.ReplicationConfig;
import com.sleepycat.je.rep.impl.RepImpl;
import com.sleepycat.je.rep.net.DataChannel;
import com.sleepycat.je.rep.net.DataChannelFactory;
import com.sleepycat.je.rep.net.InstanceParams;
import com.sleepycat.je.rep.util.ReplicationGroupAdmin;
import org.apache.commons.io.FileUtils;
import org.apache.logging.log4j.LogManager;
Expand All @@ -44,20 +49,24 @@
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.RepeatedTest;
import org.junit.jupiter.api.Test;
import org.mockito.MockedStatic;
import org.mockito.Mockito;

import java.io.File;
import java.io.IOException;
import java.lang.reflect.Field;
import java.net.DatagramSocket;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.SocketException;
import java.nio.channels.SocketChannel;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.security.SecureRandom;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
Expand All @@ -66,6 +75,49 @@ public class BDBEnvironmentTest {
private static final Logger LOG = LogManager.getLogger(BDBEnvironmentTest.class);
private static List<String> tmpDirs = new ArrayList<>();

public static class TestDataChannelFactory implements DataChannelFactory {
public TestDataChannelFactory(InstanceParams params) {
}

@Override
public DataChannel acceptChannel(SocketChannel socketChannel) {
return null;
}

@Override
public DataChannel connect(InetSocketAddress addr, InetSocketAddress localAddr,
ConnectOptions connectOptions) {
return null;
}
}

@Test
public void testReplicationGroupAdminReusesConfiguredNetworkFactory() {
ReplicationBasicConfig networkConfig = new ReplicationBasicConfig();
networkConfig.setChannelFactoryClass(TestDataChannelFactory.class.getName());
ReplicationConfig replicationConfig = new ReplicationConfig();
replicationConfig.setRepNetConfig(networkConfig);

BDBEnvironment bdbEnvironment = new BDBEnvironment(true, false);
Deencapsulation.setField(bdbEnvironment, "replicationConfig", replicationConfig);

Env mockEnv = Mockito.mock(Env.class);
Frontend frontend = new Frontend(FrontendNodeType.FOLLOWER,
"test-fe", "127.0.0.1", 9010);
frontend.setIsAlive(true);
Mockito.when(mockEnv.getFrontends(FrontendNodeType.FOLLOWER))
.thenReturn(Collections.singletonList(frontend));

try (MockedStatic<Env> mockedEnvStatic = Mockito.mockStatic(Env.class)) {
mockedEnvStatic.when(Env::getCurrentEnv).thenReturn(mockEnv);

ReplicationGroupAdmin admin = bdbEnvironment.getReplicationGroupAdmin();

DataChannelFactory channelFactory = Deencapsulation.getField(admin, "channelFactory");
Assertions.assertInstanceOf(TestDataChannelFactory.class, channelFactory);
}
}

public static String createTmpDir() throws Exception {
String dorisHome = System.getenv("DORIS_HOME");
if (Strings.isNullOrEmpty(dorisHome)) {
Expand Down
Loading