def test_get_total_clicks(self):
self.re_init_models()
clicks_list = [20, 42, 33, 14, 55]
for clicks in clicks_list:
self.add_clicks(Advertiser(), clicks)
self.assertEqual(sum(clicks_list), Advertiser.get_total_clicks())
Advertiser.get_objects().clear()
def test_advertisers_total_clicks(self):
self.re_init_models()
advertiser1_clicks = [20, 10, 15, 60]
self.create_ads_from_clicks_list(Advertiser(), advertiser1_clicks)
advertiser2_clicks = [25, 5, 35, 40]
self.create_ads_from_clicks_list(Advertiser(), advertiser2_clicks)
self.assertEqual(sum(advertiser1_clicks + advertiser2_clicks),
Advertiser.get_total_clicks())
from models import Advertiser
from models.ad import Ad
if __name__ == '__main__':
advertiser1 = Advertiser('name1')
advertiser2 = Advertiser('name2')
ad1 = Ad(title='title1',
image_url='image-url1',
link='link1',
advertiser=advertiser1)
ad2 = Ad(title='title2',
image_url='image-url2',
link='link2',
advertiser=advertiser2)
ad2.describe_me()
advertiser1.describe_me()
ad1.inc_views()
ad1.inc_views()
ad1.inc_views()
ad1.inc_views()
ad2.inc_views()
ad1.inc_clicks()
ad1.inc_clicks()
ad2.inc_clicks()
print(advertiser2.get_name())
advertiser2.set_name('new name')
print(advertiser2.get_name())
print(ad1.get_clicks())
print(advertiser2.get_clicks())
print(Advertiser.get_total_clicks())
print(Advertiser.help())
