1 20 21 package com.methodhead.shim; 22 23 import java.util.*; 24 import java.sql.*; 25 import java.io.*; 26 import junit.framework.*; 27 import org.apache.log4j.*; 28 import com.methodhead.persistable.*; 29 import com.methodhead.test.*; 30 import org.apache.cactus.*; 31 import com.methodhead.sitecontext.*; 32 import com.methodhead.*; 33 34 public class ShimUtilsTest extends ServletTestCase { 35 36 SiteMap siteMap = null; 37 SiteMap siteMap2 = null; 38 Link link = null; 39 private Module module = null; 40 private List list = null; 41 42 static { 43 TestUtils.initLogger(); 44 TestUtils.initDb(); 45 } 46 47 public ShimUtilsTest( String name ) { 48 super( name ); 49 } 50 51 100 101 protected void setUp() { 102 try { 104 ConnectionSingleton.runBatchUpdate( new FileReader( "webapp/WEB-INF/db/transfer-reset.sql" ) ); 105 106 session.getServletContext().removeAttribute( ShimGlobals.SITEMAPMAP_KEY ); 107 108 135 } 136 catch ( Exception e ) { 137 fail( e.getMessage() ); 138 } 139 } 140 141 protected void tearDown() { 142 } 143 144 public void testSetSiteMap() { 145 try { 146 TestData.createLinks(); 147 148 siteMap = new SiteMap(); 149 siteMap.setSiteContext( SiteContext.getDefaultContext() ); 150 siteMap.setRoot( TestData.link1 ); 151 siteMap.save(); 152 153 ShimUtils.setSiteMap( request, siteMap ); 154 155 Map map = ( Map )session.getServletContext().getAttribute( ShimGlobals.SITEMAPMAP_KEY ); 156 157 assertNotNull( map ); 158 159 siteMap2 = ( SiteMap )map.get( SiteContext.getDefaultContext().get( "id" ) ); 160 161 assertNotNull( siteMap2 ); 162 163 link = ( Link )siteMap2.getRoot(); 164 165 assertNotNull( link ); 166 assertEquals( TestData.link1, link ); 167 } 168 catch ( Exception e ) { 169 e.printStackTrace(); 170 fail(); 171 } 172 } 173 174 public void testGetSiteMap() { 175 try { 176 177 try { 181 siteMap2 = ShimUtils.getSiteMap( request ); 182 } 183 catch ( Exception e ) { 184 } 186 187 TestData.createSiteMap(); 188 189 SiteContext.setContext( request, SiteContext.getDefaultContext() ); 190 191 siteMap2 = ShimUtils.getSiteMap( request ); 192 193 Map map = ( Map )session.getServletContext().getAttribute( ShimGlobals.SITEMAPMAP_KEY ); 194 195 assertNotNull( map ); 196 197 siteMap2 = ( SiteMap )map.get( SiteContext.getDefaultContext().get( "id" ) ); 198 199 assertNotNull( siteMap2 ); 200 201 link = ( Link )siteMap2.getRoot(); 202 203 assertNotNull( link ); 204 assertEquals( TestData.link1.getPageId(), link.getPageId() ); 205 } 206 catch ( Exception e ) { 207 e.printStackTrace(); 208 fail(); 209 } 210 } 211 212 public void testGetModules() { 213 try { 214 list = ShimUtils.getModules( SiteContext.getDefaultContext() ); 215 216 assertNotNull( list ); 217 assertEquals( 3, list.size() ); 218 219 module = ( Module )list.get( 0 ); 220 assertNotNull( module ); 221 assertEquals( "Text", module.getName() ); 222 223 module = ( Module )list.get( 1 ); 224 assertNotNull( module ); 225 assertEquals( "Navigation", module.getName() ); 226 227 module = ( Module )list.get( 2 ); 228 assertNotNull( module ); 229 assertEquals( "Include", module.getName() ); 230 } 231 catch ( Exception e ) { 232 e.printStackTrace(); 233 fail(); 234 } 235 } 236 237 public void testGetLinkUrl() { 238 try { 239 link = new Link(); 240 link.setTitle( "Title" ); 241 link.setAlias( "title" ); 242 link.setPageId( 666 ); 243 244 assertEquals( "title.shtml", ShimUtils.getLinkUrl( link ) ); 245 } 246 catch ( Exception e ) { 247 e.printStackTrace(); 248 fail(); 249 } 250 } 251 252 public void testFileNameToLabel() { 253 assertEquals( "<null>", ShimUtils.fileNameToLabel( null ) ); 254 assertEquals( "", ShimUtils.fileNameToLabel( "" ) ); 255 assertEquals( "", ShimUtils.fileNameToLabel( " " ) ); 256 assertEquals( "", ShimUtils.fileNameToLabel( "\t" ) ); 257 assertEquals( "T", ShimUtils.fileNameToLabel( "t" ) ); 258 assertEquals( "T", ShimUtils.fileNameToLabel( "T" ) ); 259 assertEquals( "T", ShimUtils.fileNameToLabel( "t.txt" ) ); 260 assertEquals( "T", ShimUtils.fileNameToLabel( "T.txt" ) ); 261 assertEquals( "Test", ShimUtils.fileNameToLabel( "test" ) ); 262 assertEquals( "Test", ShimUtils.fileNameToLabel( "test.txt" ) ); 263 assertEquals( "Test", ShimUtils.fileNameToLabel( "Test" ) ); 264 assertEquals( "Test", ShimUtils.fileNameToLabel( "Test.txt" ) ); 265 assertEquals( "My Test", ShimUtils.fileNameToLabel( "myTest.txt" ) ); 266 assertEquals( "My Test", ShimUtils.fileNameToLabel( "MyTest.txt" ) ); 267 assertEquals( "My Test2 Test", ShimUtils.fileNameToLabel( "MyTest2Test.txt" ) ); 268 assertEquals( "M T A", ShimUtils.fileNameToLabel( "MTA.txt" ) ); 269 assertEquals( "My Test", ShimUtils.fileNameToLabel( "MyTest." ) ); 270 assertEquals( "My Test", ShimUtils.fileNameToLabel( "My Test.txt" ) ); 271 assertEquals( "My-test", ShimUtils.fileNameToLabel( "my-test.txt" ) ); 272 } 273 } 274 | Popular Tags |