def form():
if request.method == 'POST':
myCreditManager = FasterCreditManager()
inputs = []
input_type = CREDIT_TYPE
#probably want some proper form validation later
for i in range(1, 5):
key = 'input'
key += `i`
#loop through all the 5 fields
if key in request.form and request.form[key] is not u'':
inputs.append(request.form[key])
if "tag:" in request.form[key]:
input_type = GUID_TYPE
if input_type == CREDIT_TYPE:
#if input is credit type, then return the guids
return_data = myCreditManager.return_guids(inputs)
else:
#else if input are guids, then return the credits
return_data = myCreditManager.return_credits(inputs)
return render_template('form_submit.html', error='', return_data=return_data)
else:
return render_template('form_submit.html')
def test_local_data_one_credit(self):
if self.local_xml_file:
self.creditManager = FasterCreditManager(rss_feed_url=self.local_xml_file)
guids = self.creditManager.return_guids(['id_1Some Name1'])
self.assertTrue('id_1' in guids)
def setUp(self):
self.creditManager = FasterCreditManager()
#this local xml file path must be absolute
self.local_xml_file = False #'file:///Users/wenzou/PycharmProjects/developer_test/test_data/testdata5000.xml'
class TestFastCreditManager(unittest.TestCase):
def setUp(self):
self.creditManager = FasterCreditManager()
#this local xml file path must be absolute
self.local_xml_file = False #'file:///Users/wenzou/PycharmProjects/developer_test/test_data/testdata5000.xml'
def test_local_data_one_guid(self):
if self.local_xml_file:
self.creditManager = FasterCreditManager(rss_feed_url=self.local_xml_file)
credits = self.creditManager.return_credits(['id_1'])
self.assertTrue('id_1Some Name1' in credits)
def test_local_data_one_credit(self):
if self.local_xml_file:
self.creditManager = FasterCreditManager(rss_feed_url=self.local_xml_file)
guids = self.creditManager.return_guids(['id_1Some Name1'])
self.assertTrue('id_1' in guids)
def test_no_guid(self):
credits = self.creditManager.return_credits([])
self.assertEqual(credits, [])
def test_bad_guid(self):
credits = self.creditManager.return_credits(['XYDI'])
self.assertEqual(credits, [])
def test_one_guid(self):
credits = self.creditManager.return_credits(['tag:wiredrive,2011-03-18:token:128b053b916ea1f7f20233e8a26bc45d:21239'])
self.assertTrue('MTV Choose or Lose' in credits)
self.assertTrue('Bryan Buckely' in credits)
self.assertTrue('J Walter Thompson / New York' in credits)
self.assertTrue('Hungryman' in credits)
self.assertTrue('Cannes Gold Lion 2009' in credits)
self.assertTrue(len(credits), 5)
def test_multiple_guid(self):
credits = self.creditManager.return_credits(['tag:wiredrive,2011-03-18:token:128b053b916ea1f7f20233e8a26bc45d:21239',
'tag:wiredrive,2011-03-18:token:128b053b916ea1f7f20233e8a26bc45d:21248'])
self.assertTrue('MTV Choose or Lose' in credits)
self.assertTrue('Bryan Buckely' in credits)
self.assertTrue('J Walter Thompson / New York' in credits)
self.assertTrue('Hungryman' in credits)
self.assertTrue('Cannes Gold Lion 2009' in credits)
self.assertTrue('Pizza Pops' in credits)
self.assertTrue('Scott Corbett' in credits)
self.assertTrue('Holiday Films' in credits)
self.assertTrue('Cannes Silver Lion 2009' in credits)
self.assertTrue(len(credits), 9)
def test_no_credit(self):
guids = self.creditManager.return_guids([])
self.assertEqual(guids, [])
def test_bad_credit(self):
guids = self.creditManager.return_guids(['XYDI'])
self.assertEqual(guids, [])
def test_one_credit(self):
guids = self.creditManager.return_guids(['Pizza Pops'])
self.assertEqual(guids, ['tag:wiredrive,2011-03-18:token:128b053b916ea1f7f20233e8a26bc45d:21248'])
def test_two_credits(self):
guids = self.creditManager.return_guids(['Pizza Pops', 'Scott Corbett'])
self.assertEqual(guids, ['tag:wiredrive,2011-03-18:token:128b053b916ea1f7f20233e8a26bc45d:21248'])
def test_three_credits(self):
guids = self.creditManager.return_guids(['Washington Lottery', 'Jerry Brown', 'Sticks & Stones'])
self.assertEqual(guids, ['tag:wiredrive,2011-03-18:token:128b053b916ea1f7f20233e8a26bc45d:21253'])
