17
17
import socket
18
18
import time
19
19
import unittest
20
- import urllib .request
21
20
import uuid
22
21
23
22
from kubernetes .client import api_client
26
25
from kubernetes .stream import stream , portforward
27
26
from kubernetes .stream .ws_client import ERROR_CHANNEL
28
27
28
+ import six .moves .urllib .request as urllib_request
29
29
30
30
def short_uuid ():
31
31
id = str (uuid .uuid4 ())
@@ -228,7 +228,7 @@ def test_portforward_raw(self):
228
228
self .assertTrue (pf .connected )
229
229
230
230
sock = pf .socket (1236 )
231
- self .assertRaises (BrokenPipeError , sock .sendall , b'This should fail...' )
231
+ self .assertRaises (socket . error , sock .sendall , b'This should fail...' )
232
232
self .assertIsNotNone (pf .error (1236 ))
233
233
sock .close ()
234
234
@@ -246,6 +246,7 @@ def test_portforward_raw(self):
246
246
reply += data
247
247
self .assertEqual (reply , sent )
248
248
sock .close ()
249
+ time .sleep (1 )
249
250
self .assertFalse (pf .connected )
250
251
self .assertIsNone (pf .error (1234 ))
251
252
self .assertIsNone (pf .error (1235 ))
@@ -300,12 +301,12 @@ def kubernetes_create_connection(address, *args, **kwargs):
300
301
socket_create_connection = socket .create_connection
301
302
try :
302
303
socket .create_connection = kubernetes_create_connection
303
- response = urllib . request .urlopen ('http://%s.default.kubernetes/' % name )
304
+ response = urllib_request .urlopen ('http://%s.default.kubernetes/' % name )
304
305
html = response .read ().decode ('utf-8' )
305
306
finally :
306
307
socket .create_connection = socket_create_connection
307
308
308
- self .assertEqual (response .status , 200 )
309
+ self .assertEqual (response .code , 200 )
309
310
self .assertTrue ('<h1>Welcome to nginx!</h1>' in html )
310
311
311
312
resp = api .delete_namespaced_pod (name = name , body = {},
0 commit comments