1 20 21 package com.methodhead.shim; 22 23 import java.util.*; 24 import java.io.*; 25 import java.sql.*; 26 import junit.framework.*; 27 import org.apache.log4j.*; 28 import com.methodhead.persistable.*; 29 import com.methodhead.test.*; 30 import com.methodhead.sitecontext.*; 31 32 public class HtmlFragmentTest extends TestCase { 33 34 private HtmlFragment fragment = null; 35 private HtmlFragment fragment2 = null; 36 private List list = null; 37 38 private SiteContext siteContext1_ = null; 39 private SiteContext siteContext2_ = null; 40 41 private HtmlFragment fragment1_ = null; 42 private HtmlFragment fragment2_ = null; 43 private HtmlFragment fragment3_ = null; 44 45 private void createData() { 46 47 siteContext1_ = new SiteContext(); 48 siteContext1_.getDomains().add( "methodhead.com" ); 49 siteContext1_.saveNew(); 50 51 siteContext2_ = new SiteContext(); 52 siteContext2_.getDomains().add( "danhensgen.com" ); 53 siteContext2_.saveNew(); 54 55 fragment1_ = new HtmlFragment(); 56 fragment1_.setSiteContext( siteContext1_ ); 57 fragment1_.setString( "name", "Fragment1" ); 58 fragment1_.setString( "value", "Value1" ); 59 fragment1_.saveNew(); 60 61 fragment2_ = new HtmlFragment(); 62 fragment2_.setSiteContext( siteContext1_ ); 63 fragment2_.setString( "name", "Fragment2" ); 64 fragment2_.setString( "value", "Value2" ); 65 fragment2_.saveNew(); 66 67 fragment3_ = new HtmlFragment(); 68 fragment3_.setSiteContext( siteContext2_ ); 69 fragment3_.setString( "name", "Fragment3" ); 70 fragment3_.setString( "value", "Value3" ); 71 fragment3_.saveNew(); 72 } 73 74 static { 75 TestUtils.initLogger(); 76 TestUtils.initDb(); 77 } 78 79 public HtmlFragmentTest( String name ) { 80 super( name ); 81 } 82 83 protected void setUp() { 84 try { 85 ConnectionSingleton.runBatchUpdate( new FileReader( "webapp/WEB-INF/db/transfer-reset.sql" ) ); 86 } 87 catch ( Exception e ) { 88 fail( e.getMessage() ); 89 } 90 } 91 92 protected void tearDown() { 93 } 94 95 public void testProperties() { 96 try { 97 createData(); 98 99 fragment = new HtmlFragment(); 100 101 try { 102 fragment.getSiteContext(); 103 fail( "Exception not thrown." ); 104 } 105 catch ( Exception e ) {} 106 107 fragment.setSiteContext( siteContext1_ ); 108 assertEquals( siteContext1_, fragment.getSiteContext() ); 109 } 110 catch ( Exception e ) { 111 e.printStackTrace(); 112 fail(); 113 } 114 } 115 116 public void testSaveNew() { 117 try { 118 createData(); 119 120 fragment = new HtmlFragment(); 121 fragment.setSiteContext( siteContext1_ ); 122 fragment.setString( "name", "name" ); 123 fragment.setString( "value", "value" ); 124 fragment.saveNew(); 125 126 fragment2 = new HtmlFragment(); 127 fragment2.load( "sitecontext_id=1 AND id=" + fragment.getInt( "id" ) ); 128 assertEquals( "name", fragment2.getString( "name" ) ); 129 assertEquals( "value", fragment2.getString( "value" ) ); 130 } 131 catch ( Exception e ) { 132 e.printStackTrace(); 133 fail(); 134 } 135 } 136 137 public void testLoadAll() { 138 try { 139 createData(); 140 141 fragment = new HtmlFragment(); 142 fragment.setSiteContext( siteContext1_ ); 143 144 list = fragment.loadAll(); 145 assertEquals( 2, list.size() ); 146 147 fragment = ( HtmlFragment )list.get( 0 ); 148 assertEquals( "Fragment1", fragment.getString( "name" ) ); 149 150 fragment = ( HtmlFragment )list.get( 1 ); 151 assertEquals( "Fragment2", fragment.getString( "name" ) ); 152 } 153 catch ( Exception e ) { 154 e.printStackTrace(); 155 fail(); 156 } 157 } 158 159 public void testDeleteAll() { 160 try { 161 createData(); 162 163 fragment = new HtmlFragment(); 164 fragment.setSiteContext( siteContext1_ ); 165 fragment.deleteAll(); 166 167 fragment = new HtmlFragment(); 168 fragment.setSiteContext( siteContext1_ ); 169 list = fragment.loadAll(); 170 assertEquals( 0, list.size() ); 171 172 fragment = new HtmlFragment(); 173 fragment.setSiteContext( siteContext2_ ); 174 list = fragment.loadAll(); 175 assertEquals( 1, list.size() ); 176 } 177 catch ( Exception e ) { 178 e.printStackTrace(); 179 fail(); 180 } 181 } 182 } 183 | Popular Tags |