1 package org.columba.mail.gui.table.model; 19 20 import java.util.Date ; 21 22 import junit.framework.TestCase; 23 24 import org.columba.mail.folder.headercache.MemoryHeaderList; 25 import org.columba.mail.message.ColumbaHeader; 26 import org.columba.mail.message.IHeaderList; 27 import org.frapuccino.treetable.Tree; 28 import org.frapuccino.treetable.TreeTable; 29 30 34 public class HeaderTableModelTest extends TestCase { 35 36 public static String [] columns = { "Subject", "From", "columba.date"}; 37 38 protected IHeaderList createHeaderList() { 39 IHeaderList list = new MemoryHeaderList(); 40 ColumbaHeader h = new ColumbaHeader(); 41 h.set("columba.uid", new Integer (0)); 42 h.set("Subject", "Test1"); 43 h.set("From", "test@test.de"); 44 h.set("columba.date", new Date ()); 45 list.add(h, new Integer (0)); 46 47 h = new ColumbaHeader(); 48 h.set("columba.uid", new Integer (1)); 49 h.set("Subject", "Test2"); 50 h.set("From", "test@test.de"); 51 h.set("columba.date", new Date ()); 52 list.add(h, new Integer (1)); 53 54 return list; 55 } 56 57 61 public void testColumns() { 62 HeaderTableModel model = new HeaderTableModel(columns); 63 64 assertEquals(3, model.getColumnCount()); 66 } 67 68 public void testSet() { 69 HeaderTableModel model = new HeaderTableModel(columns); 70 71 TreeTable treetable = new TreeTable(); 72 treetable.setModel(model); 73 model.setTree((Tree) treetable.getTree()); 74 75 IHeaderList list = createHeaderList(); 77 78 model.set(list); 79 80 assertEquals(2, model.getRootNode().getChildCount()); 82 83 86 assertEquals(2, model.getRowCount()); 88 89 assertEquals(2, model.getTree().getRowCount()); 91 92 } 93 94 public void testRemove() { 95 HeaderTableModel model = new HeaderTableModel(columns); 96 Tree tree = new Tree(); 97 model.setTree(tree); 98 99 IHeaderList list = createHeaderList(); 100 101 model.set(list); 102 103 model.remove(new Object [] { new Integer (0)}); 105 106 assertEquals(1, model.getRootNode().getChildCount()); 108 111 assertEquals(1, model.getRowCount()); 113 114 assertEquals(1, model.getTree().getRowCount()); 116 } 117 118 public void testModify() { 119 HeaderTableModel model = new HeaderTableModel(columns); 120 Tree tree = new Tree(); 121 model.setTree(tree); 122 123 IHeaderList list = createHeaderList(); 124 125 model.set(list); 126 } 127 } | Popular Tags |