Honeynode test tool
[transportpce.git] / tests / honeynode / honeynode-plugin-impl / src / main / java / io / fd / honeycomb / transportpce / device / tools / DefaultDeviceFactory.java
diff --git a/tests/honeynode/honeynode-plugin-impl/src/main/java/io/fd/honeycomb/transportpce/device/tools/DefaultDeviceFactory.java b/tests/honeynode/honeynode-plugin-impl/src/main/java/io/fd/honeycomb/transportpce/device/tools/DefaultDeviceFactory.java
new file mode 100644 (file)
index 0000000..1372de5
--- /dev/null
@@ -0,0 +1,139 @@
+/*
+ * Copyright (c) 2018 Orange 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.honeycomb.transportpce.device.tools;
+
+import io.fd.honeycomb.transportpce.binding.converter.XMLDataObjectConverter;
+import io.fd.honeycomb.transportpce.test.common.DataStoreContext;
+
+import java.io.BufferedWriter;
+import java.io.ByteArrayInputStream;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.Writer;
+import java.util.Optional;
+
+import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev170206.org.openroadm.device.container.OrgOpenroadmDevice;
+import org.opendaylight.yangtools.yang.binding.DataObject;
+import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * Factory for creating the default device from the XML stored in
+ * classpath.
+ *
+ * @author Martial COULIBALY ( martial.coulibaly@gfi.com ) on behalf of Orange
+ */
+public class DefaultDeviceFactory {
+
+    private static final Logger LOG = LoggerFactory.getLogger(DefaultDeviceFactory.class);
+
+    /**
+     * Returns a new instance of {@link OrgOpenroadmDevice} from the loaded XML stored
+     * in File.
+     *
+     * @return {@link OrgOpenroadmDevice}
+     */
+    public OrgOpenroadmDevice createDefaultDevice(DataStoreContext dataStoreContextUtil, File device_data_config) {
+        OrgOpenroadmDevice result = null;
+        if (device_data_config.exists()) {
+            String config = device_data_config.getName();
+            LOG.info("file '{}' exists at location : {}", config, device_data_config.getAbsolutePath());
+            InputStream targetStream;
+            try {
+                targetStream = new FileInputStream(device_data_config);
+                Optional<NormalizedNode<?, ?>> transformIntoNormalizedNode = null;
+                transformIntoNormalizedNode =
+                        XMLDataObjectConverter.createWithDataStoreUtil(dataStoreContextUtil)
+                        .transformIntoNormalizedNode(targetStream);
+                if (!transformIntoNormalizedNode.isPresent()) {
+                    throw new IllegalStateException(String.format("Could not transform the input %s into normalized nodes",
+                            config));
+                }
+                Optional<DataObject> dataObject = XMLDataObjectConverter.createWithDataStoreUtil(dataStoreContextUtil)
+                        .getDataObject(transformIntoNormalizedNode.get(), OrgOpenroadmDevice.QNAME);
+                if (!dataObject.isPresent()) {
+                    throw new IllegalStateException("Could not transform normalized nodes into data object");
+                }
+                result =  (OrgOpenroadmDevice) dataObject.get();
+            } catch (FileNotFoundException e) {
+                LOG.error("File not found : {} at {}", e.getMessage(), e.getLocalizedMessage());
+            }
+        } else {
+            LOG.info("xml file not existed at : '{}'", device_data_config.getAbsolutePath());
+        }
+        return result;
+    }
+    /**
+     * Returns a new instance of {@link OrgOpenroadmDevice} from the loaded XML stored
+     * in String.
+     *
+     * @return {@link OrgOpenroadmDevice}
+     */
+    public OrgOpenroadmDevice createDefaultDevice(DataStoreContext dataStoreContextUtil, String device_data_config) {
+        OrgOpenroadmDevice result = null;
+        if (device_data_config != null) {
+            LOG.info("device data config string is ok ");
+            InputStream targetStream;
+            targetStream = new ByteArrayInputStream(device_data_config.getBytes());
+            Optional<NormalizedNode<?, ?>> transformIntoNormalizedNode = null;
+            transformIntoNormalizedNode =
+                    XMLDataObjectConverter.createWithDataStoreUtil(dataStoreContextUtil)
+                    .transformIntoNormalizedNode(targetStream);
+            if (!transformIntoNormalizedNode.isPresent()) {
+                throw new IllegalStateException(String.format("Could not transform the input %s into normalized nodes"));
+            }
+            Optional<DataObject> dataObject = XMLDataObjectConverter.createWithDataStoreUtil(dataStoreContextUtil)
+                    .getDataObject(transformIntoNormalizedNode.get(), OrgOpenroadmDevice.QNAME);
+            if (!dataObject.isPresent()) {
+                throw new IllegalStateException("Could not transform normalized nodes into data object");
+            }
+            result =  (OrgOpenroadmDevice) dataObject.get();
+        } else {
+            LOG.info("device data config string is null!");
+        }
+        return result;
+    }
+
+
+    public void createXMLFromDevice(DataStoreContext dataStoreContextUtil, OrgOpenroadmDevice device, String output) {
+        if (device != null) {
+            Optional<NormalizedNode<?, ?>> transformIntoNormalizedNode = null;
+            transformIntoNormalizedNode =
+                    XMLDataObjectConverter.createWithDataStoreUtil(dataStoreContextUtil)
+                    .toNormalizedNodes(device, OrgOpenroadmDevice.class);
+            if (!transformIntoNormalizedNode.isPresent()) {
+                throw new IllegalStateException(String.format("Could not transform the input %s into normalized nodes",
+                        device));
+            }
+            XMLDataObjectConverter createWithDataStoreUtil = XMLDataObjectConverter.createWithDataStoreUtil(dataStoreContextUtil);
+            Writer writerFromDataObject =
+                    createWithDataStoreUtil.writerFromDataObject(device, OrgOpenroadmDevice.class, createWithDataStoreUtil.dataContainer());
+            try {
+                BufferedWriter writer = new BufferedWriter(new FileWriter(output));
+                writer.write(writerFromDataObject.toString());
+                writer.close();
+            } catch (IOException e) {
+                LOG.error("Bufferwriter error ");
+            }
+            LOG.info("device xml : {}", writerFromDataObject.toString());
+        }
+    }
+}