Add missing agr and iterate on row

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

2 files changed, 10 insertions, 2 deletions

diff --git a/netbox-client/provider/src/main/java/org/onap/ccsdk/sli/adaptors/netbox/impl/NetboxClientImpl.java b/netbox-client/provider/src/main/java/org/onap/ccsdk/sli/adaptors/netbox/impl/NetboxClientImpl.java
index e7b5284a..b54a35b5 100644
--- a/netbox-client/provider/src/main/java/org/onap/ccsdk/sli/adaptors/netbox/impl/NetboxClientImpl.java
+++ b/netbox-client/provider/src/main/java/org/onap/ccsdk/sli/adaptors/netbox/impl/NetboxClientImpl.java
@@ -64,7 +64,7 @@ public class NetboxClientImpl implements NetboxClient {
 
     private static final String ASSIGN_IP_SQL_STATEMENT =
         "INSERT INTO IPAM_IP_ASSIGNEMENT (service_instance_id, vf_module_id, prefix_id, ip_address_id, ip_address, ip_status, ip_response_json, external_key) \n"
-            + "VALUES (?, ?, ?, ?, ?, ?, ?)";
+            + "VALUES (?, ?, ?, ?, ?, ?, ?, ?)";
     private static final String UNASSIGN_IP_SQL_STATEMENT =
         "UPDATE IPAM_IP_ASSIGNEMENT SET ip_status = ? WHERE service_instance_id = ? AND external_key = ?";
     private static final String GET_IP_ADDRESS_ID_SQL_STATEMENT =
@@ -169,7 +169,11 @@ public class NetboxClientImpl implements NetboxClient {
             serviceInstanceId,
             externalKey);
         try (CachedRowSet row = dbLibService.getData(GET_IP_ADDRESS_ID_SQL_STATEMENT, args, null)) {
-            ipAddressId = row.getString("ip_address_id");
+            if (row.next()) {
+                ipAddressId = row.getString("ip_address_id");
+            } else {
+                throw new SQLException("Data not found.");
+            }
         } catch (SQLException e) {
             LOG.error(SQL_EXCEPTION_MESSAGE, e);
             return QueryStatus.FAILURE;
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
index bc81f0b7..624a1906 100644
--- 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
@@ -147,6 +147,7 @@ public class NetboxClientImplTest {
 
         CachedRowSet crs = mock(CachedRowSet.class);
         doReturn("3").when(crs).getString(eq("ip_address_id"));
+        doReturn(true).when(crs).next();
         doReturn(crs).when(dbLib).getData(anyString(), any(ArrayList.class), eq(null));
 
         QueryStatus status = netboxClientMock
@@ -164,6 +165,7 @@ public class NetboxClientImplTest {
 
         CachedRowSet crs = mock(CachedRowSet.class);
         doReturn("3").when(crs).getString(eq("ip_address_id"));
+        doReturn(true).when(crs).next();
         doReturn(crs).when(dbLib).getData(anyString(), any(ArrayList.class), eq(null));
 
         QueryStatus status = netboxClient.unassignIpAddress(params, svcLogicContext);
@@ -181,6 +183,7 @@ public class NetboxClientImplTest {
         givenThat(delete(urlEqualTo(expectedUrl)).willReturn(created().withBody(response)));
 
         CachedRowSet crs = mock(CachedRowSet.class);
+        doReturn(true).when(crs).next();
         doReturn("3").when(crs).getString(eq("ip_address_id"));
         doReturn(crs).when(dbLib).getData(anyString(), any(ArrayList.class), eq(null));
 
@@ -201,6 +204,7 @@ public class NetboxClientImplTest {
 
         CachedRowSet crs = mock(CachedRowSet.class);
         doReturn("3").when(crs).getString(eq("ip_address_id"));
+        doReturn(true).when(crs).next();
         doReturn(crs).when(dbLib).getData(anyString(), any(ArrayList.class), eq(null));
 
         QueryStatus status = netboxClient.unassignIpAddress(params, svcLogicContext);