Coverage for ivatar/test_auxiliary.py: 100%

1""" 

2Test various other parts of ivatar/libravatar in order 

3to increase the overall test coverage. Test in here, didn't 

4fit anywhere else. 

5""" 

6 

7from django.test import TestCase 

8from django.contrib.auth.models import User 

9 

10from ivatar.utils import random_string 

11from ivatar.ivataraccount.models import pil_format, UserPreference 

12 

13 

14class Tester(TestCase): 

15 """ 

16 Main test class 

17 """ 

18 

19 user = None 

20 username = random_string() 

21 

22 def setUp(self): 

23 """ 

24 Prepare tests. 

25 - Create user 

26 """ 

27 self.user = User.objects.create_user( 

28 username=self.username, 

29 ) 

30 

31 def test_pil_format(self): 

32 """ 

33 Test pil format function 

34 """ 

35 self.assertEqual(pil_format("jpg"), "JPEG") 

36 self.assertEqual(pil_format("jpeg"), "JPEG") 

37 self.assertEqual(pil_format("png"), "PNG") 

38 self.assertEqual(pil_format("gif"), "GIF") 

39 self.assertEqual(pil_format("webp"), "WEBP") 

40 self.assertEqual(pil_format("abc"), None) 

41 

42 def test_userprefs_str(self): 

43 """ 

44 Test if str representation of UserPreferences is as expected 

45 """ 

46 up = UserPreference(theme="default", user=self.user) 

47 print(up)