def test_newest_election():
"""
Use newest election
"""
ena_util_in = ena_util.ENAUtils('MN', False, debug=True, db_file=db_file)
ena_util_in.setup()
assert type(ena_util_in.election) == dict
def test_timestamp():
"""
Test timestamp.
"""
ena_util_in = ena_util.ENAUtils('MN', False, debug=True, db_file=db_file)
ena_util_in.setup()
ts = ena_util_in.timestamp()
assert type(ts) == int and ts > 0
def test_instantiation():
"""
Just test basic instationation.
"""
ena_util_in = ena_util.ENAUtils('MN',
'20141104',
debug=True,
db_file=db_file)
assert isinstance(ena_util_in, ena_util.ENAUtils)
def create_scraper(election=False):
"""
Wrapper to create instance of VA scraper.
"""
if os.path.exists(db_file):
os.remove(db_file)
# Create util and scraper instance
util = ena_util.ENAUtils('VA', election, True, db_file=db_file)
return va.Scraper(util)
def test_save_meta():
"""
Test a save.
"""
ena_util_in = ena_util.ENAUtils('MN', False, debug=True, db_file=db_file)
ena_util_in.setup()
ena_util_in.save_meta('test', 1234)
rows = ena_util_in.sql.select('* FROM meta')
assert rows != []
def test_setup():
"""
Test setup. This does not seem to work in conjucntion with state level
tests and probably has something to do with scraperwiki library and the
global database name
"""
ena_util_in = ena_util.ENAUtils('MN', False, debug=True, db_file=db_file)
ena_util_in.setup()
#assert os.path.exists(ena_util_in.db_file) and os.path.isfile(ena_util_in.db_file)
assert True
def test_google_spreadsheet():
"""
Test a save.
"""
ena_util_in = ena_util.ENAUtils('MN', False, debug=True, db_file=db_file)
ena_util_in.setup()
rows = ena_util_in.google_spreadsheet(
'1f3uc7P-WEeqJPkIlN14lhtGghRyIVsV08k6QISi2JNs', 0)
assert rows != []
def test_has_table():
"""
Test has_table().
"""
ena_util_in = ena_util.ENAUtils('MN', False, debug=True, db_file=db_file)
ena_util_in.setup()
# Should not have by default
assert ena_util_in.has_table('test_table_foo') == False
# Should have after save
ena_util_in.save_meta('test', 1234)
assert ena_util_in.has_table('meta') == True
def test_save():
"""
Test a save.
"""
ena_util_in = ena_util.ENAUtils('MN', False, debug=True, db_file=db_file)
ena_util_in.setup()
ena_util_in.save(['id'], {
'id': 123,
'foo': 'bar'
}, 'test')
rows = ena_util_in.sql.select('* FROM test')
assert rows != [] and rows[0]['foo'] == 'bar'
def test_save_results():
"""
Test results save.
"""
ena_util_in = ena_util.ENAUtils('MN', False, debug=True, db_file=db_file)
ena_util_in.setup()
ena_util_in.save_results({
'id': 'test-row',
'state': ena_util_in.state,
'election': ena_util_in.election_id,
'updated': ena_util_in.timestamp(),
'contest_id': 'contest-test-id',
'choice': 'Candidate Name',
'winner': False,
'party': 'R',
'votes': 10,
'percentage': 25.0
})
rows = ena_util_in.sql.select('* FROM results')
assert rows != [] and rows[0]['id'] == 'test-row'
def test_save_contests():
"""
Test contests save.
"""
ena_util_in = ena_util.ENAUtils('MN', False, debug=True, db_file=db_file)
ena_util_in.setup()
ena_util_in.save_contests({
'id': 'test-row',
'state': ena_util_in.state,
'election': ena_util_in.election_id,
'title': 'Contest Race',
'sub_title': '',
'precincts_reporting': 10,
'total_precincts': 100,
'percent_reporting': 10.0,
'total_votes': 1111,
'seats': 1,
'updated': ena_util_in.timestamp()
})
rows = ena_util_in.sql.select('* FROM contests')
assert rows != [] and rows[0]['id'] == 'test-row'