Added netbox client to assign/unassign ip

Change-Id: Ied317c7f251936ced116b6a3ea81789c82095df6
Issue-ID: CCSDK-462
Signed-off-by: Alexis de Talhouët <adetalhouet89@gmail.com>

4 files changed, 392 insertions, 0 deletions

diff --git a/netbox-client/provider/src/test/java/org/onap/ccsdk/sli/adaptors/netbox/impl/NetboxClientImplTest.java b/netbox-client/provider/src/test/java/org/onap/ccsdk/sli/adaptors/netbox/impl/NetboxClientImplTest.java
new file mode 100644
index 00000000..19b178c9
--- /dev/null
+++ b/netbox-client/provider/src/test/java/org/onap/ccsdk/sli/adaptors/netbox/impl/NetboxClientImplTest.java
@@ -0,0 +1,185 @@
+/*
+ * Copyright (C) 2018 Bell Canada.
+ *
+ * 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 org.onap.ccsdk.sli.adaptors.netbox.impl;
+
+import static com.github.tomakehurst.wiremock.client.WireMock.created;
+import static com.github.tomakehurst.wiremock.client.WireMock.delete;
+import static com.github.tomakehurst.wiremock.client.WireMock.deleteRequestedFor;
+import static com.github.tomakehurst.wiremock.client.WireMock.equalTo;
+import static com.github.tomakehurst.wiremock.client.WireMock.givenThat;
+import static com.github.tomakehurst.wiremock.client.WireMock.ok;
+import static com.github.tomakehurst.wiremock.client.WireMock.post;
+import static com.github.tomakehurst.wiremock.client.WireMock.postRequestedFor;
+import static com.github.tomakehurst.wiremock.client.WireMock.serverError;
+import static com.github.tomakehurst.wiremock.client.WireMock.urlEqualTo;
+import static com.github.tomakehurst.wiremock.client.WireMock.verify;
+import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.wireMockConfig;
+import static org.apache.http.HttpHeaders.ACCEPT;
+import static org.apache.http.HttpHeaders.AUTHORIZATION;
+import static org.apache.http.HttpHeaders.CONTENT_TYPE;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.mock;
+
+import com.github.tomakehurst.wiremock.junit.WireMockRule;
+import com.google.common.base.Charsets;
+import com.google.common.io.Resources;
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URL;
+import java.nio.charset.StandardCharsets;
+import org.apache.http.HttpEntity;
+import org.apache.http.HttpResponse;
+import org.apache.http.StatusLine;
+import org.junit.After;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.runners.MockitoJUnitRunner;
+import org.onap.ccsdk.sli.adaptors.netbox.api.IpamException;
+import org.onap.ccsdk.sli.adaptors.netbox.model.IPAddress;
+import org.onap.ccsdk.sli.adaptors.netbox.model.Prefix;
+import org.onap.ccsdk.sli.adaptors.netbox.model.Status.Values;
+
+@RunWith(MockitoJUnitRunner.class)
+public class NetboxClientImplTest {
+
+    private static final String APPLICATION_JSON = "application/json";
+
+    @Rule
+    public WireMockRule wm = new WireMockRule(wireMockConfig().dynamicPort());
+
+    private String token = "token";
+
+    private NetboxHttpClient httpClient;
+    private NetboxClientImpl netboxClient;
+
+    @Before
+    public void setup() {
+        String baseUrl = "http://localhost:" + wm.port();
+
+        httpClient = new NetboxHttpClient(baseUrl, token);
+        httpClient.init();
+
+        netboxClient = new NetboxClientImpl(httpClient);
+
+        wm.addMockServiceRequestListener(
+            (request, response) -> {
+                System.out.println("Request URL :" + request.getAbsoluteUrl());
+                System.out.println("Request body :" + request.getBodyAsString());
+                System.out.println("Response status :" + response.getStatus());
+                System.out.println("Response body :" + response.getBodyAsString());
+            });
+    }
+
+    @After
+    public void tearDown() throws IOException {
+        httpClient.close();
+    }
+
+    @Test
+    public void nextAvailableIpInPrefixTestNoId() {
+        Prefix prefix = mock(Prefix.class);
+        doReturn(null).when(prefix).getId();
+        try {
+            netboxClient.assign(prefix);
+        } catch (IpamException e) {
+            Assert.assertEquals("Id must be set", e.getMessage());
+            return;
+        }
+        Assert.fail();
+    }
+
+    @Test
+    public void nextAvailableIpInPrefixTest() throws IOException, IpamException {
+        Integer id = 3;
+        Prefix prefix = mock(Prefix.class);
+        doReturn(id).when(prefix).getId();
+
+        URL url = Resources.getResource("nextAvailableIpResponse.json");
+        String response = Resources.toString(url, Charsets.UTF_8);
+
+        String expectedUrl = "/api/ipam/prefixes/" + id + "/available-ips/";
+        givenThat(post(urlEqualTo(expectedUrl)).willReturn(created().withBody(response)));
+
+        netboxClient.assign(prefix);
+
+        verify(postRequestedFor(urlEqualTo(expectedUrl))
+            .withHeader(ACCEPT, equalTo(APPLICATION_JSON))
+            .withHeader(CONTENT_TYPE, equalTo(APPLICATION_JSON))
+            .withHeader(AUTHORIZATION, equalTo("Token " + token)));
+    }
+
+    @Test
+    public void deleteIpTestError500() {
+        Integer id = 3;
+        IPAddress ipAddress = mock(IPAddress.class);
+        doReturn(id).when(ipAddress).getId();
+
+        String expectedUrl = "/api/ipam/ip-addresses/" + id + "/";
+        givenThat(delete(urlEqualTo(expectedUrl)).willReturn(serverError()));
+        try {
+            netboxClient.unassign(ipAddress);
+        } catch (IpamException e) {
+            Assert.assertEquals(IllegalStateException.class, e.getCause().getClass());
+            Assert.assertTrue(e.getMessage().contains(
+                "Fail to unassign IP for IPAddress(id= 3). java.lang.IllegalStateException: Netbox request failed with status: HTTP/1.1 500 Server Error"));
+            return;
+        }
+        Assert.fail();
+    }
+
+    @Test
+    public void deleteIpTest() throws IpamException {
+        Integer id = 3;
+        IPAddress ipAddress = mock(IPAddress.class);
+        doReturn(id).when(ipAddress).getId();
+
+        String expectedUrl = "/api/ipam/ip-addresses/" + id + "/";
+        givenThat(delete(urlEqualTo(expectedUrl)).willReturn(ok()));
+        netboxClient.unassign(ipAddress);
+        verify(deleteRequestedFor(urlEqualTo(expectedUrl))
+            .withHeader(ACCEPT, equalTo(APPLICATION_JSON))
+            .withHeader(CONTENT_TYPE, equalTo(APPLICATION_JSON))
+            .withHeader(AUTHORIZATION, equalTo("Token " + token)));
+    }
+
+
+    @Test
+    public void getIpAddressTest() throws IOException {
+        StatusLine statusLine = mock(StatusLine.class);
+        doReturn(201).when(statusLine).getStatusCode();
+
+        URL url = Resources.getResource("nextAvailableIpResponse.json");
+        String response = Resources.toString(url, Charsets.UTF_8);
+        InputStream stream = new ByteArrayInputStream(response.getBytes(StandardCharsets.UTF_8));
+
+        HttpEntity entity = mock(HttpEntity.class);
+        doReturn(stream).when(entity).getContent();
+
+        HttpResponse httpResponse = mock(HttpResponse.class);
+        doReturn(statusLine).when(httpResponse).getStatusLine();
+        doReturn(entity).when(httpResponse).getEntity();
+
+        IPAddress ipAddress = netboxClient.getIpAddress(httpResponse);
+
+        Assert.assertEquals("192.168.20.7/32", ipAddress.getAddress());
+        Assert.assertEquals(Integer.valueOf(8), ipAddress.getId());
+        Assert.assertEquals(Values.ACTIVE, ipAddress.getStatus());
+    }
+}
\ No newline at end of file
diff --git a/netbox-client/provider/src/test/java/org/onap/ccsdk/sli/adaptors/netbox/impl/NetboxHttpClientTest.java b/netbox-client/provider/src/test/java/org/onap/ccsdk/sli/adaptors/netbox/impl/NetboxHttpClientTest.java
new file mode 100644
index 00000000..ee2861c0
--- /dev/null
+++ b/netbox-client/provider/src/test/java/org/onap/ccsdk/sli/adaptors/netbox/impl/NetboxHttpClientTest.java
@@ -0,0 +1,127 @@
+/*
+ * Copyright (C) 2018 Bell Canada.
+ *
+ * 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 org.onap.ccsdk.sli.adaptors.netbox.impl;
+
+import static com.github.tomakehurst.wiremock.client.WireMock.aResponse;
+import static com.github.tomakehurst.wiremock.client.WireMock.delete;
+import static com.github.tomakehurst.wiremock.client.WireMock.deleteRequestedFor;
+import static com.github.tomakehurst.wiremock.client.WireMock.equalTo;
+import static com.github.tomakehurst.wiremock.client.WireMock.givenThat;
+import static com.github.tomakehurst.wiremock.client.WireMock.ok;
+import static com.github.tomakehurst.wiremock.client.WireMock.post;
+import static com.github.tomakehurst.wiremock.client.WireMock.postRequestedFor;
+import static com.github.tomakehurst.wiremock.client.WireMock.urlEqualTo;
+import static com.github.tomakehurst.wiremock.client.WireMock.verify;
+import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.wireMockConfig;
+import static org.apache.http.HttpHeaders.ACCEPT;
+import static org.apache.http.HttpHeaders.CONTENT_TYPE;
+
+import com.github.tomakehurst.wiremock.http.Fault;
+import com.github.tomakehurst.wiremock.junit.WireMockRule;
+import java.io.IOException;
+import java.util.concurrent.CompletionException;
+import org.junit.After;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.onap.ccsdk.sli.adaptors.netbox.api.IpamException;
+
+public class NetboxHttpClientTest {
+
+    private static final String APPLICATION_JSON = "application/json";
+
+    @Rule
+    public WireMockRule wm = new WireMockRule(wireMockConfig().dynamicPort());
+
+    private NetboxHttpClient httpClient;
+
+    @Before
+    public void setup() {
+        String baseUrl = "http://localhost:" + wm.port();
+        String token = "token";
+
+        httpClient = new NetboxHttpClient(baseUrl, token);
+        httpClient.init();
+
+        wm.addMockServiceRequestListener(
+            (request, response) -> {
+                System.out.println("Request URL :" + request.getAbsoluteUrl());
+                System.out.println("Request body :" + request.getBodyAsString());
+                System.out.println("Response status :" + response.getStatus());
+                System.out.println("Response body :" + response.getBodyAsString());
+            });
+    }
+
+    @After
+    public void tearDown() throws IOException {
+        httpClient.close();
+    }
+
+    @Test
+    public void postTest() {
+        String expectedUrl = "/testPost";
+        givenThat(post(urlEqualTo(expectedUrl)).willReturn(ok()));
+
+        httpClient.post(expectedUrl, "").toCompletableFuture().join();
+
+        verify(postRequestedFor(urlEqualTo(expectedUrl))
+            .withHeader(ACCEPT, equalTo(APPLICATION_JSON))
+            .withHeader(CONTENT_TYPE, equalTo(APPLICATION_JSON)));
+    }
+
+    @Test
+    public void postTestException() {
+        String expectedUrl = "/testPost";
+        givenThat(post(urlEqualTo(expectedUrl)).willReturn(aResponse().withFault(Fault.MALFORMED_RESPONSE_CHUNK)));
+
+        try {
+            httpClient.post(expectedUrl, "").toCompletableFuture().join();
+        } catch (CompletionException e) {
+            Assert.assertEquals(IpamException.class, e.getCause().getClass());
+            Assert.assertEquals("Netbox request failed", e.getCause().getMessage());
+            return;
+        }
+        Assert.fail();
+    }
+
+    @Test
+    public void deleteTest() {
+        String expectedUrl = "/testDelete";
+        givenThat(delete(urlEqualTo(expectedUrl)).willReturn(ok()));
+
+        httpClient.delete(expectedUrl).toCompletableFuture().join();
+
+        verify(deleteRequestedFor(urlEqualTo(expectedUrl))
+            .withHeader(ACCEPT, equalTo(APPLICATION_JSON))
+            .withHeader(CONTENT_TYPE, equalTo(APPLICATION_JSON)));
+    }
+
+    @Test
+    public void deleteTestException() {
+        String expectedUrl = "/testDelete";
+        givenThat(delete(urlEqualTo(expectedUrl)).willReturn(aResponse().withFault(Fault.MALFORMED_RESPONSE_CHUNK)));
+
+        try {
+            httpClient.delete(expectedUrl).toCompletableFuture().join();
+        } catch (CompletionException e) {
+            Assert.assertEquals(IpamException.class, e.getCause().getClass());
+            Assert.assertEquals("Netbox request failed", e.getCause().getMessage());
+            return;
+        }
+        Assert.fail();
+    }
+}
diff --git a/netbox-client/provider/src/test/java/org/onap/ccsdk/sli/adaptors/netbox/property/NetboxPropertiesTest.java b/netbox-client/provider/src/test/java/org/onap/ccsdk/sli/adaptors/netbox/property/NetboxPropertiesTest.java
new file mode 100644
index 00000000..d9916707
--- /dev/null
+++ b/netbox-client/provider/src/test/java/org/onap/ccsdk/sli/adaptors/netbox/property/NetboxPropertiesTest.java
@@ -0,0 +1,67 @@
+/*
+ * Copyright (C) 2018 Bell Canada.
+ *
+ * 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 org.onap.ccsdk.sli.adaptors.netbox.property;
+
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+
+import ch.qos.logback.classic.spi.ILoggingEvent;
+import ch.qos.logback.core.Appender;
+import java.util.List;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.ArgumentCaptor;
+import org.mockito.Captor;
+import org.mockito.Mock;
+import org.mockito.runners.MockitoJUnitRunner;
+import org.slf4j.LoggerFactory;
+
+@RunWith(MockitoJUnitRunner.class)
+public class NetboxPropertiesTest {
+
+    private NetboxProperties props;
+
+    @Mock
+    private Appender<ILoggingEvent> appender;
+    @Captor
+    private ArgumentCaptor<ILoggingEvent> captor;
+
+    @Before
+    public void setup() {
+        ch.qos.logback.classic.Logger logger = (ch.qos.logback.classic.Logger) LoggerFactory
+            .getLogger(org.slf4j.Logger.ROOT_LOGGER_NAME);
+        logger.addAppender(appender);
+    }
+
+    @Test
+    public void testMissingFile() {
+        props = new NetboxProperties();
+
+        verifyLogEntry(
+            "Missing configuration properties resource for Netbox: netbox.properties");
+    }
+
+
+    private void verifyLogEntry(String message) {
+        verify(appender, times(1)).doAppend(captor.capture());
+        List<ILoggingEvent> allValues = captor.getAllValues();
+        for (ILoggingEvent loggingEvent : allValues) {
+            Assert.assertTrue(loggingEvent.getFormattedMessage().contains(message));
+        }
+    }
+}
\ No newline at end of file
diff --git a/netbox-client/provider/src/test/resources/nextAvailableIpResponse.json b/netbox-client/provider/src/test/resources/nextAvailableIpResponse.json
new file mode 100644
index 00000000..dec1245e
--- /dev/null
+++ b/netbox-client/provider/src/test/resources/nextAvailableIpResponse.json
@@ -0,0 +1,13 @@
+{
+  "id": 8,
+  "address": "192.168.20.7/32",
+  "vrf": null,
+  "tenant": 1,
+  "status": 1,
+  "role": null,
+  "interface": null,
+  "description": "",
+  "nat_inside": null,
+  "created": "2018-08-15",
+  "last_updated": "2018-08-15T15:51:26.634903Z"
+}
\ No newline at end of file