* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
-
package org.opendaylight.controller.sal.binding.test.bugfix;
import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertTrue;
-import com.google.common.util.concurrent.SettableFuture;
+import static org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType.OPERATIONAL;
+
+import com.google.common.collect.ImmutableSet;
+import java.util.Set;
import java.util.concurrent.TimeUnit;
import org.junit.Test;
-import org.opendaylight.controller.md.sal.common.api.data.DataChangeEvent;
-import org.opendaylight.controller.sal.binding.api.data.DataChangeListener;
-import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
-import org.opendaylight.controller.sal.binding.test.AbstractDataServiceTest;
+import org.opendaylight.controller.md.sal.binding.api.DataBroker;
+import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
+import org.opendaylight.controller.md.sal.binding.test.AbstractDataTreeChangeListenerTest;
+import org.opendaylight.mdsal.binding.spec.reflect.BindingReflections;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.md.sal.test.augment.rev140709.TreeComplexUsesAugment;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.md.sal.test.augment.rev140709.TreeComplexUsesAugmentBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.md.sal.test.augment.rev140709.complex.from.grouping.ContainerWithUsesBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.md.sal.test.list.rev140701.two.level.list.TopLevelList;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.md.sal.test.list.rev140701.two.level.list.TopLevelListBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.md.sal.test.list.rev140701.two.level.list.TopLevelListKey;
-import org.opendaylight.yangtools.concepts.ListenerRegistration;
-import org.opendaylight.yangtools.yang.binding.DataObject;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.opendaylight.yangtools.yang.binding.YangModuleInfo;
-public class WriteParentListenAugmentTest extends AbstractDataServiceTest {
+public class WriteParentListenAugmentTest extends AbstractDataTreeChangeListenerTest {
private static final String TLL_NAME = "foo";
private static final TopLevelListKey TLL_KEY = new TopLevelListKey(TLL_NAME);
- private static final InstanceIdentifier<TopLevelList> TLL_INSTANCE_ID_BA = InstanceIdentifier.builder(Top.class) //
- .child(TopLevelList.class, TLL_KEY).toInstance();
+ private static final InstanceIdentifier<TopLevelList> TLL_INSTANCE_ID_BA = InstanceIdentifier.builder(Top.class)
+ .child(TopLevelList.class, TLL_KEY).build();
private static final InstanceIdentifier<TreeComplexUsesAugment> AUGMENT_WILDCARDED_PATH = InstanceIdentifier
- .builder(Top.class).child(TopLevelList.class).augmentation(TreeComplexUsesAugment.class).toInstance();
+ .builder(Top.class).child(TopLevelList.class).augmentation(TreeComplexUsesAugment.class).build();
private static final InstanceIdentifier<TreeComplexUsesAugment> AUGMENT_TLL_PATH = InstanceIdentifier
- .builder(Top.class).child(TopLevelList.class, TLL_KEY).augmentation(TreeComplexUsesAugment.class).toInstance();
+ .builder(Top.class).child(TopLevelList.class, TLL_KEY).augmentation(TreeComplexUsesAugment.class).build();
+
+ @Override
+ protected Set<YangModuleInfo> getModuleInfos() throws Exception {
+ return ImmutableSet.of(BindingReflections.getModuleInfo(Top.class),
+ BindingReflections.getModuleInfo(TreeComplexUsesAugment.class));
+ }
@Test
public void writeNodeListenAugment() throws Exception {
- final SettableFuture<DataChangeEvent<InstanceIdentifier<?>, DataObject>> event = SettableFuture.create();
-
- ListenerRegistration<DataChangeListener> dclRegistration = baDataService.registerDataChangeListener(
- AUGMENT_WILDCARDED_PATH, new DataChangeListener() {
+ DataBroker dataBroker = getDataBroker();
- @Override
- public void onDataChanged(final DataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
- event.set(change);
- }
- });
+ final TreeComplexUsesAugment treeComplexUsesAugment = treeComplexUsesAugment("one");
- DataModificationTransaction modification = baDataService.beginTransaction();
+ final TestListener<TreeComplexUsesAugment> listener = createListener(OPERATIONAL, AUGMENT_WILDCARDED_PATH,
+ added(AUGMENT_TLL_PATH, treeComplexUsesAugment));
- TopLevelList tll = new TopLevelListBuilder() //
- .setKey(TLL_KEY) //
- .addAugmentation(TreeComplexUsesAugment.class, treeComplexUsesAugment("one")).build();
- modification.putOperationalData(TLL_INSTANCE_ID_BA, tll);
- modification.commit().get();
+ final WriteTransaction transaction = dataBroker.newWriteOnlyTransaction();
- DataChangeEvent<InstanceIdentifier<?>, DataObject> receivedEvent = event.get(1000, TimeUnit.MILLISECONDS);
- assertTrue(receivedEvent.getCreatedOperationalData().containsKey(AUGMENT_TLL_PATH));
+ TopLevelList tll = new TopLevelListBuilder().withKey(TLL_KEY)
+ .addAugmentation(TreeComplexUsesAugment.class, treeComplexUsesAugment).build();
+ transaction.put(OPERATIONAL, TLL_INSTANCE_ID_BA, tll, true);
+ transaction.submit().get(5, TimeUnit.SECONDS);
- dclRegistration.close();
+ listener.verify();
- DataModificationTransaction mod2 = baDataService.beginTransaction();
- mod2.putOperationalData(AUGMENT_TLL_PATH, treeComplexUsesAugment("two"));
- mod2.commit().get();
+ final WriteTransaction transaction2 = dataBroker.newWriteOnlyTransaction();
+ transaction2.put(OPERATIONAL, AUGMENT_TLL_PATH, treeComplexUsesAugment("two"));
+ transaction2.submit().get(5, TimeUnit.SECONDS);
- TreeComplexUsesAugment readedAug = (TreeComplexUsesAugment) baDataService.readOperationalData(AUGMENT_TLL_PATH);
+ TreeComplexUsesAugment readedAug = dataBroker.newReadOnlyTransaction().read(
+ OPERATIONAL, AUGMENT_TLL_PATH).get(5, TimeUnit.SECONDS).get();
assertEquals("two", readedAug.getContainerWithUses().getLeafFromGrouping());
-
}
- private TreeComplexUsesAugment treeComplexUsesAugment(final String value) {
- return new TreeComplexUsesAugmentBuilder() //
- .setContainerWithUses(new ContainerWithUsesBuilder().setLeafFromGrouping(value).build()) //
+ private static TreeComplexUsesAugment treeComplexUsesAugment(final String value) {
+ return new TreeComplexUsesAugmentBuilder()
+ .setContainerWithUses(new ContainerWithUsesBuilder().setLeafFromGrouping(value).build())
.build();
}
-}
\ No newline at end of file
+}