Allocate Rule

[holmes/rule-management.git] / rulemgt / src / test / java / org / onap / holmes / rulemgt / bolt / enginebolt / EngineWrapperTest.java

diff --git a/rulemgt/src/test/java/org/onap/holmes/rulemgt/bolt/enginebolt/EngineWrapperTest.java b/rulemgt/src/test/java/org/onap/holmes/rulemgt/bolt/enginebolt/EngineWrapperTest.java
index cc8c047..0664db7 100644 (file)
--- a/rulemgt/src/test/java/org/onap/holmes/rulemgt/bolt/enginebolt/EngineWrapperTest.java
+++ b/rulemgt/src/test/java/org/onap/holmes/rulemgt/bolt/enginebolt/EngineWrapperTest.java
@@ -64,12 +64,13 @@ public class EngineWrapperTest {
         thrown.expectMessage("Failed to call the rule deployment RESTful API.");\r
 \r
         EasyMock.expect(\r
-                engineServiceMock.deploy(EasyMock.anyObject(CorrelationDeployRule4Engine.class)))\r
+                engineServiceMock.deploy(EasyMock.anyObject(CorrelationDeployRule4Engine.class),\r
+                        EasyMock.anyObject(String.class)))\r
                 .andThrow(\r
                         new RuntimeException(""));\r
         PowerMock.replayAll();\r
 \r
-        engineWrapper.deployEngine(new CorrelationDeployRule4Engine());\r
+        engineWrapper.deployEngine(new CorrelationDeployRule4Engine(),"10.96.33.34");\r
 \r
         PowerMock.verifyAll();\r
     }\r
@@ -80,13 +81,14 @@ public class EngineWrapperTest {
         thrown.expectMessage("Failed to deploy the rule!");\r
 \r
         EasyMock.expect(\r
-                engineServiceMock.deploy(EasyMock.anyObject(CorrelationDeployRule4Engine.class)))\r
+                engineServiceMock.deploy(EasyMock.anyObject(CorrelationDeployRule4Engine.class),\r
+                        EasyMock.anyObject(String.class)))\r
                 .andReturn(httpResponse);\r
         EasyMock.expect(httpResponse.getStatusLine()).andReturn(statusLineMock);\r
         EasyMock.expect(statusLineMock.getStatusCode()).andReturn(400);\r
         PowerMock.replayAll();\r
 \r
-        engineWrapper.deployEngine(new CorrelationDeployRule4Engine());\r
+        engineWrapper.deployEngine(new CorrelationDeployRule4Engine(),"10.96.33.34");\r
 \r
         PowerMock.verifyAll();\r
     }\r
@@ -100,14 +102,15 @@ public class EngineWrapperTest {
         thrown.expectMessage(\r
                 "Failed to parse the value returned by the engine management service.");\r
         EasyMock.expect(\r
-                engineServiceMock.deploy(EasyMock.anyObject(CorrelationDeployRule4Engine.class)))\r
+                engineServiceMock.deploy(EasyMock.anyObject(CorrelationDeployRule4Engine.class),\r
+                        EasyMock.anyObject(String.class)))\r
                 .andReturn(httpResponse);\r
         EasyMock.expect(httpResponse.getStatusLine()).andReturn(statusLineMock);\r
         EasyMock.expect(statusLineMock.getStatusCode()).andReturn(200);\r
         PowerMockito.when(HttpsUtils.extractResponseEntity(httpResponse)).thenReturn(content);\r
         PowerMock.replayAll();\r
 \r
-        engineWrapper.deployEngine(new CorrelationDeployRule4Engine());\r
+        engineWrapper.deployEngine(new CorrelationDeployRule4Engine(),"10.96.33.34");\r
 \r
         PowerMock.verifyAll();\r
     }\r
@@ -118,14 +121,15 @@ public class EngineWrapperTest {
         String content = "{\"packageName\":\"test\"}";\r
         PowerMockito.mockStatic(HttpsUtils.class);\r
         EasyMock.expect(\r
-                engineServiceMock.deploy(EasyMock.anyObject(CorrelationDeployRule4Engine.class)))\r
+                engineServiceMock.deploy(EasyMock.anyObject(CorrelationDeployRule4Engine.class),\r
+                        EasyMock.anyObject(String.class)))\r
                 .andReturn(httpResponse);\r
         EasyMock.expect(httpResponse.getStatusLine()).andReturn(statusLineMock);\r
         EasyMock.expect(statusLineMock.getStatusCode()).andReturn(200);\r
         PowerMockito.when(HttpsUtils.extractResponseEntity(httpResponse)).thenReturn(content);\r
         PowerMock.replayAll();\r
 \r
-        String result = engineWrapper.deployEngine(new CorrelationDeployRule4Engine());\r
+        String result = engineWrapper.deployEngine(new CorrelationDeployRule4Engine(),"10.96.33.34");\r
 \r
         assertThat(result, equalTo("test"));\r
 \r
@@ -136,11 +140,12 @@ public class EngineWrapperTest {
         thrown.expect(CorrelationException.class);\r
         thrown.expectMessage("Failed to call the rule deleting RESTful API.");\r
 \r
-        EasyMock.expect(engineServiceMock.delete(EasyMock.anyObject(String.class))).andThrow(\r
+        EasyMock.expect(engineServiceMock.delete(EasyMock.anyObject(String.class),\r
+                EasyMock.anyObject(String.class))).andThrow(\r
                 new RuntimeException(""));\r
         PowerMock.replayAll();\r
 \r
-        engineWrapper.deleteRuleFromEngine("");\r
+        engineWrapper.deleteRuleFromEngine("","10.96.33.34");\r
 \r
         PowerMock.verifyAll();\r
     }\r
@@ -150,28 +155,30 @@ public class EngineWrapperTest {
         thrown.expect(CorrelationException.class);\r
         thrown.expectMessage("Failed to delete the rule!");\r
 \r
-        EasyMock.expect(engineServiceMock.delete(EasyMock.anyObject(String.class)))\r
+        EasyMock.expect(engineServiceMock.delete(EasyMock.anyObject(String.class),\r
+                EasyMock.anyObject(String.class)))\r
                 .andReturn(httpResponse);\r
         EasyMock.expect(httpResponse.getStatusLine()).andReturn(statusLineMock);\r
         EasyMock.expect(statusLineMock.getStatusCode()).andReturn(400);\r
 \r
         PowerMock.replayAll();\r
 \r
-        engineWrapper.deleteRuleFromEngine("");\r
+        engineWrapper.deleteRuleFromEngine("","10.96.33.34");\r
 \r
         PowerMock.verifyAll();\r
     }\r
 \r
     @Test\r
     public void deleteRuleFromEngine_success() throws Exception {\r
-        EasyMock.expect(engineServiceMock.delete(EasyMock.anyObject(String.class)))\r
+        EasyMock.expect(engineServiceMock.delete(EasyMock.anyObject(String.class),\r
+                EasyMock.anyObject(String.class)))\r
                 .andReturn(httpResponse);\r
         EasyMock.expect(httpResponse.getStatusLine()).andReturn(statusLineMock);\r
         EasyMock.expect(statusLineMock.getStatusCode()).andReturn(200);\r
 \r
         PowerMock.replayAll();\r
 \r
-        boolean result = engineWrapper.deleteRuleFromEngine("");\r
+        boolean result = engineWrapper.deleteRuleFromEngine("","10.96.33.34");\r
 \r
         assertThat(result, equalTo(true));\r
     }\r
@@ -182,12 +189,13 @@ public class EngineWrapperTest {
         thrown.expectMessage("Failed to call the rule verification RESTful API.");\r
 \r
         EasyMock.expect(\r
-                engineServiceMock.check(EasyMock.anyObject(CorrelationCheckRule4Engine.class)))\r
+                engineServiceMock.check(EasyMock.anyObject(CorrelationCheckRule4Engine.class),\r
+                        EasyMock.anyObject(String.class)))\r
                 .andThrow(\r
                         new RuntimeException(""));\r
         PowerMock.replayAll();\r
 \r
-        engineWrapper.checkRuleFromEngine(new CorrelationCheckRule4Engine());\r
+        engineWrapper.checkRuleFromEngine(new CorrelationCheckRule4Engine(),"10.96.33.34");\r
 \r
         PowerMock.verifyAll();\r
     }\r
@@ -195,14 +203,15 @@ public class EngineWrapperTest {
     @Test\r
     public void checkRuleFromEngine_success() throws Exception {\r
         EasyMock.expect(\r
-                engineServiceMock.check(EasyMock.anyObject(CorrelationCheckRule4Engine.class)))\r
+                engineServiceMock.check(EasyMock.anyObject(CorrelationCheckRule4Engine.class),\r
+                        EasyMock.anyObject(String.class)))\r
                 .andReturn(httpResponse);\r
         EasyMock.expect(httpResponse.getStatusLine()).andReturn(statusLineMock);\r
         EasyMock.expect(statusLineMock.getStatusCode()).andReturn(200);\r
 \r
         PowerMock.replayAll();\r
 \r
-        boolean result = engineWrapper.checkRuleFromEngine(new CorrelationCheckRule4Engine());\r
+        boolean result = engineWrapper.checkRuleFromEngine(new CorrelationCheckRule4Engine(),"10.96.33.34");\r
 \r
         assertThat(result, equalTo(true));\r
     }\r