def test_get_connection(self):
# As a full test would require a working AWS secret key,
# only rudimentary tests are performed on this function
with self.settings(DJURK=None, DJURK_CONFIG_FILE=None):
self.assertRaises(InvalidDjurkSettings, get_connection)
with self.settings(DJURK={'aws_access_key_id': '123'}):
self.assertRaises(KeyError, get_connection)
with self.settings(DJURK={'aws_secret_access_key': '123'}):
self.assertRaises(KeyError, get_connection)
with self.settings(DJURK={'aws_access_key_id': '123',
'aws_secret_access_key': '456'}):
mtc = get_connection()
self.assertTrue(isinstance(get_connection(),
boto.mturk.connection.MTurkConnection))
with self.settings(DJURK=None,
DJURK_CONFIG_FILE=self.djurk_config_filename):
f = open(self.djurk_config_filename, 'w')
f.write("[Connection]\naws_access_key_id: 123\n"
"aws_secret_access_key: 456\n")
f.close()
self.assertTrue(isinstance(get_connection(),
boto.mturk.connection.MTurkConnection))
def test_get_connection(self):
# As a full test would require a working AWS secret key,
# only rudimentary tests are performed on this function
with self.settings(DJURK=None, DJURK_CONFIG_FILE=None):
self.assertRaises(InvalidDjurkSettings, get_connection)
with self.settings(DJURK={'aws_access_key_id': '123'}):
self.assertRaises(KeyError, get_connection)
with self.settings(DJURK={'aws_secret_access_key': '123'}):
self.assertRaises(KeyError, get_connection)
with self.settings(DJURK={
'aws_access_key_id': '123',
'aws_secret_access_key': '456'
}):
mtc = get_connection()
self.assertTrue(
isinstance(get_connection(),
boto.mturk.connection.MTurkConnection))
with self.settings(DJURK=None,
DJURK_CONFIG_FILE=self.djurk_config_filename):
f = open(self.djurk_config_filename, 'w')
f.write("[Connection]\naws_access_key_id: 123\n"
"aws_secret_access_key: 456\n")
f.close()
self.assertTrue(
isinstance(get_connection(),
boto.mturk.connection.MTurkConnection))
def handle(self, *args, **options):
self.mtc = get_connection()
if options['account_balance']:
self.check_account_balance()
if options['sample_hit']:
def update_reviewable_hits(do_update_assignments=False):
"""Get only reviewable HITS from Amazon"""
connection = get_connection()
_update_hits(connection.get_reviewable_hits(),
do_update_assignments=do_update_assignments)
def update_all_hits(do_update_assignments=False):
"""Get All HITS from Amazon"""
connection = get_connection()
_update_hits(connection.get_all_hits(),
do_update_assignments=do_update_assignments)
def update_reviewable_hits(do_update_assignments=False):
"""Get only reviewable HITS from Amazon"""
connection = get_connection()
_update_hits(connection.get_reviewable_hits(),
do_update_assignments=do_update_assignments)
def update_all_hits(do_update_assignments=False):
"""Get All HITS from Amazon"""
connection = get_connection()
_update_hits(connection.get_all_hits(),
do_update_assignments=do_update_assignments)
