← Back to branch summary

~awuerl/blitzortung-python/master

« back to all changes in this revision

Viewing changes to tests/db/test_db_table.py

  • Committer: Andreas Wuerl
  • Date: 2019-06-24 19:32:25 UTC
  • mto: This revision was merged to the branch mainline in revision 394.
  • Revision ID: git-v1:29d4be282c05035bdff8d4828573537ed6d8ceea
update tests

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/db/test_db_table.py
18
18
 
19
19
"""
20
20
 
21
 
import unittest
22
21
import datetime
23
22
 
 
23
import psycopg2
24
24
import pytz
 
25
from assertpy import assert_that
25
26
from mock import Mock, call
26
 
from hamcrest import assert_that, is_, equal_to, none
27
 
import psycopg2
28
27
 
29
 
import blitzortung
30
28
import blitzortung.db.table
31
29
 
32
30
 
44
42
        return args
45
43
 
46
44
 
47
 
class BaseTest(unittest.TestCase):
48
 
    def setUp(self):
 
45
class BaseTest(object):
 
46
    def setup_method(self):
49
47
        self.connection_pool = Mock()
50
48
        self.connection = self.connection_pool.getconn()
51
49
        self.cursor = self.connection.cursor()
72
70
 
73
71
    def test_is_connected(self):
74
72
        self.connection.closed = True
75
 
        assert_that(self.base.is_connected(), is_(False))
 
73
        assert_that(self.base.is_connected()).is_equal_to(False)
76
74
 
77
75
        self.connection.closed = False
78
 
        assert_that(self.base.is_connected())
 
76
        assert_that(self.base.is_connected()).is_true()
79
77
 
80
78
        self.base.conn = None
81
 
        assert_that(self.base.is_connected(), is_(False))
 
79
        assert_that(self.base.is_connected()).is_false()
82
80
 
83
81
    def test_full_table_name(self):
84
 
        assert_that(self.base.full_table_name, is_(""))
85
 
        assert_that(self.base.schema_name, is_(""))
 
82
        assert_that(self.base.full_table_name).is_equal_to("")
 
83
        assert_that(self.base.schema_name).is_equal_to("")
86
84
 
87
85
        self.base.table_name = "foo"
88
86
 
89
 
        assert_that(self.base.full_table_name, is_("foo"))
 
87
        assert_that(self.base.full_table_name).is_equal_to("foo")
90
88
 
91
89
        self.base.schema_name = "bar"
92
90
 
93
 
        assert_that(self.base.full_table_name, is_('"bar"."foo"'))
94
 
        assert_that(self.base.schema_name, is_("bar"))
 
91
        assert_that(self.base.full_table_name).is_equal_to('"bar"."foo"')
 
92
        assert_that(self.base.schema_name).is_equal_to("bar")
95
93
 
96
94
    def test_get_timezone(self):
97
 
        assert_that(self.base.get_timezone(), is_(equal_to(pytz.UTC)))
 
95
        assert_that(self.base.get_timezone()).is_equal_to(pytz.UTC)
98
96
 
99
97
    def test_fix_timezone(self):
100
 
        assert_that(self.base.fix_timezone(None), is_(none()))
 
98
        assert_that(self.base.fix_timezone(None)).is_none()
101
99
 
102
100
        time = datetime.datetime(2013, 1, 1, 12, 0, 0, tzinfo=pytz.timezone("CET"))
103
101
        utc_time = datetime.datetime(2013, 1, 1, 11, 0, 0, tzinfo=pytz.timezone("UTC"))
104
102
 
105
 
        assert_that(self.base.fix_timezone(time), is_(utc_time))
 
103
        assert_that(self.base.fix_timezone(time)).is_equal_to(utc_time)
106
104
 
107
105
    def test_from_bare_utc_to_timezone(self):
108
106
        self.base.set_timezone(pytz.timezone("CET"))
110
108
        time = datetime.datetime(2013, 1, 1, 12, 0, 0, tzinfo=pytz.timezone("CET"))
111
109
        utc_time = datetime.datetime(2013, 1, 1, 11, 0, 0)
112
110
 
113
 
        assert_that(self.base.from_bare_utc_to_timezone(utc_time), is_(equal_to(time)))
 
111
        assert_that(self.base.from_bare_utc_to_timezone(utc_time)).is_equal_to(time)
114
112
 
115
113
    def test_from_timezone_to_bare_utc(self):
116
114
        self.base.set_timezone(pytz.timezone("CET"))
118
116
        time = datetime.datetime(2013, 1, 1, 12, 0, 0, tzinfo=pytz.timezone("CET"))
119
117
        utc_time = datetime.datetime(2013, 1, 1, 11, 0, 0)
120
118
 
121
 
        assert_that(self.base.from_timezone_to_bare_utc(time), is_(equal_to(utc_time)))
 
119
        assert_that(self.base.from_timezone_to_bare_utc(time)).is_equal_to(utc_time)
122
120
 
123
121
    def test_commit(self):
124
122
        self.connection.commit.assert_not_called()