def test_dont_save_tickets_twice(self):
client = Vivo(TEST_DIR)
client._parse()
client._save_tickets()
client._save_tickets()
wallet = coopy.base.init_persistent_system(Wallet(), basedir=TEST_DIR)
count = wallet.count_tickets()
self.assertEqual(10, count)
def test_return_avalaible_tickets_at_create(self):
client = Vivo(TEST_DIR)
client._parse()
availables = client._save_tickets()
self.assertEqual(1, len(availables))
self.assertEqual('QUERO MATAR MEU CHEFE 2', availables[0].name)
availables = client._save_tickets()
self.assertEqual(0, len(availables))
def test_cancel_in_avaliability(self):
client = Vivo()
client._parse()
expected_list = [
{'date': '22/02/2015', 'name': 'NOITE INFELIZ - A COMEDIA MUSICAL DAS MALDADES', 'avaliabilty': 'CA', 'id': "ede5d5105c2098e56ef88fc0987765a4", 'link': 'detalharEvento.action?caMktEvtCod=PCT20672&k=ede5d5105c2098e56ef88fc0987765a4'},
]
for expected_item in expected_list:
self.assertIn(expected_item, client.tickets)
def test_save_tickets(self):
client = Vivo(TEST_DIR)
client._parse()
client._save_tickets()
wallet = coopy.base.init_persistent_system(Wallet(), basedir=TEST_DIR)
ticket = wallet.get_ticket("7efe1e063a6bfbaa5c865f31d0a57e46")
self.assertEqual('QUERO MATAR MEU CHEFE 2', ticket.name)
self.assertEqual('BK', ticket.avaliabilty)
self.assertEqual('7efe1e063a6bfbaa5c865f31d0a57e46', ticket.id)
self.assertEqual('03/12/2014', ticket.date)
def test_parse_html(self):
client = Vivo()
client._parse()
expected_list = [
{'date': '02/12/2014', 'name': 'OS CARAS DE PAUS', 'avaliabilty': 'CL', 'id': "6146707aa13ca3946e60757349c0a9d6", 'link': 'detalharEvento.action?caMktEvtCod=PRE20018&k=6146707aa13ca3946e60757349c0a9d6'},
{'date': '03/12/2014', 'name': 'QUERO MATAR MEU CHEFE 2', 'avaliabilty': 'BK', 'id': "7efe1e063a6bfbaa5c865f31d0a57e46", 'link': 'detalharEvento.action?caMktEvtCod=PRE20002&k=7efe1e063a6bfbaa5c865f31d0a57e46'},
{'date': '04/12/2014', 'name': u'CHUVA CONSTANTE - A FELICIDADE \xc9 FORA DA LEI', 'avaliabilty': 'SO', 'id': "49d33f9135849ba4db71589662dcf1d3", 'link': 'detalharEvento.action?caMktEvtCod=PCT20032&k=49d33f9135849ba4db71589662dcf1d3'},
]
for expected_item in expected_list:
self.assertIn(expected_item, client.tickets)
