parkrunDb/test.py at master · jones139/parkrunDb · GitHub

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
#!/usr/bin/python
"""
Tests for parkrunDb
"""
import unittest
import json
from parkrunDbLib import parkrunDbLib

class dbTests(unittest.TestCase):
    def setUp(self):
        print "setup()"
        self.db = parkrunDbLib("test.db")
        self.db.initialiseDb("createdb.sqlite")


class TestDates(dbTests):
    def testDateConv(self):
        dateStr = "31/01/2018"
        ts = self.db.dateStr2ts(dateStr)
        dateStr2 = self.db.ts2dateStr(ts)
        print "DateStr=%s, ts=%d, DateStr2=%s" % (dateStr,ts,dateStr2)
        self.assertEqual(dateStr,dateStr2)

class TestEvents(dbTests):
    def testAddEvent(self):
        dateStr = "31/01/2018"
        eventNo = 999
        prId = 1
        evId = self.db.addEvent(eventNo,prId,self.db.dateStr2ts(dateStr))
        evId2 = self.db.getEventId(prId,self.db.dateStr2ts(dateStr))
        self.assertEqual(evId,evId2)

        evId3 = self.db.getEventIdFromEventNo(prId,eventNo)
        self.assertEqual(evId,evId3)

    def testAddRun(self):
        dateStr = "31/01/2018"
        eventNo = 999
        prId = 1
        evId = self.db.addEvent(eventNo,prId,self.db.dateStr2ts(dateStr))

        runnerId = 1
        roleId = 1

        self.assertEqual(self.db.getRunId(evId,runnerId,roleId),-1)
        runId = self.db.addRun(evId,runnerId,roleId,999,"V45",60.2,1,3,"Note")
        runId2 = self.db.getRunId(evId,runnerId,roleId)
        self.assertEqual(runId,runId2)

    def testAddRunner(self):
        self.assertEqual(self.db.getRunnerId(123456),-1)
        runnerId = self.db.addRunner(123456,"Blogs, Joe","HBRH","M")
        runnerId2 = self.db.getRunnerId(123456)
        self.assertEqual(runnerId,runnerId2)

        runnerData = self.db.getRunner(runnerId)
        self.assertEqual(runnerData[0],123456)


        runnerId = self.db.addRunner(1234567,"Blogs, Joe2","HBRH","M")
        runnerId2 = self.db.getRunnerId(1234567)
        self.assertEqual(runnerId,runnerId2)

    def testUpdateRunner(self):
        # make sure we have a clean database
        self.assertEqual(self.db.getRunnerId(123456),-1)
        runnerId = self.db.addRunner(123456,"Blogs, Joe","HBRH","F")
        nRows = self.db.updateRunner(runnerId, 123456,"Blogs, Joe","HBRH","M")
        self.assertEqual(nRows,1)
        runnerData = self.db.getRunner(runnerId)
        self.assertEqual(runnerData[3],"M")
        runnerId2 = self.db.getRunnerId(123456)
        self.assertEqual(runnerId,runnerId2)


    def testFindRunner(self):
        self.db = parkrunDbLib("test.db","iddb.json")
        self.db.initialiseDb("createdb.sqlite")

        nameStr = "Roy WATKINS"
        runnerNo = self.db.getRunnerNoFromName(nameStr)
        self.assertEqual(runnerNo,1455360)

        nameStr = "Bobby WILLIAMS"
        runnerNo = self.db.getRunnerNoFromName(nameStr)
        self.assertEqual(runnerNo,1488327)


        nameStr = "NON EXISTANT RUNNER"
        runnerNo = self.db.getRunnerNoFromName(nameStr)
        self.assertEqual(runnerNo,-1)


if __name__ == '__main__':
    unittest.main()