ioam: export plugin - support for ioam-export plugin in hc2vpp

Change-Id: I2314a2de3b9c2a1d9c2e5950659e44cb9bdda985
Signed-off-by: Sagar Srivastav <sagsriva@cisco.com>

7 files changed, 403 insertions, 18 deletions

diff --git a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/VppIoamModule.java b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/VppIoamModule.java
index 204d003c9..228653c1c 100755
--- a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/VppIoamModule.java
+++ b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/VppIoamModule.java
@@ -20,12 +20,15 @@ import com.google.inject.AbstractModule;
 import com.google.inject.Provider;
 import com.google.inject.Singleton;
 import com.google.inject.multibindings.Multibinder;
-import io.fd.hc2vpp.vppioam.impl.oper.VppIoamReaderFactory;
-import io.fd.honeycomb.translate.read.ReaderFactory;
-import io.fd.honeycomb.translate.write.WriterFactory;
 import io.fd.hc2vpp.vppioam.impl.config.VppIoamWriterFactory;
+import io.fd.hc2vpp.vppioam.impl.oper.VppIoamReaderFactory;
+import io.fd.hc2vpp.vppioam.impl.util.JVppIoamExportProvider;
 import io.fd.hc2vpp.vppioam.impl.util.JVppIoamPotProvider;
 import io.fd.hc2vpp.vppioam.impl.util.JVppIoamTraceProvider;
+import io.fd.honeycomb.translate.read.ReaderFactory;
+import io.fd.honeycomb.translate.write.WriterFactory;
+import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexport;
+import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexportFacade;
 import io.fd.vpp.jvpp.ioampot.future.FutureJVppIoampot;
 import io.fd.vpp.jvpp.ioampot.future.FutureJVppIoampotFacade;
 import io.fd.vpp.jvpp.ioamtrace.future.FutureJVppIoamtrace;
@@ -41,16 +44,19 @@ public final class VppIoamModule extends AbstractModule {
     private static final Logger LOG = LoggerFactory.getLogger(VppIoamModule.class);
     private final Class<? extends Provider<FutureJVppIoamtraceFacade>> jvppIoamTraceProviderClass;
     private final Class<? extends Provider<FutureJVppIoampotFacade>> jvppIoamPotProviderClass;
+    private final Class<? extends Provider<FutureJVppIoamexportFacade>> jvppIoamExportProviderClass;
 
     public VppIoamModule() {
-        this(JVppIoamTraceProvider.class, JVppIoamPotProvider.class);
+        this(JVppIoamTraceProvider.class, JVppIoamPotProvider.class, JVppIoamExportProvider.class);
     }
 
     @VisibleForTesting
     VppIoamModule(Class<? extends Provider<FutureJVppIoamtraceFacade>> jvppIoamTraceProvider,
-                  Class<? extends Provider<FutureJVppIoampotFacade>> jvppIoamPotProviderClass) {
+                  Class<? extends Provider<FutureJVppIoampotFacade>> jvppIoamPotProviderClass,
+                  Class<? extends Provider<FutureJVppIoamexportFacade>> jvppIoamExportProviderClass) {
         this.jvppIoamTraceProviderClass = jvppIoamTraceProvider;
         this.jvppIoamPotProviderClass = jvppIoamPotProviderClass;
+        this.jvppIoamExportProviderClass = jvppIoamExportProviderClass;
     }
 
     @Override
@@ -60,6 +66,7 @@ public final class VppIoamModule extends AbstractModule {
         // Bind to Plugin's JVPP.
         bind(FutureJVppIoamtrace.class).toProvider(jvppIoamTraceProviderClass).in(Singleton.class);
         bind(FutureJVppIoampot.class).toProvider(jvppIoamPotProviderClass).in(Singleton.class);
+        bind(FutureJVppIoamexport.class).toProvider(jvppIoamExportProviderClass).in(Singleton.class);
 
         // Below are classes picked up by HC framework
         Multibinder.newSetBinder(binder(), WriterFactory.class).addBinding().to(VppIoamWriterFactory.class);
diff --git a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizer.java b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizer.java
new file mode 100644
index 000000000..2172750d4
--- /dev/null
+++ b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizer.java
@@ -0,0 +1,95 @@
+/*
+ * Copyright (c) 2016 Cisco and its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.fd.hc2vpp.vppioam.impl.config;
+
+import io.fd.hc2vpp.common.translate.util.Ipv4Translator;
+import io.fd.hc2vpp.common.translate.util.JvppReplyConsumer;
+import io.fd.hc2vpp.vppioam.impl.util.FutureJVppIoamexportCustomizer;
+import io.fd.honeycomb.translate.spi.write.WriterCustomizer;
+import io.fd.honeycomb.translate.write.WriteContext;
+import io.fd.honeycomb.translate.write.WriteFailedException;
+import io.fd.vpp.jvpp.ioamexport.dto.IoamExportIp6EnableDisable;
+import io.fd.vpp.jvpp.ioamexport.dto.IoamExportIp6EnableDisableReply;
+import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexport;
+import javax.annotation.Nonnull;
+import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.export.rev170206.IoamExport;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class IoamExportWriterCustomizer extends FutureJVppIoamexportCustomizer
+        implements WriterCustomizer<IoamExport>, JvppReplyConsumer, Ipv4Translator {
+
+    private static final Logger LOG = LoggerFactory.getLogger(IoamExportWriterCustomizer.class);
+
+    public IoamExportWriterCustomizer(FutureJVppIoamexport jVppIoamexport){
+        super(jVppIoamexport);
+    }
+
+    @Override
+    public void writeCurrentAttributes(@Nonnull final InstanceIdentifier<IoamExport> instanceIdentifier,
+                                       @Nonnull final IoamExport ioamExport, @Nonnull final WriteContext writeContext)
+            throws WriteFailedException {
+        addExportProfile(ioamExport,instanceIdentifier);
+        LOG.info("Export profile {} created, id: {}", ioamExport, instanceIdentifier);
+    }
+
+    @Override
+    public void updateCurrentAttributes(@Nonnull final InstanceIdentifier<IoamExport> instanceIdentifier,
+                                        @Nonnull final IoamExport dataBefore, @Nonnull final IoamExport dataAfter,
+                                        @Nonnull final WriteContext writeContext) throws WriteFailedException {
+        addExportProfile(dataAfter,instanceIdentifier);
+        LOG.info("Export profile {} updated , id: {}", dataAfter, instanceIdentifier);
+    }
+
+    @Override
+    public void deleteCurrentAttributes(@Nonnull final InstanceIdentifier<IoamExport> instanceIdentifier,
+                                        @Nonnull final IoamExport ioamExport, @Nonnull final WriteContext writeContext)
+            throws WriteFailedException {
+        deleteExportProfile(ioamExport,instanceIdentifier);
+        LOG.info("Export profile {} deleted, id: {}", ioamExport, instanceIdentifier);
+    }
+
+    private IoamExportIp6EnableDisableReply addExportProfile(IoamExport ioamExport, InstanceIdentifier<IoamExport> id)
+            throws WriteFailedException {
+        IoamExportIp6EnableDisable request = new IoamExportIp6EnableDisable();
+        if (ioamExport.isDisable()==null)
+            request.isDisable = 1;
+        else
+            request.isDisable = (byte)(ioamExport.isDisable()?1:0);
+        request.srcAddress = ipv4AddressNoZoneToArray(ioamExport.getSourceAddress().getValue());
+        request.collectorAddress = ipv4AddressNoZoneToArray(ioamExport.getCollectorAddress().getValue());
+        return getReplyForCreate(getFutureJVppIoamexport()
+                .ioamExportIp6EnableDisable(request)
+                .toCompletableFuture(),id,ioamExport);
+
+    }
+
+    private IoamExportIp6EnableDisableReply deleteExportProfile(IoamExport ioamExport,
+                                                                InstanceIdentifier<IoamExport> id)
+            throws WriteFailedException {
+        IoamExportIp6EnableDisable request = new IoamExportIp6EnableDisable();
+        request.isDisable = 1;     //disable when deleted
+        request.srcAddress = ipv4AddressNoZoneToArray(ioamExport.getSourceAddress().getValue());
+        request.collectorAddress = ipv4AddressNoZoneToArray(ioamExport.getCollectorAddress().getValue());
+
+        return getReplyForDelete(getFutureJVppIoamexport()
+                .ioamExportIp6EnableDisable(request)
+                .toCompletableFuture(),id);
+    }
+
+}
diff --git a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/VppIoamWriterFactory.java b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/VppIoamWriterFactory.java
index 15d312ff7..ba105eb39 100755
--- a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/VppIoamWriterFactory.java
+++ b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/VppIoamWriterFactory.java
@@ -19,11 +19,14 @@ package io.fd.hc2vpp.vppioam.impl.config;
 import com.google.common.collect.Sets;
 import com.google.inject.Inject;
 import io.fd.honeycomb.translate.impl.write.GenericListWriter;
+import io.fd.honeycomb.translate.impl.write.GenericWriter;
 import io.fd.honeycomb.translate.write.WriterFactory;
 import io.fd.honeycomb.translate.write.registry.ModifiableWriterRegistryBuilder;
+import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexport;
 import io.fd.vpp.jvpp.ioampot.future.FutureJVppIoampot;
 import io.fd.vpp.jvpp.ioamtrace.future.FutureJVppIoamtrace;
 import javax.annotation.Nonnull;
+import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.export.rev170206.IoamExport;
 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.trace.rev160512.IoamTraceConfig;
 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.trace.rev160512.ioam.trace.config.TraceConfig;
 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.trace.rev160512.ioam.trace.config.trace.config.NodeInterfaces;
@@ -38,12 +41,16 @@ public class VppIoamWriterFactory implements WriterFactory {
     private final FutureJVppIoamtrace jVppIoamtrace;
     @Nonnull
     private final FutureJVppIoampot jVppIoampot;
+    @Nonnull
+    private final FutureJVppIoamexport jVppIoamexport;
 
     @Inject
     public VppIoamWriterFactory(@Nonnull final FutureJVppIoamtrace jVppIoamtrace,
-                                @Nonnull final FutureJVppIoampot jVppIoampot) {
+                                @Nonnull final FutureJVppIoampot jVppIoampot,
+                                @Nonnull final FutureJVppIoamexport jVppIoamexport) {
         this.jVppIoamtrace = jVppIoamtrace;
         this.jVppIoampot = jVppIoampot;
+        this.jVppIoamexport = jVppIoamexport;
     }
 
     @Override
@@ -60,6 +67,10 @@ public class VppIoamWriterFactory implements WriterFactory {
         registry.subtreeAdd(Sets.newHashSet(InstanceIdentifier.create(PotProfileSet.class)
                         .child(PotProfileList.class)),
                 new GenericListWriter<>(potId, new IoamPotWriterCustomizer(jVppIoampot)));
+        //Export Config
+        final InstanceIdentifier<IoamExport> exportId =
+                InstanceIdentifier.create(IoamExport.class);
+        registry.add(new GenericWriter<>(exportId,new IoamExportWriterCustomizer(jVppIoamexport)));
 
     }
 }
diff --git a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/FutureJVppIoamexportCustomizer.java b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/FutureJVppIoamexportCustomizer.java
new file mode 100644
index 000000000..adc2ffc10
--- /dev/null
+++ b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/FutureJVppIoamexportCustomizer.java
@@ -0,0 +1,41 @@
+/*
+ * Copyright (c) 2016 Cisco and its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.fd.hc2vpp.vppioam.impl.util;
+
+import com.google.common.base.Preconditions;
+import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexport;
+import javax.annotation.Nonnull;
+
+public abstract class FutureJVppIoamexportCustomizer {
+
+    private final FutureJVppIoamexport futureJVppIoamexport;
+
+    public FutureJVppIoamexportCustomizer (@Nonnull final FutureJVppIoamexport futureJVppIoamexport) {
+        this.futureJVppIoamexport = Preconditions.checkNotNull(futureJVppIoamexport,
+                "futureJVppIoamexport should not be null");
+    }
+
+    /**
+     * Get Ioam POT Api reference
+     *
+     * @return Ioam POT Api reference
+     */
+    public FutureJVppIoamexport getFutureJVppIoamexport() {
+        return futureJVppIoamexport;
+    }
+
+}
diff --git a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/JVppIoamExportProvider.java b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/JVppIoamExportProvider.java
new file mode 100644
index 000000000..fa4d20004
--- /dev/null
+++ b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/util/JVppIoamExportProvider.java
@@ -0,0 +1,55 @@
+/*
+ * Copyright (c) 2016 Cisco and its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.fd.hc2vpp.vppioam.impl.util;
+
+import com.google.inject.Inject;
+import io.fd.honeycomb.infra.distro.ProviderTrait;
+import io.fd.vpp.jvpp.JVppRegistry;
+import io.fd.vpp.jvpp.ioamexport.JVppIoamexportImpl;
+import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexportFacade;
+import java.io.IOException;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class JVppIoamExportProvider extends ProviderTrait<FutureJVppIoamexportFacade> {
+
+    private static final Logger LOG = LoggerFactory.getLogger(JVppIoamExportProvider.class);
+
+    @Inject
+    private JVppRegistry registry;
+
+    @Override
+    protected FutureJVppIoamexportFacade create() {
+        try {
+            final JVppIoamexportImpl jVppIoamexport = new JVppIoamexportImpl();
+            // Free jvpp-ioam-export plugin's resources on shutdown
+            Runtime.getRuntime().addShutdownHook(new Thread() {
+                @Override
+                public void run() {
+                    LOG.info("Unloading jvpp-ioam-export plugin");
+                    jVppIoamexport.close();
+                    LOG.info("Successfully unloaded jvpp-ioam-export plugin");
+                }
+            });
+
+            LOG.info("Successfully loaded jvpp-ioam-export plugin");
+            return new FutureJVppIoamexportFacade(registry, jVppIoamexport);
+        } catch (IOException e) {
+            throw new IllegalStateException("Unable to open VPP management connection", e);
+        }
+    }
+}
diff --git a/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/VppIoamModuleTest.java b/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/VppIoamModuleTest.java
index c11ee2dc4..45b673218 100644
--- a/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/VppIoamModuleTest.java
+++ b/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/VppIoamModuleTest.java
@@ -16,6 +16,14 @@
 
 package io.fd.hc2vpp.vppioam.impl;
 
+import static org.hamcrest.CoreMatchers.is;
+import static org.hamcrest.CoreMatchers.not;
+import static org.hamcrest.Matchers.empty;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertThat;
+import static org.mockito.Mockito.mock;
+import static org.mockito.MockitoAnnotations.initMocks;
+
 import com.google.inject.Guice;
 import com.google.inject.Inject;
 import com.google.inject.Provider;
@@ -27,24 +35,16 @@ import io.fd.honeycomb.translate.impl.write.registry.FlatWriterRegistryBuilder;
 import io.fd.honeycomb.translate.read.ReaderFactory;
 import io.fd.honeycomb.translate.write.WriterFactory;
 import io.fd.vpp.jvpp.JVppRegistry;
+import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexportFacade;
 import io.fd.vpp.jvpp.ioampot.future.FutureJVppIoampotFacade;
 import io.fd.vpp.jvpp.ioamtrace.future.FutureJVppIoamtraceFacade;
+import java.util.HashSet;
+import java.util.Set;
 import org.junit.Before;
 import org.junit.Test;
 import org.mockito.Mock;
 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
 
-import java.util.HashSet;
-import java.util.Set;
-
-import static org.hamcrest.CoreMatchers.is;
-import static org.hamcrest.CoreMatchers.not;
-import static org.hamcrest.Matchers.empty;
-import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertThat;
-import static org.mockito.Mockito.mock;
-import static org.mockito.MockitoAnnotations.initMocks;
-
 
 public class VppIoamModuleTest {
 
@@ -73,7 +73,10 @@ public class VppIoamModuleTest {
 
         initMocks(this);
 
-        Guice.createInjector(new VppIoamModule(MockJVppIoamTraceProvider.class,MockJVppIoamPotProvider.class), BoundFieldModule.of(this)).injectMembers(this);
+        Guice.createInjector(new VppIoamModule(MockJVppIoamTraceProvider.class,
+                MockJVppIoamPotProvider.class,
+                        MockJVppIoamExportProvider.class),
+                BoundFieldModule.of(this)).injectMembers(this);
     }
 
     @Test
@@ -109,5 +112,13 @@ public class VppIoamModuleTest {
             return mock(FutureJVppIoampotFacade.class);
         }
     }
+
+    private static final class MockJVppIoamExportProvider implements Provider<FutureJVppIoamexportFacade> {
+
+        @Override
+        public FutureJVppIoamexportFacade get() {
+            return mock(FutureJVppIoamexportFacade.class);
+        }
+    }
 }
 
diff --git a/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizerTest.java b/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizerTest.java
new file mode 100644
index 000000000..b2dfaf308
--- /dev/null
+++ b/ioam/impl/src/test/java/io/fd/hc2vpp/vppioam/impl/config/IoamExportWriterCustomizerTest.java
@@ -0,0 +1,165 @@
+/*
+ * Copyright (c) 2016 Cisco and/or its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.fd.hc2vpp.vppioam.impl.config;
+
+import static org.junit.Assert.fail;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.verify;
+
+import io.fd.hc2vpp.common.test.write.WriterCustomizerTest;
+import io.fd.hc2vpp.common.translate.util.Ipv4Translator;
+import io.fd.honeycomb.translate.write.WriteFailedException;
+import io.fd.vpp.jvpp.ioamexport.dto.IoamExportIp6EnableDisable;
+import io.fd.vpp.jvpp.ioamexport.dto.IoamExportIp6EnableDisableReply;
+import io.fd.vpp.jvpp.ioamexport.future.FutureJVppIoamexport;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.export.rev170206.IoamExport;
+import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.ioam.sb.export.rev170206.IoamExportBuilder;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+
+public class IoamExportWriterCustomizerTest extends WriterCustomizerTest
+        implements Ipv4Translator{
+
+    @Mock
+    FutureJVppIoamexport jVppIoamexport;
+
+    private IoamExportWriterCustomizer customizer;
+
+    private static final Logger LOG = LoggerFactory.getLogger(IoamExportWriterCustomizerTest.class);
+
+    @Override
+    public void setUpTest() throws Exception {
+        customizer = new IoamExportWriterCustomizer(jVppIoamexport);
+    }
+
+    private static IoamExport generateExportProfile(){
+        IoamExportBuilder builder = new IoamExportBuilder();
+        builder.setDisable(false);
+        builder.setSourceAddress(new Ipv4Address("127.0.0.1"));
+        builder.setCollectorAddress(new Ipv4Address("127.0.0.2"));
+        return builder.build();
+    }
+
+    private void whenExportAddThenSuccess() {
+        final IoamExportIp6EnableDisableReply reply = new IoamExportIp6EnableDisableReply();
+        reply.context = 1;
+        doReturn(future(reply)).when(jVppIoamexport).ioamExportIp6EnableDisable(any(IoamExportIp6EnableDisable.class));
+    }
+
+    private void whenExportAddThenFailure() {
+        doReturn(failedFuture()).when(jVppIoamexport).ioamExportIp6EnableDisable(any(IoamExportIp6EnableDisable.class));
+    }
+
+    private void whenExportDelThenSuccess() {
+        final IoamExportIp6EnableDisableReply reply = new IoamExportIp6EnableDisableReply();
+        reply.context = 1;
+        doReturn(future(reply)).when(jVppIoamexport).ioamExportIp6EnableDisable(any(IoamExportIp6EnableDisable.class));
+    }
+
+    private void whenExportDelThenFailure() {
+        doReturn(failedFuture()).when(jVppIoamexport).ioamExportIp6EnableDisable(any(IoamExportIp6EnableDisable.class));
+    }
+
+    private IoamExportIp6EnableDisable generateIoamExportIp6EnableDisable(boolean disable) {
+        IoamExportIp6EnableDisable request = new IoamExportIp6EnableDisable();
+        request.isDisable = (byte)(disable?1:0);
+        request.srcAddress = ipv4AddressNoZoneToArray("127.0.0.1");
+        request.collectorAddress = ipv4AddressNoZoneToArray("127.0.0.2");
+
+        return request;
+    }
+
+    @Test
+    public void testCreate() throws Exception {
+        final IoamExport ioamExport = generateExportProfile();
+        final InstanceIdentifier<IoamExport> id = InstanceIdentifier.create(IoamExport.class);
+
+        whenExportAddThenSuccess();
+
+        customizer.writeCurrentAttributes(id, ioamExport, writeContext);
+
+        verify(jVppIoamexport).ioamExportIp6EnableDisable(generateIoamExportIp6EnableDisable(false));
+    }
+
+    @Test
+    public void testCreateFailed() throws Exception {
+        final IoamExport ioamExport = generateExportProfile();
+        final InstanceIdentifier<IoamExport> id = InstanceIdentifier.create(IoamExport.class);
+
+        whenExportAddThenFailure();
+
+        try {
+            customizer.writeCurrentAttributes(id, ioamExport, writeContext);
+        } catch (WriteFailedException e) {
+            verify(jVppIoamexport).ioamExportIp6EnableDisable(generateIoamExportIp6EnableDisable(false));
+
+            return;
+        }
+        fail("WriteFailedException.CreateFailedException was expected");
+    }
+
+    @Test
+    public void testDelete() throws Exception {
+        final IoamExport ioamExport = generateExportProfile();
+        final InstanceIdentifier<IoamExport> id = InstanceIdentifier.create(IoamExport.class);
+
+        whenExportDelThenSuccess();
+
+        customizer.deleteCurrentAttributes(id, ioamExport, writeContext);
+
+        verify(jVppIoamexport).ioamExportIp6EnableDisable(generateIoamExportIp6EnableDisable(true));
+    }
+
+    @Test
+    public void testDeleteFailed() throws Exception {
+        final IoamExport ioamExport = generateExportProfile();
+        final InstanceIdentifier<IoamExport> id = InstanceIdentifier.create(IoamExport.class);
+
+        whenExportDelThenFailure();
+
+        try {
+            customizer.deleteCurrentAttributes(id, ioamExport, writeContext);
+        } catch (WriteFailedException e) {
+            verify(jVppIoamexport).ioamExportIp6EnableDisable(generateIoamExportIp6EnableDisable(true));
+
+            return;
+        }
+        fail("WriteFailedException.CreateFailedException was expected");
+    }
+
+    @Test
+    public void testCreateWithMissingDisabledField() throws Exception {
+        IoamExportBuilder builder = new IoamExportBuilder();
+        builder.setSourceAddress(new Ipv4Address("127.0.0.1"));
+        builder.setCollectorAddress(new Ipv4Address("127.0.0.2"));
+        final IoamExport ioamExport = builder.build();
+        final InstanceIdentifier<IoamExport> id = InstanceIdentifier.create(IoamExport.class);
+
+        whenExportAddThenSuccess();
+
+        customizer.writeCurrentAttributes(id, ioamExport, writeContext);
+
+        verify(jVppIoamexport).ioamExportIp6EnableDisable(generateIoamExportIp6EnableDisable(true));
+    }
+
+}