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]
