1from flask_testing import TestCase 

2from webapp.app import create_app 

3from webapp.handlers import badge_counter, badge_logged_in_counter 

4 

5 

6class TestsBadgePrometheusCounter(TestCase): 

7 def setUp(self): 

8 self.endpoint_url = "/static/images/badges/en/snap-store-black.svg" 

9 badge_counter._value.set(0.0) 

10 badge_logged_in_counter._value.set(0.0) 

11 

12 def create_app(self): 

13 app = create_app(testing=True) 

14 app.secret_key = "secret_key" 

15 app.config["WTF_CSRF_METHODS"] = [] 

16 

17 return app 

18 

19 def tests_increment_counter(self): 

20 self.client.get(self.endpoint_url, buffered=True) 

21 assert badge_counter._value.get() == 1.0 

22 

23 def tests_no_increment(self): 

24 with self.client.session_transaction() as s: 

25 s["session"] = "content" 

26 

27 self.client.get(self.endpoint_url, buffered=True) 

28 assert badge_counter._value.get() == 0.0 

29 

30 def tests_increment_counter_logged_in(self): 

31 self.client.get(self.endpoint_url, buffered=True) 

32 assert badge_logged_in_counter._value.get() == 0.0 

33 

34 def tests_no_increment_logged_in(self): 

35 with self.client.session_transaction() as s: 

36 s["session"] = "content" 

37 

38 self.client.get(self.endpoint_url, buffered=True) 

39 assert badge_logged_in_counter._value.get() == 1.0