@@ -79,109 +79,110 @@ def test_server_accept(self):
7979
8080 def test_missing_data (self ):
8181 self .write_line (b'' )
82- self .assertEqual (self .channel .last ,
82+ self .assertEqual (self .channel .socket . last ,
8383 b'500 Error: bad syntax\r \n ' )
8484
8585 def test_EHLO_not_implemented (self ):
8686 self .write_line (b'EHLO test.example' )
87- self .assertEqual (self .channel .last ,
87+ self .assertEqual (self .channel .socket . last ,
8888 b'502 Error: command "EHLO" not implemented\r \n ' )
8989
9090 def test_HELO (self ):
9191 name = socket .getfqdn ()
9292 self .write_line (b'HELO test.example' )
93- self .assertEqual (self .channel .last ,
93+ self .assertEqual (self .channel .socket . last ,
9494 '250 {}\r \n ' .format (name ).encode ('ascii' ))
9595
9696 def test_HELO_bad_syntax (self ):
9797 self .write_line (b'HELO' )
98- self .assertEqual (self .channel .last ,
98+ self .assertEqual (self .channel .socket . last ,
9999 b'501 Syntax: HELO hostname\r \n ' )
100100
101101 def test_HELO_duplicate (self ):
102102 self .write_line (b'HELO test.example' )
103103 self .write_line (b'HELO test.example' )
104- self .assertEqual (self .channel .last ,
104+ self .assertEqual (self .channel .socket . last ,
105105 b'503 Duplicate HELO/EHLO\r \n ' )
106106
107107 def test_NOOP (self ):
108108 self .write_line (b'NOOP' )
109- self .assertEqual (self .channel .last , b'250 Ok\r \n ' )
109+ self .assertEqual (self .channel .socket . last , b'250 Ok\r \n ' )
110110
111111 def test_NOOP_bad_syntax (self ):
112112 self .write_line (b'NOOP hi' )
113- self .assertEqual (self .channel .last ,
113+ self .assertEqual (self .channel .socket . last ,
114114 b'501 Syntax: NOOP\r \n ' )
115115
116116 def test_QUIT (self ):
117117 self .write_line (b'QUIT' )
118- self .assertEqual (self .channel .last , b'221 Bye\r \n ' )
118+ self .assertEqual (self .channel .socket . last , b'221 Bye\r \n ' )
119119
120120 def test_QUIT_arg_ignored (self ):
121121 self .write_line (b'QUIT bye bye' )
122- self .assertEqual (self .channel .last , b'221 Bye\r \n ' )
122+ self .assertEqual (self .channel .socket . last , b'221 Bye\r \n ' )
123123
124124 def test_bad_state (self ):
125- self .channel ._SMTPChannel__state = 'BAD STATE'
125+ self .channel .smtp_state = 'BAD STATE'
126126 self .write_line (b'HELO' )
127- self .assertEqual (self .channel .last , b'451 Internal confusion\r \n ' )
127+ self .assertEqual (self .channel .socket .last ,
128+ b'451 Internal confusion\r \n ' )
128129
129130 def test_need_MAIL (self ):
130131 self .write_line (b'RCPT to:spam@example' )
131- self .assertEqual (self .channel .last ,
132+ self .assertEqual (self .channel .socket . last ,
132133 b'503 Error: need MAIL command\r \n ' )
133134
134135 def test_MAIL_syntax (self ):
135136 self .write_line (b'MAIL from eggs@example' )
136- self .assertEqual (self .channel .last ,
137+ self .assertEqual (self .channel .socket . last ,
137138 b'501 Syntax: MAIL FROM:<address>\r \n ' )
138139
139140 def test_MAIL_missing_from (self ):
140141 self .write_line (b'MAIL from:' )
141- self .assertEqual (self .channel .last ,
142+ self .assertEqual (self .channel .socket . last ,
142143 b'501 Syntax: MAIL FROM:<address>\r \n ' )
143144
144145 def test_MAIL_chevrons (self ):
145146 self .write_line (b'MAIL from:<eggs@example>' )
146- self .assertEqual (self .channel .last , b'250 Ok\r \n ' )
147+ self .assertEqual (self .channel .socket . last , b'250 Ok\r \n ' )
147148
148149 def test_nested_MAIL (self ):
149150 self .write_line (b'MAIL from:eggs@example' )
150151 self .write_line (b'MAIL from:spam@example' )
151- self .assertEqual (self .channel .last ,
152+ self .assertEqual (self .channel .socket . last ,
152153 b'503 Error: nested MAIL command\r \n ' )
153154
154155 def test_need_RCPT (self ):
155156 self .write_line (b'MAIL From:eggs@example' )
156157 self .write_line (b'DATA' )
157- self .assertEqual (self .channel .last ,
158+ self .assertEqual (self .channel .socket . last ,
158159 b'503 Error: need RCPT command\r \n ' )
159160
160161 def test_RCPT_syntax (self ):
161162 self .write_line (b'MAIL From:eggs@example' )
162163 self .write_line (b'RCPT to eggs@example' )
163- self .assertEqual (self .channel .last ,
164+ self .assertEqual (self .channel .socket . last ,
164165 b'501 Syntax: RCPT TO: <address>\r \n ' )
165166
166167 def test_data_dialog (self ):
167168 self .write_line (b'MAIL From:eggs@example' )
168- self .assertEqual (self .channel .last , b'250 Ok\r \n ' )
169+ self .assertEqual (self .channel .socket . last , b'250 Ok\r \n ' )
169170 self .write_line (b'RCPT To:spam@example' )
170- self .assertEqual (self .channel .last , b'250 Ok\r \n ' )
171+ self .assertEqual (self .channel .socket . last , b'250 Ok\r \n ' )
171172
172173 self .write_line (b'DATA' )
173- self .assertEqual (self .channel .last ,
174+ self .assertEqual (self .channel .socket . last ,
174175 b'354 End data with <CR><LF>.<CR><LF>\r \n ' )
175176 self .write_line (b'data\r \n more\r \n .' )
176- self .assertEqual (self .channel .last , b'250 Ok\r \n ' )
177+ self .assertEqual (self .channel .socket . last , b'250 Ok\r \n ' )
177178 self .assertEqual (self .server .messages [- 1 ],
178179 ('peer' , 'eggs@example' , ['spam@example' ], 'data\n more' ))
179180
180181 def test_DATA_syntax (self ):
181182 self .write_line (b'MAIL From:eggs@example' )
182183 self .write_line (b'RCPT To:spam@example' )
183184 self .write_line (b'DATA spam' )
184- self .assertEqual (self .channel .last , b'501 Syntax: DATA\r \n ' )
185+ self .assertEqual (self .channel .socket . last , b'501 Syntax: DATA\r \n ' )
185186
186187 def test_multiple_RCPT (self ):
187188 self .write_line (b'MAIL From:eggs@example' )
@@ -197,13 +198,13 @@ def test_manual_status(self):
197198 self .write_line (b'RCPT To:spam@example' )
198199 self .write_line (b'DATA' )
199200 self .write_line (b'return status\r \n .' )
200- self .assertEqual (self .channel .last , b'250 Okish\r \n ' )
201+ self .assertEqual (self .channel .socket . last , b'250 Okish\r \n ' )
201202
202203 def test_RSET (self ):
203204 self .write_line (b'MAIL From:eggs@example' )
204205 self .write_line (b'RCPT To:spam@example' )
205206 self .write_line (b'RSET' )
206- self .assertEqual (self .channel .last , b'250 Ok\r \n ' )
207+ self .assertEqual (self .channel .socket . last , b'250 Ok\r \n ' )
207208 self .write_line (b'MAIL From:foo@example' )
208209 self .write_line (b'RCPT To:eggs@example' )
209210 self .write_line (b'DATA' )
@@ -213,7 +214,7 @@ def test_RSET(self):
213214
214215 def test_RSET_syntax (self ):
215216 self .write_line (b'RSET hi' )
216- self .assertEqual (self .channel .last , b'501 Syntax: RSET\r \n ' )
217+ self .assertEqual (self .channel .socket . last , b'501 Syntax: RSET\r \n ' )
217218
218219
219220def test_main ():
0 commit comments