|
1
9 package org.jboss.portal.test.setup.dl;
10
11
12 import org.jboss.portal.setup.impl.dl.dbloader.hibernate.HibernateDataLoader;
13 import org.jboss.portal.setup.impl.dl.dbloader.hibernate.HibernateDataLoaderConfigImpl;
14 import org.jboss.portal.setup.impl.pm.HibernatePersistenceManager;
15 import org.jboss.portal.setup.impl.pm.HibernatePesistenceSession;
16 import org.jboss.portal.setup.PortalSetupException;
17
18
19 import org.jboss.portal.common.util.XML;
20 import org.jboss.portal.test.setup.PortalSetupBaseTestCase;
21 import org.jboss.portal.core.impl.user.UserModuleImpl;
22 import org.jboss.portal.core.model.User;
23 import org.w3c.dom.Document  ;
24 import org.w3c.dom.Element ;
25 import org.w3c.dom.NodeList ;
26 import org.hibernate.Session;
27
28 import java.io.File ;
29 import java.io.FileInputStream ;
30 import java.io.InputStream ;
31 import java.util.List ;
32 import java.util.Set ;
33
34
35 import junit.framework.Assert;
36
37
38
44 public class HibernateDataLoaderTestCase extends PortalSetupBaseTestCase
45 {
46
47 protected HibernateDataLoader m_dataLoader = null;
48 protected HibernateDataLoaderConfigImpl m_config = null;
49 private UserModuleImpl m_userModule = null;
50 private HibernatePersistenceManager m_pm = null;
51 private HibernatePesistenceSession m_ps = null;
52
53
54 public HibernateDataLoaderTestCase(String name)
55 {
56 super(name);
57
58 }
59
60 public void setUp() throws Exception
61 {
62 try
63 {
64 super.setUp();
65 setDataLoaderConfig();
66 m_dataLoader = new HibernateDataLoader();
67 m_dataLoader.setDataLoaderConfiguration(m_config);
68 }
69 catch (PortalSetupException pse)
70 {
71 pse.printStackTrace();
72 assertTrue(false);
73 }
74
75 m_userModule = new UserModuleImpl()
77 {
78 protected Session getSession()
79 {
80 return m_ps.getSession();
81 }
82 };
83
84 }
85
86 public void testLoadData() throws Exception
87 {
88 String uri = "org/jboss/portal/setup/impl/dl/dbloader/portal-dl.xml";
90 if (m_dataLoader.checkDataInitialized())
91 {
92 System.out.println("Database data have been initialized!");
94 }
95 else
96 {
97 InputStream is = null;
98 File file = new File (uri);
99 if (file.exists())
100 {
101 is = new FileInputStream (file);
102 }
103 else
104 {
105 is = Thread.currentThread().getContextClassLoader().getResourceAsStream(uri);
106 }
107
108
109 Document dmt = XML.getDocumentBuilderFactory().newDocumentBuilder().parse(is);
110 Element docElmt = dmt.getDocumentElement();
111 List actions = XML.getChildren(docElmt, "action");
112 for (int i = 0; i < actions.size(); i++)
113 {
114 Element action = (Element )actions.get(i);
115 m_dataLoader.loadData(action);
116 }
117 }
118 Set users = m_userModule.findUsers(0, 200);
119 User user = m_userModule.findUserByUserName("admin");
121 boolean enabled = user.getEnabled();
122 Assert.assertTrue(enabled);
123
124
125 }
126
127 public void tearDown() throws Exception
128 {
129 m_dataLoader.destroy();
130 m_dataLoader = null;
131 m_ps.close();
132 m_pm.destroy();
133 super.tearDown();
134
135 }
136
137 protected void setDataLoaderConfig() throws PortalSetupException
138 {
139
140 m_config = new HibernateDataLoaderConfigImpl();
141 m_config.setDataQueryString("SELECT JBP_UNAME FROM JBP_USERS WHERE JBP_UID = 1");
142 m_config.setHibernateConfig(m_hbConfig);
143 m_pm = new HibernatePersistenceManager();
144 m_pm.initialize(m_hbConfig, null);
145 m_ps = (HibernatePesistenceSession)m_pm.getSession();
146 }
147
148
162 }
163 |
Popular Tags |
Java API By Example, From Geeks To Geeks.