diff --git a/src/test/java/ntnu/idatt2016/v233/SmartMat/util/CategoryUtilTest.java b/src/test/java/ntnu/idatt2016/v233/SmartMat/util/CategoryUtilTest.java new file mode 100644 index 0000000000000000000000000000000000000000..8dce0ac5f024a149ca5c84023eea4645807b37f9 --- /dev/null +++ b/src/test/java/ntnu/idatt2016/v233/SmartMat/util/CategoryUtilTest.java @@ -0,0 +1,44 @@ +package ntnu.idatt2016.v233.SmartMat.util; + +import org.junit.jupiter.api.Test; + +import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertThrows; + +public class CategoryUtilTest { + + @Test + void testDefineCategory() { + String name = "kjøtt"; + String description = "Prime cut of beef"; + assertEquals("meat, fish and chicken", CategoryUtil.defineCategory(name, description)); + + name = "brød"; + description = "Whole wheat bread"; + assertEquals("baked goods and grains", CategoryUtil.defineCategory(name, description)); + + name = "melk"; + description = "Whole milk"; + assertEquals("dairy and egg ", CategoryUtil.defineCategory(name, description)); + + name = "eple"; + description = "Fresh green apple"; + assertEquals("fruit and vegetables", CategoryUtil.defineCategory(name, description)); + + name = "Something"; + description = "Unknown product"; + assertEquals("other", CategoryUtil.defineCategory(name, description)); + } + + @Test + void testGetCategoryName() { + assertEquals("meat, fish and chicken", CategoryUtil.getCategoryName(1)); + assertEquals("baked goods and grains", CategoryUtil.getCategoryName(2)); + assertEquals("dairy and egg", CategoryUtil.getCategoryName(3)); + assertEquals("other", CategoryUtil.getCategoryName(4)); + assertEquals("fruit and vegetables", CategoryUtil.getCategoryName(5)); + + int invalidCategoryNumber = 99; + assertThrows(IllegalArgumentException.class, () -> CategoryUtil.getCategoryName(invalidCategoryNumber)); + } +}