1
|
package eu.dnetlib.data.mapreduce.hbase.statsExport.utils;
|
2
|
|
3
|
import java.io.InputStream;
|
4
|
import java.util.Properties;
|
5
|
|
6
|
import org.apache.hadoop.conf.Configuration;
|
7
|
import org.apache.log4j.Logger;
|
8
|
import org.junit.Before;
|
9
|
import org.junit.Test;
|
10
|
|
11
|
import eu.dnetlib.data.mapreduce.hbase.statsExport.daos.SqlDAO;
|
12
|
import eu.dnetlib.data.mapreduce.hbase.statsExport.daos.SqlStore;
|
13
|
import eu.dnetlib.data.mapreduce.hbase.statsExport.drivers.DBDriver;
|
14
|
import eu.dnetlib.data.mapreduce.hbase.statsExport.drivers.SqoopDriver;
|
15
|
import eu.dnetlib.data.mapreduce.hbase.statsExport.drivers.StatsJobDriver;
|
16
|
|
17
|
public class DaoTest {
|
18
|
|
19
|
private Logger log = Logger.getLogger(this.getClass());
|
20
|
|
21
|
private SqlDAO statsDao;
|
22
|
private SqlStore statsStore;
|
23
|
|
24
|
private Properties props;
|
25
|
|
26
|
@Before
|
27
|
public void init() throws Exception {
|
28
|
InputStream file = ClassLoader.getSystemResourceAsStream("eu/dnetlib/data/mapreduce/hbase/statsExport/StatsProperties");
|
29
|
props = new Properties();
|
30
|
props.load(file);
|
31
|
file.close();
|
32
|
|
33
|
statsStore = new SqlStore();
|
34
|
statsStore.setDbDriver(props.getProperty("Stats_db_Driver"));
|
35
|
statsStore.setDB_URL(props.getProperty("Stats_db_Url"));
|
36
|
statsStore.setDbUser(props.getProperty("Stats_db_User"));
|
37
|
statsStore.setDbPassword(props.getProperty("Stats_db_Pass"));
|
38
|
|
39
|
statsDao = new SqlDAO();
|
40
|
|
41
|
statsDao.setStore(statsStore);
|
42
|
|
43
|
}
|
44
|
|
45
|
@Test
|
46
|
public void getReports() throws Exception {
|
47
|
|
48
|
statsDao.generateReports();
|
49
|
|
50
|
}
|
51
|
|
52
|
@Test
|
53
|
public void installProcedures() throws Exception {
|
54
|
statsStore.getConnection();
|
55
|
|
56
|
InputStream in = ClassLoader.getSystemResourceAsStream("eu/dnetlib/data/mapreduce/hbase/statsExport/reportProcedures.sql");
|
57
|
byte[] b = new byte[in.available()];
|
58
|
in.read(b);
|
59
|
String q = new String(b);
|
60
|
|
61
|
in.close();
|
62
|
|
63
|
log.info(" Creating Report Procedures in Temporary Database ....");
|
64
|
// CREATE USER 'jeffrey'@'localhost' IDENTIFIED BY 'mypass';
|
65
|
try {
|
66
|
statsStore.executeCommand(q);
|
67
|
|
68
|
} catch (Exception e) {
|
69
|
log.error("Could not create Report Procedures in Temporary database :" + e.getMessage());
|
70
|
throw new Exception("Could not create Report Procedures in Temporary database :", e);
|
71
|
} finally {
|
72
|
statsStore.closeConnection();
|
73
|
}
|
74
|
|
75
|
log.info(" Done.");
|
76
|
|
77
|
}
|
78
|
|
79
|
// @Test
|
80
|
public void executeExtraInserts() throws Exception {
|
81
|
|
82
|
statsDao.executeExtraInserts();
|
83
|
|
84
|
}
|
85
|
|
86
|
// @Test
|
87
|
public void buildIndexesDb() throws Exception {
|
88
|
|
89
|
statsDao.executeExtraInserts();
|
90
|
|
91
|
}
|
92
|
|
93
|
}
|