def test_close(self): conn = HTTPSConnection('testhost', 8025, {'var': 'val'}, True, 7) conn.sock = self.mox.CreateMockAnything() conn.sock.unwrap() conn.sock.close() self.mox.ReplayAll() conn.close()
def test_close(self): if six.PY2: conn = HTTPSConnection("testhost", 8025, {"var": "val"}, True, 7) else: conn = HTTPSConnection("testhost", 8025, {"var": "val"}, 7) conn.sock = self.mox.CreateMockAnything() conn.sock.unwrap() conn.sock.close() self.mox.ReplayAll() conn.close()
def test_connect(self): self.mox.StubOutWithMock(socket, 'create_connection') self.mox.StubOutWithMock(ssl, 'SSLSocket') sslsock = self.mox.CreateMockAnything() socket.create_connection(('testhost', 8025), 7).AndReturn(9) ssl.SSLSocket(9, var='val').AndReturn(sslsock) sslsock.do_handshake() self.mox.ReplayAll() conn = HTTPSConnection('testhost', 8025, {'var': 'val'}, True, 7) conn.connect() self.assertEqual(sslsock, conn.sock)
def test_connect(self): self.mox.StubOutWithMock(socket, 'create_connection') self.mox.StubOutWithMock(ssl, 'SSLSocket') sslsock = self.mox.CreateMockAnything() socket.create_connection(('testhost', 8025), 7).AndReturn(9) ssl.SSLSocket(9, var='val').AndReturn(sslsock) sslsock.do_handshake() self.mox.ReplayAll() conn = HTTPSConnection('testhost', 8025, {'var': 'val'}, True, 7) conn.connect() assert_equal(sslsock, conn.sock)
def test_connect(self): self.mox.StubOutWithMock(socket, "create_connection") self.mox.StubOutWithMock(ssl, "SSLSocket") sslsock = self.mox.CreateMockAnything() socket.create_connection(("testhost", 8025), 7).AndReturn(9) ssl.SSLSocket(9, var="val").AndReturn(sslsock) sslsock.do_handshake() self.mox.ReplayAll() # strict HTTPConnection became deprecated if six.PY2: conn = HTTPSConnection("testhost", 8025, {"var": "val"}, True, 7) else: conn = HTTPSConnection("testhost", 8025, {"var": "val"}, 7) conn.connect() self.assertEqual(sslsock, conn.sock)