def setUp(self):
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
# Create a proposal with token 'proposaltoken'.
db_loader.create_proposal()
def test_proposal_list_method(self):
''' Add a proposal to the database then query. '''
db_loader.create_proposal()
response = generic_service_method_success_test(self, 'proposal', 'list')
self.assertEqual(response['response']['type'], 'Proposals',
'System proposal list service method failed.')
self.assertEqual(len(response['response']['content']['proposals']), 1,
'Failed to return the correct number of proposals.')
def test_proposal_get_method(self):
''' Add one private and one public proposal to the database then query. '''
proposal_key = db_loader.create_proposal()
key = proposal_key.urlsafe()
response = generic_service_method_success_test(self, 'proposal', 'get', params={'key':encrypt(key)})
self.assertEqual(response['response']['type'], 'Proposal',
'Proposal get service method failed.')
self.assertEqual(response['response']['content']['key'], key,
'Proposal get method returned the wrong proposal.')
