@@ -50,27 +50,39 @@ def test_discord_user(self):
50
50
uid = '172150183260323840' ,
51
51
username = 'Henry G. Tiger' ,
52
52
discriminator = '1738' )
53
- self .assertEqual (str (u ), '%s.%s' % (u .username , u .discriminator ))
53
+ self .assertEqual (str (u ), '%s.%s' % (u .username , u .discriminator ))
54
54
55
55
56
56
class TestDiscordInvite (TestCase ):
57
57
58
58
def setUp (self ):
59
59
self .red = Group .objects .create (name = 'Red Team' )
60
60
self .blue = Group .objects .create (name = 'Blue Team' )
61
+ self .invite_json = {"code" : "aY1XeQGDKL8iCRvJ" ,
62
+ "guild" : {
63
+ "splash" : None ,
64
+ "id" : "132196927679889408" ,
65
+ "icon" : "b472e4221e5b24f4de1583be747ca1bd" ,
66
+ "name" : "Alea Iacta Est" },
67
+ "channel" : {"type" : "text" ,
68
+ "id" : "132196927679889408" ,
69
+ "name" : "general" }}
61
70
62
71
def tearDown (self ):
63
72
self .red .delete ()
64
73
self .blue .delete ()
65
74
pass
66
75
67
76
def test_discord_invite (self ):
68
- DiscordInvite .objects .create (code = 'EzqS3tytSpUCa7Lr' , active = True )
77
+ DiscordInvite .objects .create (code = 'EzqS3tytSpUCa7Lr' ,
78
+ active = True )
69
79
70
- inv = DiscordInvite .objects .create (code = '8UQcx0Ychpq1Kmqs' , active = True )
80
+ inv = DiscordInvite .objects .create (code = '8UQcx0Ychpq1Kmqs' ,
81
+ active = True )
71
82
inv .groups .add (self .blue )
72
83
73
- inv = DiscordInvite .objects .create (code = '9dbFpT89bmXzejG9' , active = False )
84
+ inv = DiscordInvite .objects .create (code = '9dbFpT89bmXzejG9' ,
85
+ active = False )
74
86
inv .groups .add (self .red )
75
87
76
88
inv = DiscordInvite .objects .create (code = 'pMqXoCQT41S7e4LK' )
@@ -79,30 +91,30 @@ def test_discord_invite(self):
79
91
80
92
with self .assertRaises (IntegrityError ):
81
93
with transaction .atomic ():
82
- DiscordInvite .objects .create (code = 'EzqS3tytSpUCa7Lr' , active = True )
94
+ DiscordInvite .objects .create (code = 'EzqS3tytSpUCa7Lr' ,
95
+ active = True )
83
96
84
- self .assertEqual (DiscordInvite .objects .filter (groups = self .blue ).count (), 2 )
85
- self .assertEqual (DiscordInvite .objects .filter (groups = self .red ).count (), 2 )
86
- self .assertEqual (DiscordInvite .objects .filter (groups__isnull = True ).count (), 1 )
97
+ self .assertEqual (DiscordInvite .objects .filter (groups = self .blue
98
+ ).count (), 2 )
99
+ self .assertEqual (DiscordInvite .objects .filter (groups = self .red
100
+ ).count (), 2 )
101
+ self .assertEqual (DiscordInvite .objects .filter (groups__isnull = True
102
+ ).count (), 1 )
87
103
88
- self .assertEqual (DiscordInvite .objects .filter (active = True ).count (), 2 )
89
- self .assertEqual (DiscordInvite .objects .filter (active = False ).count (), 2 )
104
+ self .assertEqual (DiscordInvite .objects .filter (active = True
105
+ ).count (), 2 )
106
+ self .assertEqual (DiscordInvite .objects .filter (active = False
107
+ ).count (), 2 )
90
108
91
109
@patch ('discord_bind.models.requests.get' )
92
110
def test_invite_update (self , mock_get ):
93
111
mock_get .return_value = mock_response = Mock ()
94
112
mock_response .status_code = 200
95
- mock_response .json .return_value = {"code" : "aY1XeQGDKL8iCRvJ" ,
96
- "guild" : {"splash" : None ,
97
- "id" : "132196927679889408" ,
98
- "icon" : "b472e4221e5b24f4de1583be747ca1bd" ,
99
- "name" : "Alea Iacta Est" },
100
- "channel" : {"type" : "text" ,
101
- "id" : "132196927679889408" ,
102
- "name" : "general" }}
113
+ mock_response .json .return_value = self .invite_json
103
114
obj = DiscordInvite .objects .create (code = 'aY1XeQGDKL8iCRvJ' )
104
115
obj .update_context ()
105
- mock_get .assert_called_once_with ('https://discordapp.com/api/invites/aY1XeQGDKL8iCRvJ' )
116
+ mock_get .assert_called_once_with ('https://discordapp.com/api'
117
+ '/invites/aY1XeQGDKL8iCRvJ' )
106
118
self .assertEqual (obj .guild_id , "132196927679889408" )
107
119
self .assertEqual (obj .guild_name , "Alea Iacta Est" )
108
120
self .assertEqual (obj .guild_icon , "b472e4221e5b24f4de1583be747ca1bd" )
0 commit comments