class TestWatchlist(unittest.TestCase):
class TestData:
def __init__(self, name):
self.name = name
self.verbose = True
def setUp(self):
self.recognition = Recognition()
def create_watchlists(self):
print "Creating some watchlists"
test_data = [
TestWatchlist.TestData(name='Blacklist'),
TestWatchlist.TestData(name='Whitelist'),
TestWatchlist.TestData(name='Office')
]
for this_test in test_data:
print "** Testing {0} **".format(this_test.name)
watchlist = self.recognition.create_watchlist(name=this_test.name)
print 'created watchlist {name} {watchlist_id}'.format(name=watchlist['name'], watchlist_id=watchlist['watchlist_id'])
self.assertIsNotNone(watchlist['watchlist_id'])
def list_watchlists(self):
print "List all the watchlists"
watchlists = self.recognition.list_watchlists()
self.assertIsNotNone(watchlists)
for watchlist in watchlists:
print '{0} has id {1}'.format(watchlist['name'], watchlist['watchlist_id'])
def delete_watchlists(self):
watchlists = self.recognition.list_watchlists()
for watchlist in watchlists:
print 'Going to delete {0}'.format(watchlist['watchlist_id'])
deleted_id = self.recognition.delete_watchlist(watchlist['watchlist_id'])
self.assertEqual(watchlist['watchlist_id'], deleted_id)
