Coverage for src/tests/core/mail/test_recipients.py: 100%

1"""Module for testing Recipients class.""" 

2 

3from kwai_core.domain.value_objects.email_address import EmailAddress 

4from kwai_core.mail.recipient import Recipient, Recipients 

5 

6 

7def test_recipients_add_cc(): 

8 """Test adding cc.""" 

9 recipients = Recipients( 

10 Recipient(email=EmailAddress("jigoro.kano@kwai.com"), name="Jigoro Kano") 

11 ) 

12 recipients = recipients.add_cc( 

13 Recipient(email=EmailAddress("kyuzo.mifune@kwai.com"), name="Kyuzo Mifune") 

14 ).add_cc( 

15 Recipient( 

16 email=EmailAddress("kunisaburo.iizuka@kwai.com"), name="Kunisaburo Iizuka" 

17 ) 

18 ) 

19 

20 assert len(recipients.cc) == 2, "There should be 2 cc recipients." 

21 

22 

23def test_recipients_with_cc(): 

24 """Test with cc.""" 

25 recipients = Recipients( 

26 Recipient(email=EmailAddress("jigoro.kano@kwai.com"), name="Jigoro Kano") 

27 ) 

28 recipients = recipients.with_cc( 

29 Recipient(email=EmailAddress("kyuzo.mifune@kwai.com"), name="Kyuzo Mifune"), 

30 Recipient( 

31 email=EmailAddress("kunisaburo.iizuka@kwai.com"), name="Kunisaburo Iizuka" 

32 ), 

33 ) 

34 

35 assert len(recipients.cc) == 2, "There should be 2 cc recipients." 

36 

37 

38def test_recipients_add_to(): 

39 """Test adding receivers.""" 

40 recipients = Recipients( 

41 Recipient(email=EmailAddress("jigoro.kano@kwai.com"), name="Jigoro Kano") 

42 ) 

43 recipients = recipients.add_to( 

44 Recipient(email=EmailAddress("kyuzo.mifune@kwai.com"), name="Kyuzo Mifune") 

45 ).add_to( 

46 Recipient( 

47 email=EmailAddress("kunisaburo.iizuka@kwai.com"), name="Kunisaburo Iizuka" 

48 ) 

49 ) 

50 

51 assert len(recipients.to) == 2, "There should be 2 to recipients." 

52 

53 

54def test_recipients_with_to(): 

55 """Test setting a receiver.""" 

56 recipients = Recipients( 

57 Recipient(email=EmailAddress("jigoro.kano@kwai.com"), name="Jigoro Kano") 

58 ) 

59 recipients = recipients.with_to( 

60 Recipient(email=EmailAddress("kyuzo.mifune@kwai.com"), name="Kyuzo Mifune"), 

61 Recipient( 

62 email=EmailAddress("kunisaburo.iizuka@kwai.com"), name="Kunisaburo Iizuka" 

63 ), 

64 ) 

65 

66 assert len(recipients.to) == 2, "There should be 2 to recipients." 

67 

68 

69def test_recipients_add_bcc(): 

70 """Test adding bcc.""" 

71 recipients = Recipients( 

72 Recipient(email=EmailAddress("jigoro.kano@kwai.com"), name="Jigoro Kano") 

73 ) 

74 recipients = recipients.add_bcc( 

75 Recipient(email=EmailAddress("kyuzo.mifune@kwai.com"), name="Kyuzo Mifune") 

76 ).add_bcc( 

77 Recipient( 

78 email=EmailAddress("kunisaburo.iizuka@kwai.com"), name="Kunisaburo Iizuka" 

79 ) 

80 ) 

81 

82 assert len(recipients.bcc) == 2, "There should be 2 bcc recipients." 

83 

84 

85def test_recipients_with_bcc(): 

86 """Test setting bcc.""" 

87 recipients = Recipients( 

88 Recipient(email=EmailAddress("jigoro.kano@kwai.com"), name="Jigoro Kano") 

89 ) 

90 recipients = recipients.with_bcc( 

91 Recipient(email=EmailAddress("kyuzo.mifune@kwai.com"), name="Kyuzo Mifune"), 

92 Recipient( 

93 email=EmailAddress("kunisaburo.iizuka@kwai.com"), name="Kunisaburo Iizuka" 

94 ), 

95 ) 

96 

97 assert len(recipients.bcc) == 2, "There should be 2 cc recipients."