NoticeTreeItemTest.java

package com.noticeditorteam.noticeditor.model;

import static org.junit.Assert.*;
import org.junit.Test;

public class NoticeTreeItemTest {

    @Test
    public void testBranch() {
        NoticeTreeItem branch = new NoticeTreeItem("branch");
        assertTrue(branch.isBranch());
        assertFalse(branch.isLeaf());
        assertEquals(0, branch.getInternalChildren().size());

        branch.addChild(new NoticeTreeItem("branch"));
        branch.addChild(new NoticeTreeItem("leaf", "content"));
        assertEquals(2, branch.getInternalChildren().size());

        assertEquals("leaf", ((NoticeTreeItem) branch.getInternalChildren().get(1)).getTitle());
    }

    @Test
    public void testLeaf() {
        NoticeTreeItem leaf = new NoticeTreeItem("leaf", "content");
        assertEquals("leaf", leaf.getTitle());
        assertFalse(leaf.isBranch());
        assertTrue(leaf.isLeaf());
        assertEquals(0, leaf.getInternalChildren().size());
    }

    @Test
    public void testAddChild() {
        NoticeTreeItem branch = new NoticeTreeItem("branch");
        assertEquals(0, branch.getInternalChildren().size());
        for (int i = 0; i < 10; i++) {
            branch.addChild(null);
        }
        assertEquals(10, branch.getInternalChildren().size());
    }

    @Test
    public void testChangeContent() {
        NoticeTreeItem branch = new NoticeTreeItem("branch");
        assertNull(branch.getContent());
        branch.changeContent("test");
        assertNull(branch.getContent());

        NoticeTreeItem leaf = new NoticeTreeItem("leaf", "content");
        assertEquals("content", leaf.getContent());
        leaf.changeContent("test");
        assertEquals("test", leaf.getContent());
    }

}