new ChildShardContext(DOM_DATA_TREE_IDENTIFIER, READABLE_WRITEABLE_DOM_DATA_TREE_SHARD);
private static final Map<DOMDataTreeIdentifier, ChildShardContext> CHILD_SHARDS =
ImmutableMap.of(DOM_DATA_TREE_IDENTIFIER, CHILD_SHARD_CONTEXT);
+ private InMemoryDOMDataTreeShardProducer mockProducer;
@Before
public void setUp() throws Exception {
final InMemoryDOMDataTreeShardChangePublisher inMemoryDOMDataTreeShardChangePublisher =
new InMemoryDOMDataTreeShardChangePublisher(MoreExecutors.newDirectExecutorService(), 1, DATA_TREE,
YANG_INSTANCE_IDENTIFIER, CHILD_SHARDS);
+ mockProducer = mock(InMemoryDOMDataTreeShardProducer.class);
+ doNothing().when(mockProducer).transactionReady(any(), any());
+ doNothing().when(mockProducer).onTransactionCommited(any());
+ doNothing().when(mockProducer).transactionAborted(any());
inmemoryDOMDataTreeShardWriteTransaction =
- new InmemoryDOMDataTreeShardWriteTransaction(shardDataModification, DATA_TREE,
+ new InmemoryDOMDataTreeShardWriteTransaction(mockProducer, shardDataModification, DATA_TREE,
inMemoryDOMDataTreeShardChangePublisher,
MoreExecutors.listeningDecorator(Executors.newSingleThreadExecutor()));
}