Project

General

Profile

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
}
(2-2/5)