import org.opendaylight.controller.cluster.datastore.entityownership.messages.RegisterListenerLocal;
import org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal;
import org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterListenerLocal;
+import org.opendaylight.controller.cluster.datastore.entityownership.selectionstrategy.EntityOwnerSelectionStrategyConfig;
import org.opendaylight.controller.cluster.datastore.messages.GetShardDataTree;
import org.opendaylight.controller.cluster.datastore.utils.MockClusterWrapper;
import org.opendaylight.controller.md.cluster.datastore.model.SchemaContextHelper;
@Test
public void testEntityOwnershipShardCreated() throws Exception {
- DistributedEntityOwnershipService service = new DistributedEntityOwnershipService(dataStore);
+ DistributedEntityOwnershipService service = new DistributedEntityOwnershipService(dataStore,
+ EntityOwnerSelectionStrategyConfig.newBuilder().build());
service.start();
Future<ActorRef> future = dataStore.getActorContext().findLocalShardAsync(
@Test
public void testRegisterCandidate() throws Exception {
final TestShardBuilder shardBuilder = new TestShardBuilder();
- DistributedEntityOwnershipService service = new DistributedEntityOwnershipService(dataStore) {
+ DistributedEntityOwnershipService service = new DistributedEntityOwnershipService(dataStore,
+ EntityOwnerSelectionStrategyConfig.newBuilder().build()) {
@Override
protected EntityOwnershipShard.Builder newShardBuilder() {
return shardBuilder;
@Test
public void testCloseCandidateRegistration() throws Exception {
final TestShardBuilder shardBuilder = new TestShardBuilder();
- DistributedEntityOwnershipService service = new DistributedEntityOwnershipService(dataStore) {
+ DistributedEntityOwnershipService service = new DistributedEntityOwnershipService(dataStore,
+ EntityOwnerSelectionStrategyConfig.newBuilder().build()) {
@Override
protected EntityOwnershipShard.Builder newShardBuilder() {
return shardBuilder;
@Test
public void testListenerRegistration() {
final TestShardBuilder shardBuilder = new TestShardBuilder();
- DistributedEntityOwnershipService service = new DistributedEntityOwnershipService(dataStore) {
+ DistributedEntityOwnershipService service = new DistributedEntityOwnershipService(dataStore,
+ EntityOwnerSelectionStrategyConfig.newBuilder().build()) {
@Override
protected EntityOwnershipShard.Builder newShardBuilder() {
return shardBuilder;
@Test
public void testGetOwnershipState() throws Exception {
final TestShardBuilder shardBuilder = new TestShardBuilder();
- DistributedEntityOwnershipService service = new DistributedEntityOwnershipService(dataStore) {
+ DistributedEntityOwnershipService service = new DistributedEntityOwnershipService(dataStore,
+ EntityOwnerSelectionStrategyConfig.newBuilder().build()) {
@Override
protected EntityOwnershipShard.Builder newShardBuilder() {
return shardBuilder;
static class TestShardBuilder extends EntityOwnershipShard.Builder {
TestShardBuilder() {
- localMemberName("member-1");
+ localMemberName("member-1").ownerSelectionStrategyConfig(
+ EntityOwnerSelectionStrategyConfig.newBuilder().build());
}
private final AtomicReference<CountDownLatch> messageReceived = new AtomicReference<>();
@Override
public Props props() {
+ verify();
return Props.create(TestEntityOwnershipShard.class,this, messageClass, messageReceived,
receivedMessage, dataTree);
}