1 files changed, 38 insertions, 26 deletions
diff --git a/ONAP-PAP-REST/src/test/java/org/onap/policy/pap/xacml/rest/components/BRMSPolicyTest.java b/ONAP-PAP-REST/src/test/java/org/onap/policy/pap/xacml/rest/components/BRMSPolicyTest.java
index c13d1f1fe..a31d5a393 100644
--- a/ONAP-PAP-REST/src/test/java/org/onap/policy/pap/xacml/rest/components/BRMSPolicyTest.java
+++ b/ONAP-PAP-REST/src/test/java/org/onap/policy/pap/xacml/rest/components/BRMSPolicyTest.java
@@ -19,36 +19,48 @@
  */
 package org.onap.policy.pap.xacml.rest.components;
 
-import org.junit.Rule;
-import org.junit.Test;
-import org.junit.rules.ExpectedException;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
 import java.io.IOException;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+import org.mockito.Mockito;
 import org.onap.policy.rest.dao.CommonClassDao;
 
 public class BRMSPolicyTest {
-	@Rule
-    public ExpectedException thrown = ExpectedException.none();
+  @Rule
+  public ExpectedException thrown = ExpectedException.none();
+
+  @Test
+  public void testConstructor1() {
+    CreateBRMSRuleTemplate template = new CreateBRMSRuleTemplate();
+    assertNotNull(template);
+  }
+
+  @Test
+  public void testConstructor2() {
+    CommonClassDao commonClassDao = null;
+    CreateBRMSRuleTemplate template = new CreateBRMSRuleTemplate(commonClassDao);
+    assertNotNull(template);
+  }
+
+  @Test
+  public void testReadFile() throws IOException {
+    String goodRule = "declare Params\nparam1 : int\nend\n";
+    String badRule = "declare Params\nparam1+ : int\nend\n";
+    assertEquals(CreateBRMSRuleTemplate.validateRuleParams(goodRule), true);
+    assertEquals(CreateBRMSRuleTemplate.validateRuleParams(badRule), false);
+  }
 
-	@Test
-	public void testConstructor1() {
-		CreateBRMSRuleTemplate template = new CreateBRMSRuleTemplate();
-		assertNotNull(template);
-	}
-	
-	@Test
-	public void testConstructor2() {
-		CommonClassDao commonClassDao = null;
-		CreateBRMSRuleTemplate template = new CreateBRMSRuleTemplate(commonClassDao);
-		assertNotNull(template);
-	}
-	
-	@Test
-	public void testReadFile() throws IOException {
-		String goodRule = "declare Params\nparam1 : int\nend\n";
-		String badRule = "declare Params\nparam1+ : int\nend\n";
-		assertEquals(CreateBRMSRuleTemplate.validateRuleParams(goodRule), true);
-		assertEquals(CreateBRMSRuleTemplate.validateRuleParams(badRule), false);
-	}
-}
-\ No newline at end of file
+  @Test
+  public void testAdd() {
+    CommonClassDao dao = Mockito.mock(CommonClassDao.class);
+    CreateBRMSRuleTemplate template = new CreateBRMSRuleTemplate(dao);
+    String rule = "package foo\n";
+    String ruleName = "testName";
+    String description = "testDesc";
+    String userID = "testID";
+    assertEquals(1, template.addRule(rule, ruleName, description, userID).size());
+  }
+}