1 package compbio.stat.collector;
\r
3 import static org.testng.Assert.assertEquals;
\r
4 import static org.testng.Assert.assertNotNull;
\r
6 import java.sql.SQLException;
\r
7 import java.sql.Timestamp;
\r
8 import java.util.Calendar;
\r
9 import java.util.Date;
\r
10 import java.util.List;
\r
12 import org.testng.Assert;
\r
13 import org.testng.annotations.BeforeClass;
\r
14 import org.testng.annotations.Test;
\r
16 import compbio.stat.servlet.util.StatCollection;
\r
17 import compbio.stat.servlet.util.Totals;
\r
18 import compbio.ws.client.Services;
\r
20 public class StatDBTester {
\r
24 public void init() {
\r
26 statdb = new StatDB(true);
\r
27 } catch (SQLException e) {
\r
28 e.printStackTrace();
\r
29 Assert.fail(e.getLocalizedMessage());
\r
34 public void testReadYearDataAndShutdown() {
\r
35 Calendar cal = Calendar.getInstance();
\r
36 cal.roll(Calendar.YEAR, false);
\r
38 List<JobStat> jobs = statdb.readData(
\r
39 new Timestamp(cal.getTimeInMillis()), new Timestamp(
\r
40 new Date().getTime()), Services.MuscleWS, false);
\r
41 assertNotNull(jobs);
\r
42 assertEquals(jobs.size(), 1294);
\r
43 // statdb.shutdownDBServer();
\r
44 } catch (SQLException e) {
\r
45 e.printStackTrace();
\r
46 Assert.fail(e.getMessage());
\r
51 public void testReadOneMonthData() {
\r
52 Calendar cal = Calendar.getInstance();
\r
53 cal.roll(Calendar.MONTH, false);
\r
55 List<JobStat> jobs = statdb.readData(
\r
56 new Timestamp(cal.getTimeInMillis()), new Timestamp(
\r
57 new Date().getTime()), Services.TcoffeeWS, false);
\r
58 assertNotNull(jobs);
\r
59 assertEquals(jobs.size(), 47);
\r
61 jobs = statdb.readData(new Timestamp(cal.getTimeInMillis()),
\r
62 new Timestamp(new Date().getTime()), Services.ClustalWS,
\r
65 assertNotNull(jobs);
\r
66 assertEquals(jobs.size(), 144);
\r
68 jobs = statdb.readData(new Timestamp(cal.getTimeInMillis()),
\r
69 new Timestamp(new Date().getTime()), Services.MafftWS,
\r
71 assertNotNull(jobs);
\r
72 assertEquals(jobs.size(), 149);
\r
74 jobs = statdb.readData(new Timestamp(cal.getTimeInMillis()),
\r
75 new Timestamp(new Date().getTime()), Services.ProbconsWS,
\r
77 assertNotNull(jobs);
\r
78 assertEquals(jobs.size(), 12);
\r
80 jobs = statdb.readData(new Timestamp(cal.getTimeInMillis()),
\r
81 new Timestamp(new Date().getTime()), Services.MuscleWS,
\r
83 assertNotNull(jobs);
\r
84 assertEquals(jobs.size(), 76);
\r
86 } catch (SQLException e) {
\r
87 e.printStackTrace();
\r
88 Assert.fail(e.getMessage());
\r
93 public void testGetEarliestRecord() {
\r
96 StatDB db = new StatDB();
\r
97 Date earliestRec = db.getEarliestRecord();
\r
98 // System.out.println(earliestRec);
\r
99 } catch (SQLException e) {
\r
100 e.printStackTrace();
\r
101 Assert.fail(e.getLocalizedMessage());
\r
107 public void testVerifyJobsCount() {
\r
110 Calendar cal = Calendar.getInstance();
\r
111 cal.add(Calendar.MONTH, -5);
\r
112 Timestamp from = new Timestamp(cal.getTimeInMillis());
\r
113 cal.add(Calendar.MONTH, 1);
\r
114 Timestamp to = new Timestamp(cal.getTimeInMillis());
\r
115 StatCollection sc = StatCollection.newStatCollecton(from, to);
\r
116 Totals t = Totals.sumStats(sc.getAllStat());
\r
118 System.out.println(sc.getAllStat());
\r
119 StatDB db = new StatDB();
\r
120 assertEquals(t.getTotal(), db.getTotalJobsCount(from, to));
\r
121 assertEquals(t.getAbandoned(), db.getAbandonedCount(from, to));
\r
122 assertEquals(t.getCancelled(), db.getCancelledCount(from, to));
\r
123 assertEquals(t.getIncomplete(), db.getIncompleteCount(from, to));
\r
125 } catch (SQLException e) {
\r
126 e.printStackTrace();
\r
127 Assert.fail(e.getLocalizedMessage());
\r