diff --git a/api/src/test/java/org/openmrs/api/db/hibernate/HibernateFormDAOTest.java b/api/src/test/java/org/openmrs/api/db/hibernate/HibernateFormDAOTest.java index 8460c59c2a3b..dc1c465752b4 100644 --- a/api/src/test/java/org/openmrs/api/db/hibernate/HibernateFormDAOTest.java +++ b/api/src/test/java/org/openmrs/api/db/hibernate/HibernateFormDAOTest.java @@ -12,12 +12,11 @@ import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertNotNull; -import java.util.Arrays; -import java.util.Collections; -import java.util.List; +import java.util.*; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; +import org.openmrs.EncounterType; import org.openmrs.Field; import org.openmrs.Form; import org.openmrs.FormField; @@ -59,4 +58,34 @@ public void shouldGetFormFieldsByForm() { assertEquals(form.getFormId(), formField.getForm().getFormId()); } } + + @Test + public void shouldGetFormFieldsByField() { + Field field = new Field(1); + FormField formField = new FormField(2); + formField.setField(field); + List formFields = dao.getFormFieldsByField(field); + assertNotNull(formFields); + assertEquals(3, formFields.size()); + } + + @Test + public void shouldGetFormCount() { + String partialName = "Basic Form"; + Boolean published = false; + Collection encounterTypes = new ArrayList<>(); + Boolean retired = false; + Collection containingAnyFormField = new ArrayList<>(); + Collection containingAllFormFields = new ArrayList<>(); + Collection fields = new ArrayList<>(); + Integer formCount = dao.getFormCount(partialName, + published, + encounterTypes, + retired, + containingAnyFormField, + containingAllFormFields, + fields); + assertNotNull(formCount); + assertEquals(2, formCount); + } }