def test_utc(self):
base = PostgreSQLBase(config=self.config)
# Verify that we've got 'timezone=utc' set
sql = 'SHOW TIMEZONE'
results = base.query(sql)
assert 'UTC' in results[0], (
'Please set PostgreSQL to use the UTC timezone. Documentation on how to do this is '
'included in the installation instructions.')
def test_utc(self):
base = PostgreSQLBase(config=self.config)
# Verify that we've got 'timezone=utc' set
sql = 'SHOW TIMEZONE'
results = base.query(sql)
assert 'UTC' in results[0], (
'Please set PostgreSQL to use the UTC timezone. Documentation on how to do this is '
'included in the installation instructions.'
)
def test_utc(self):
base = PostgreSQLBase(config=self.config)
# Verify that we've got 'timezone=utc' set
sql = 'SHOW TIMEZONE'
results = base.query(sql)
self.assertTrue('UTC' in results[0],
"""Please set PostgreSQL to use the UTC timezone.
Documentation on how to do this is included in
the INSTALL instructions.""")
def test_utc(self):
base = PostgreSQLBase(config=self.config)
# Verify that we've got 'timezone=utc' set
sql = 'SHOW TIMEZONE'
results = base.query(sql)
ok_(
'UTC' in results[0],
"""Please set PostgreSQL to use the UTC timezone.
Documentation on how to do this is included in
the INSTALL instructions.""")
def test_count(self):
base = PostgreSQLBase(config=self.config)
# A working count
sql = 'SELECT count(*) FROM reports'
count = base.count(sql)
eq_(count, 2)
# A working count with parameters
sql = 'SELECT count(*) FROM reports WHERE url=%(url)s'
params = {'url': 'http://mywebsite.com'}
count = base.count(sql, params)
eq_(count, 1)
# A failing count
sql = 'SELECT count(`invalid_field_name`) FROM reports'
assert_raises(DatabaseError, base.count, sql)
def test_count(self):
base = PostgreSQLBase(config=self.config)
# A working count
sql = 'SELECT count(*) FROM reports'
count = base.count(sql)
eq_(count, 2)
# A working count with parameters
sql = 'SELECT count(*) FROM reports WHERE url=%(url)s'
params = {'url': 'http://mywebsite.com'}
count = base.count(sql, params)
eq_(count, 1)
# A failing count
sql = 'SELECT count(`invalid_field_name`) FROM reports'
assert_raises(DatabaseError, base.count, sql)
def get_instance(self, config=None):
"""Return an instance of PostgreSQLBase with the config parameter as
a context or the default one if config is None.
"""
args = {
"config": config or self.get_dummy_context()
}
return PostgreSQLBase(**args)
def test_count(self):
base = PostgreSQLBase(config=self.config)
# A working count
sql = "SELECT count(*) FROM reports"
count = base.count(sql)
self.assertEqual(count, 2)
# A working count with parameters
sql = "SELECT count(*) FROM reports WHERE url=%(url)s"
params = {"url": "http://mywebsite.com"}
count = base.count(sql, params)
self.assertEqual(count, 1)
# A failing count
sql = "SELECT count(`invalid_field_name`) FROM reports"
self.assertRaises(DatabaseError, base.count, sql)
def test_count(self):
base = PostgreSQLBase(config=self.config)
# A working count
sql = 'SELECT count(*) FROM release_channels'
count = base.count(sql)
assert count == 5
# A working count with parameters
sql = 'SELECT count(*) FROM release_channels WHERE release_channel=%(release_channel)s'
params = {'release_channel': 'Nightly'}
count = base.count(sql, params)
assert count == 1
# A failing count
sql = 'SELECT count(`invalid_field_name`) FROM release_channels'
with pytest.raises(DatabaseError):
base.count(sql)
def test_query(self):
base = PostgreSQLBase(config=self.config)
# A working query
sql = 'SELECT * FROM reports'
results = base.query(sql)
eq_(len(results), 2)
ok_('http://mywebsite.com' in results[0])
ok_('*****@*****.**' in results[1])
# A working query with parameters
sql = 'SELECT * FROM reports WHERE url=%(url)s'
params = {'url': 'http://mywebsite.com'}
results = base.query(sql, params)
eq_(len(results), 1)
ok_('http://mywebsite.com' in results[0])
# A failing query
sql = 'SELECT FROM reports LIMIT notanumber'
assert_raises(DatabaseError, base.query, sql)
def test_query(self):
base = PostgreSQLBase(config=self.config)
# A working query
sql = 'SELECT * FROM reports'
results = base.query(sql)
eq_(len(results), 2)
ok_('http://mywebsite.com' in results[0])
ok_('*****@*****.**' in results[1])
# A working query with parameters
sql = 'SELECT * FROM reports WHERE url=%(url)s'
params = {'url': 'http://mywebsite.com'}
results = base.query(sql, params)
eq_(len(results), 1)
ok_('http://mywebsite.com' in results[0])
# A failing query
sql = 'SELECT FROM reports LIMIT notanumber'
assert_raises(DatabaseError, base.query, sql)
def test_query(self):
base = PostgreSQLBase(config=self.config)
# A working query
sql = "SELECT * FROM reports"
results = base.query(sql)
self.assertEqual(len(results), 2)
self.assertTrue("http://mywebsite.com" in results[0])
self.assertTrue("*****@*****.**" in results[1])
# A working query with parameters
sql = "SELECT * FROM reports WHERE url=%(url)s"
params = {"url": "http://mywebsite.com"}
results = base.query(sql, params)
self.assertEqual(len(results), 1)
self.assertTrue("http://mywebsite.com" in results[0])
# A failing query
sql = "SELECT FROM reports"
self.assertRaises(DatabaseError, base.query, sql)
def test_query(self):
base = PostgreSQLBase(config=self.config)
# A working query
sql = 'SELECT * FROM release_channels ORDER BY sort'
results = base.query(sql)
assert len(results) == 5
assert results[0] == ('Nightly', 1)
assert results[1] == ('Aurora', 2)
# A working query with parameters
sql = 'SELECT * FROM release_channels WHERE release_channel=%(release_channel)s'
params = {'release_channel': 'Nightly'}
results = base.query(sql, params)
assert len(results) == 1
assert results[0] == ('Nightly', 1)
# A failing query
sql = 'SELECT FROM release_channels LIMIT notanumber'
with pytest.raises(DatabaseError):
base.query(sql)
def test_count(self):
base = PostgreSQLBase(config=self.config)
# A working count
sql = 'SELECT count(*) FROM release_channels'
count = base.count(sql)
assert count == 5
# A working count with parameters
sql = 'SELECT count(*) FROM release_channels WHERE release_channel=%(release_channel)s'
params = {'release_channel': 'Nightly'}
count = base.count(sql, params)
assert count == 1
# A failing count
sql = 'SELECT count(`invalid_field_name`) FROM release_channels'
with pytest.raises(DatabaseError):
base.count(sql)
def test_query(self):
base = PostgreSQLBase(config=self.config)
# A working query
sql = 'SELECT * FROM reports'
results = base.query(sql)
assert len(results) == 2
assert 'http://mywebsite.com' in results[0]
assert '*****@*****.**' in results[1]
# A working query with parameters
sql = 'SELECT * FROM reports WHERE url=%(url)s'
params = {'url': 'http://mywebsite.com'}
results = base.query(sql, params)
assert len(results) == 1
assert 'http://mywebsite.com' in results[0]
# A failing query
sql = 'SELECT FROM reports LIMIT notanumber'
with pytest.raises(DatabaseError):
base.query(sql)
def test_query(self):
base = PostgreSQLBase(config=self.config)
# A working query
sql = 'SELECT * FROM release_channels ORDER BY sort'
results = base.query(sql)
assert len(results) == 5
assert results[0] == ('Nightly', 1)
assert results[1] == ('Aurora', 2)
# A working query with parameters
sql = 'SELECT * FROM release_channels WHERE release_channel=%(release_channel)s'
params = {'release_channel': 'Nightly'}
results = base.query(sql, params)
assert len(results) == 1
assert results[0] == ('Nightly', 1)
# A failing query
sql = 'SELECT FROM release_channels LIMIT notanumber'
with pytest.raises(DatabaseError):
base.query(sql)
