Springboot 2.0 upgrade

[so.git] / mso-catalog-db / src / test / java / org / onap / so / db / catalog / data / repository / CloudSiteRepositoryTest.java

diff --git a/mso-catalog-db/src/test/java/org/onap/so/db/catalog/data/repository/CloudSiteRepositoryTest.java b/mso-catalog-db/src/test/java/org/onap/so/db/catalog/data/repository/CloudSiteRepositoryTest.java
index c5ae3cc..8978f38 100644 (file)
--- a/mso-catalog-db/src/test/java/org/onap/so/db/catalog/data/repository/CloudSiteRepositoryTest.java
+++ b/mso-catalog-db/src/test/java/org/onap/so/db/catalog/data/repository/CloudSiteRepositoryTest.java
@@ -24,6 +24,7 @@ import org.junit.Assert;
 import org.junit.Test;
 import org.onap.so.db.catalog.BaseTest;
 import org.onap.so.db.catalog.beans.CloudSite;
+import org.onap.so.db.catalog.exceptions.NoEntityFoundException;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.util.CollectionUtils;
 
@@ -43,7 +44,9 @@ public class CloudSiteRepositoryTest extends BaseTest {
 
     @Test
     public void findOneTest() throws Exception {
-        CloudSite cloudSite = cloudSiteRepository.findOne("mtn13");
+        CloudSite cloudSite = cloudSiteRepository.findById("mtn13").
+                       orElseThrow(() -> new NoEntityFoundException("Cannot Find Operation"));
+        
         Assert.assertNotNull(cloudSite);
         Assert.assertEquals("mtn13",cloudSite.getId());
     }