source: MotivateDuringTherapy/src/test/java/tudelft/mentalhealth/motivatepersisting/CategoryTest.java

package tudelft.mentalhealth.motivatepersisting;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;

import org.junit.Test;

import tudelft.utilities.junit.GeneralTests;

/**
 * This also covers the {@link Subcategory} as we test them in connection
 *
 */
public class CategoryTest extends GeneralTests<Category> {

        @Override
        public List<List<Category>> getGeneralTestData() {
                List<List<Category>> list = new LinkedList<>();
                list.add(Arrays.asList(Category.MOTIVATION, Category.MOTIVATION));
                list.add(Arrays.asList(Category.EMPATHY));
                list.add(Arrays.asList(Category.FUTURE));
                return list;
        }

        @Override
        public List<String> getGeneralTestStrings() {
                return Arrays.asList("MOTIVATION", "EMPATHY", "FUTURE");
        }

        @Test
        public void isSubAllowedTest() {
                assertFalse(Category.MOTIVATION.isSubcategoryAllowed(Subcategory.ALL));
                assertTrue(Category.MOTIVATION
                                .isSubcategoryAllowed(Subcategory.COMPLIMENT));
                assertTrue(
                                Category.MOTIVATION.isSubcategoryAllowed(Subcategory.HOLDON));
                assertFalse(Category.MOTIVATION.isSubcategoryAllowed(Subcategory.LOW));
        }

        @Test
        public void subcatTest() {
                assertEquals(4, Category.MOTIVATION.getSubCategories().size());
                assertEquals(1, Category.EMPATHY.getSubCategories().size());
                assertEquals(1, Category.FUTURE.getSubCategories().size());
        }

}