Coverage for polls/tests/test_models.py: 100.00%

1import datetime (empty)

2 

3from django.contrib.auth import get_user_model (empty)

4from django.test import TestCase (empty)

5from django.utils import timezone (empty)

6 

7from ..models import Choice, Question (empty)

8 

9 

10def create_question(question_text, days): (empty)

11 """ 

12 Create a question with the given 'question_text' and published 

13 the given number of 'days' offset to now (negative for questions 

14 published in the past, positive for questions that have yet to 

15 be published. 

16 """ 

17 time = timezone.now() + datetime.timedelta(days=days) (empty)

18 return Question.objects.create(question_text=question_text, pub_date=time) (empty)

19 

20 

21def create_choice(question, choice_text, votes): (empty)

22 return Choice.objects.create( (empty)

23 question=question, choice_text=choice_text, votes=votes 

24 ) 

25 

26 

27class ChoiceTests(TestCase): (empty)

28 def setUp(self): (empty)

29 self.question = create_question(question_text="Current question.", days=0) (empty)

30 self.choice = create_choice( (empty)

31 question=self.question, choice_text="Seven logs", votes=3 

32 ) 

33 

34 def test___str__(self): (empty)

35 assert self.choice.__str__() == self.choice.choice_text 1 ctx1e

36 assert str(self.choice) == self.choice.choice_text 1 ctx1e

37 

38 

39class QuestionTests(TestCase): (empty)

40 def setUp(self): (empty)

41 self.user = get_user_model().objects.create_user( (empty)

42 username="johndoe", 

43 email="johndoe@example.com", 

44 password="secret", 

45 ) 

46 self.question = create_question(question_text="Current question.", days=0) (empty)

47 

48 """ (empty)

49 self.question = Question.objects.create( 

50 question_text="How much wood can a woodchuck chuck?", 

51 pub_date=timezone.now(), 

52 ) 

53 """ 

54 

55 def test___str__(self): (empty)

56 assert self.question.__str__() == self.question.question_text 1 ctx1f

57 assert str(self.question) == self.question.question_text 1 ctx1f

58 

59 def test_was_published_recently_with_old_question(self): (empty)

60 """ 

61 was_published_recently() returns False for 

62 questions whose pub_date is older than 1 day. 

63 """ 

64 time = timezone.now() - datetime.timedelta(days=1, seconds=1) 1 ctx1b

65 old_question = Question(pub_date=time) 1 ctx1b

66 self.assertIs(old_question.was_published_recently(), False) 1 ctx1b

67 

68 def test_was_published_recently_with_recent_question(self): (empty)

69 """ 

70 was_published_recently() returns True for 

71 questions whose pub_date is within the last day. 

72 """ 

73 time = timezone.now() - datetime.timedelta(hours=23, minutes=59, seconds=59) 1 ctx1c

74 recent_question = Question(pub_date=time) 1 ctx1c

75 self.assertIs(recent_question.was_published_recently(), True) 1 ctx1c

76 

77 def test_was_published_recently_with_future_question(self): (empty)

78 """ 

79 was_published_recently() returns False for 

80 questions whose pub_date is in the future. 

81 """ 

82 time = timezone.now() + datetime.timedelta(days=30) 1 ctx1d

83 future_question = Question(pub_date=time) 1 ctx1d

84 

85 self.assertIs(future_question.was_published_recently(), False) 1 ctx1d