-
Notifications
You must be signed in to change notification settings - Fork 51
/
ArquillianTest.java
102 lines (91 loc) · 3.51 KB
/
ArquillianTest.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*
* Copyright 2010, Red Hat, Inc. and individual contributors as indicated by the
* @author tags. See the copyright.txt file in the distribution for a full
* listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License as published by the Free
* Software Foundation; either version 2.1 of the License, or (at your option)
* any later version.
*
* This software is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
* details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this software; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA, or see the FSF
* site: http://www.fsf.org.
*/
package org.zanata;
import javax.naming.InitialContext;
import javax.sql.DataSource;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.operation.DatabaseOperation;
import org.jboss.arquillian.junit.Arquillian;
import org.junit.After;
import org.junit.Before;
import org.junit.runner.RunWith;
import org.zanata.provider.DBUnitProvider;
/**
* Base class for all Arquillian tests. Provides DB Unit facilities for
* in-container tests.
*
* @author Carlos Munoz <a
* href="mailto:camunoz@redhat.com">camunoz@redhat.com</a>
*/
@RunWith(Arquillian.class)
public abstract class ArquillianTest {
private DBUnitProvider dbUnitProvider = new DBUnitProvider(
ArquillianTest::getConnection);
/**
* Implement this in a subclass.
* <p/>
* Use it to stack DBUnit <tt>DataSetOperation</tt>'s with the
* <tt>beforeTestOperations</tt> and <tt>afterTestOperations</tt> lists.
*/
protected abstract void prepareDBUnitOperations();
public void
addBeforeTestOperation(DBUnitProvider.DataSetOperation operation) {
dbUnitProvider.addBeforeTestOperation(operation);
}
public void
addAfterTestOperation(DBUnitProvider.DataSetOperation operation) {
dbUnitProvider.addAfterTestOperation(operation);
}
/**
* Invoked on the arquillian container before the test is run.
*/
@Before
public void prepareDataBeforeTest() {
addBeforeTestOperation(new DBUnitProvider.DataSetOperation(
"org/zanata/test/model/ClearAllTables.dbunit.xml",
DatabaseOperation.DELETE_ALL));
prepareDBUnitOperations();
addAfterTestOperation(new DBUnitProvider.DataSetOperation(
"org/zanata/test/model/ClearAllTables.dbunit.xml",
DatabaseOperation.DELETE_ALL));
dbUnitProvider.prepareDataBeforeTest();
}
/**
* Invoked in the arquillian container after the test is run.
*/
@After
public void cleanDataAfterTest() {
dbUnitProvider.cleanDataAfterTest();
}
private static IDatabaseConnection getConnection() {
try {
DataSource dataSource =
(DataSource) new InitialContext().lookup(
"java:jboss/datasources/zanataDatasource");
DatabaseConnection dbConn =
new DatabaseConnection(dataSource.getConnection());
return dbConn;
} catch (Exception e) {
throw new RuntimeException(e);
}
}
}