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)