def test_update_mail_sent(self): sub = Subscriber() sub.lastname = 'toto' sub.issues_to_receive = 0 sub.save() sub = Subscriber() sub.lastname = 'tata' sub.issues_to_receive = 1 sub.save() Subscriber.update_mail_sent() toto = Subscriber.get_subscribers_from_lastname('toto')[0] tata = Subscriber.get_subscribers_from_lastname('tata')[0] self.assertEqual(1, toto.mail_sent) self.assertEqual(0, tata.mail_sent)
def test_wrong_date(self): """Test what appens when we store a wrong date in the db and try to format it""" self.sub.lastname = 'toto' self.sub.subscription_date = datetime.date(211, 7, 31) self.sub.save() sub = Subscriber.get_subscribers_from_lastname('toto')[0] actual_date = sub.subscription_date
def test_date(self): """Test the format of the date retrieved from the DB""" self.sub.lastname = 'toto' self.sub.subscription_date = datetime.date(2011, 3, 31) self.sub.save() sub = Subscriber.get_subscribers_from_lastname('toto')[0] actual_date = sub.subscription_date self.assertEqual(actual_date.strftime('%d/%m/%Y'), '31/03/2011')
def test_decrement_special_issues_to_receive(self): self.sub.lastname = 'toto' self.sub.hors_serie1 = 3 self.sub.save() self.sub = Subscriber() self.sub.lastname = 'tata' self.sub.hors_serie1 = 4 self.sub.save() self.sub = Subscriber() self.sub.lastname = 'titi' self.sub.hors_serie1 = 0 self.sub.save() Subscriber.decrement_special_issues_to_receive() sub = Subscriber.get_subscribers_from_lastname('toto')[0] self.assertEqual(sub.hors_serie1, 2) sub = Subscriber.get_subscribers_from_lastname('tata')[0] self.assertEqual(sub.hors_serie1, 3) sub = Subscriber.get_subscribers_from_lastname('titi')[0] self.assertEqual(sub.hors_serie1, 0)
def test_decrement_issues_to_receive(self): self.sub.lastname = 'toto' self.sub.issues_to_receive = 6 self.sub.save() self.sub = Subscriber() self.sub.lastname = 'tata' self.sub.issues_to_receive = 5 self.sub.save() self.sub = Subscriber() self.sub.lastname = 'titi' self.sub.issues_to_receive = 0 self.sub.save() Subscriber.decrement_issues_to_receive() sub = Subscriber.get_subscribers_from_lastname('toto')[0] self.assertEqual(sub.issues_to_receive, 5) sub = Subscriber.get_subscribers_from_lastname('tata')[0] self.assertEqual(sub.issues_to_receive, 4) sub = Subscriber.get_subscribers_from_lastname('titi')[0] self.assertEqual(sub.issues_to_receive, 0)
def test_create_and_delete_subscriber(self): sub = Subscriber() sub.lastname = 'toto' sub.save() ident = sub.identifier SubscriberAdapter.delete_from_id(ident) actual_result_count = len( Subscriber.get_subscribers_from_lastname('toto') ) self.assertEqual(0, actual_result_count)
def test_subscriber_address(self): sub = Subscriber() address = Address() sub.lastname = 'toto' address.address1 = '11 rue bilou' address.address2 = 'batiment 1' address.post_code = '66066' address.city = 'NYC' sub.address = address sub.save() retrieved_sub = Subscriber.get_subscribers_from_lastname('toto')[0] retieved_address = retrieved_sub.address self.assertEqual('NYC', retieved_address.city)
def test_subscriber_with_id(self): db_sub = Subscriber() db_sub.lastname = 'toto' db_sub.firstname = 'tata' db_sub.save() ident = db_sub.identifier sub = { 'subscriber_id': ident, 'lastname': 'toto', 'firstname': 'titi' } actual = self._save(sub) self.assertEquals(ident, actual['subscriber_id']) subs_list = Subscriber.get_subscribers_from_lastname('toto') self.assertEqual(len(subs_list), 1) retrieved_sub = subs_list[0] self.assertEquals(ident, retrieved_sub.identifier)
def _retrieve_one_from_lastname(self, lastname): """Retrieve the first sub in db having provided lastname""" subs_list = Subscriber.get_subscribers_from_lastname(lastname) return subs_list[0]