Add ImfRegInfo code .

Issue-ID: USECASEUI-696
Change-Id: Ifba61f0950e6c77aafec19cfc107017dae6e9d90
Signed-off-by: hekeguang <hekeguang@chinamobile.com>

2 files changed, 157 insertions, 0 deletions

diff --git a/intentanalysis/src/main/java/org/onap/usecaseui/intentanalysis/util/ListArrayTypeHandler.java b/intentanalysis/src/main/java/org/onap/usecaseui/intentanalysis/util/ListArrayTypeHandler.java
new file mode 100644
index 0000000..9f9b51d
--- /dev/null
+++ b/intentanalysis/src/main/java/org/onap/usecaseui/intentanalysis/util/ListArrayTypeHandler.java
@@ -0,0 +1,67 @@
+/*
+ * Copyright (C) 2022 CMCC, Inc. and others. All rights reserved.
+ *
+ * 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.usecaseui.intentanalysis.util;
+
+import org.apache.ibatis.type.BaseTypeHandler;
+import org.apache.ibatis.type.JdbcType;
+import org.apache.ibatis.type.MappedJdbcTypes;
+import org.apache.ibatis.type.MappedTypes;
+
+import java.sql.*;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+@MappedJdbcTypes({ JdbcType.ARRAY })
+@MappedTypes({ Object.class })
+public class ListArrayTypeHandler extends BaseTypeHandler<List<?>> {
+
+    @Override
+    public void setNonNullParameter(PreparedStatement ps, int i,
+                                    List<?> parameter, JdbcType jdbcType) throws SQLException {
+        //  JDBC type is required
+        Array array = ps.getConnection().createArrayOf("TEXT", parameter.toArray());
+        try {
+            ps.setArray(i, array);
+        } finally {
+            array.free();
+        }
+    }
+
+    @Override
+    public List<?> getNullableResult(ResultSet rs, String columnName) throws SQLException {
+        return extractArray(rs.getArray(columnName));
+    }
+
+    @Override
+    public List<?> getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
+        return extractArray(rs.getArray(columnIndex));
+    }
+
+    @Override
+    public List<?> getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
+        return extractArray(cs.getArray(columnIndex));
+    }
+
+    protected List<?> extractArray(Array array) throws SQLException {
+        if (array == null) {
+            return null;
+        }
+        Object javaArray = array.getArray();
+        array.free();
+        return new ArrayList<>(Arrays.asList((Object[])javaArray));
+    }
+}
diff --git a/intentanalysis/src/test/java/org/onap/usecaseui/intentanalysis/service/impl/ImfRegInfoServiceImplTest.java b/intentanalysis/src/test/java/org/onap/usecaseui/intentanalysis/service/impl/ImfRegInfoServiceImplTest.java
new file mode 100644
index 0000000..3606eea
--- /dev/null
+++ b/intentanalysis/src/test/java/org/onap/usecaseui/intentanalysis/service/impl/ImfRegInfoServiceImplTest.java
@@ -0,0 +1,90 @@
+/*
+ * Copyright (C) 2022 CMCC, Inc. and others. All rights reserved.
+ *
+ * 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.usecaseui.intentanalysis.service.impl;
+
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.onap.usecaseui.intentanalysis.bean.enums.IntentGoalType;
+import org.onap.usecaseui.intentanalysis.bean.enums.SupportArea;
+import org.onap.usecaseui.intentanalysis.bean.enums.SupportInterface;
+import org.onap.usecaseui.intentanalysis.bean.models.Intent;
+import org.onap.usecaseui.intentanalysis.bean.models.IntentGoalBean;
+import org.onap.usecaseui.intentanalysis.bean.models.IntentManagementFunctionRegInfo;
+import org.onap.usecaseui.intentanalysis.mapper.IMFRegInfoMapper;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import static org.mockito.ArgumentMatchers.any;
+
+@SpringBootTest(classes = ImfRegInfoServiceImplTest.class)
+@RunWith(SpringJUnit4ClassRunner.class)
+public class ImfRegInfoServiceImplTest {
+
+    @InjectMocks
+    private ImfRegInfoServiceImpl imfRegInfoServiceImpl;
+
+    @Mock
+    private IMFRegInfoMapper imfRegInfoMapper;
+    Intent intent = new Intent();
+    IntentGoalBean intentGoalBean = new IntentGoalBean();
+
+    @Before
+    public void setUp() {
+        intent.setIntentId("intentid");
+        intent.setIntentName("CLLBusiness");
+        intentGoalBean.setIntent(intent);
+        intentGoalBean.setIntentGoalType(IntentGoalType.CREATE);
+    }
+
+    @Test
+    public void itCanInsertIMFRegInfoRegInfo() {
+        Mockito.when(imfRegInfoMapper.insertIMFRegInfoRegInfo(any())).thenReturn(1);
+        imfRegInfoServiceImpl.insertIMFRegInfoRegInfo(new IntentManagementFunctionRegInfo());
+        Assert.assertTrue(true);
+    }
+
+    @Test
+    public void itCanGetImfRegInfoList() {
+        imfRegInfoServiceImpl.getImfRegInfoList();
+    }
+
+    @Test
+    public void itCanGetImfRegInfoListWithIntentGoalBean() {
+        List<IntentManagementFunctionRegInfo> list = new ArrayList<>();
+        IntentManagementFunctionRegInfo info = new IntentManagementFunctionRegInfo();
+        List<SupportArea> areas = new ArrayList<>();
+        areas.add(SupportArea.CLLBUSINESS);
+        info.setSupportArea(areas);
+
+        List<SupportInterface> interList = new ArrayList<>();
+        interList.add(SupportInterface.CREATE);
+        info.setSupportInterfaces(interList);
+
+        list.add(info);
+
+        Mockito.when(imfRegInfoMapper.getImfRegInfoList()).thenReturn(list);
+        imfRegInfoServiceImpl.getImfRegInfo(intentGoalBean);
+        Assert.assertTrue(true);
+    }
+}
\ No newline at end of file