python3: fix cookie handler in test

* tries both cases for the set-cookie header to make sure we catch cookie transitions. * tampers with basic_connection fixture to demonstrate closes: #93
vmware · Jul 29, 2014 · 124a8ac · 124a8ac
1 parent abbb274
commit 124a8ac
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 6 deletions.
diff --git a/pyVmomi/SoapAdapter.py b/pyVmomi/SoapAdapter.py
@@ -1241,7 +1241,10 @@ def InvokeMethod(self, mo, info, args, outerStub=None):
          # The server is probably sick, drop all of the cached connections.
          self.DropConnections()
          raise
-      cookie = resp.getheader('set-cookie')
+      cookie = resp.getheader('Set-Cookie')
+      if cookie is None:
+          # try lower-case header for backwards compat. with old vSphere
+          cookie = resp.getheader('set-cookie')
       status = resp.status
 
       if cookie:

diff --git a/tests/fixtures/basic_connection.yaml b/tests/fixtures/basic_connection.yaml
@@ -52,7 +52,7 @@ interactions:
       content-length: ['3332']
       content-type: [text/xml; charset=utf-8]
       date: ['Mon, 21 Jul 2014 22:31:05 GMT']
-      set-cookie: [vmware_soap_session="52970dd3-2b0f-647b-22b3-44bda6d49983"; Path=/;
+      set-cookie: [vmware_soap_session="52773cd3-35c6-b40a-17f1-fe664a9f08f3"; Path=/;
           HttpOnly; Secure;]
     status: {code: 200, message: OK}
 - request:
@@ -68,7 +68,7 @@ interactions:
     headers:
       Accept-Encoding: ['gzip, deflate']
       Content-Type: [text/xml; charset=UTF-8]
-      Cookie: [vmware_soap_session="52970dd3-2b0f-647b-22b3-44bda6d49983"; Path=/;
+      Cookie: [vmware_soap_session="52773cd3-35c6-b40a-17f1-fe664a9f08f3"; Path=/;
           HttpOnly; Secure;]
       SOAPAction: ['"urn:vim25/4.1"']
     method: POST
@@ -99,7 +99,7 @@ interactions:
     headers:
       Accept-Encoding: ['gzip, deflate']
       Content-Type: [text/xml; charset=UTF-8]
-      Cookie: [vmware_soap_session="52970dd3-2b0f-647b-22b3-44bda6d49983"; Path=/;
+      Cookie: [vmware_soap_session="52773cd3-35c6-b40a-17f1-fe664a9f08f3"; Path=/;
           HttpOnly; Secure;]
       SOAPAction: ['"urn:vim25/4.1"']
     method: POST
@@ -155,7 +155,7 @@ interactions:
     headers:
       Accept-Encoding: ['gzip, deflate']
       Content-Type: [text/xml; charset=UTF-8]
-      Cookie: [vmware_soap_session="52970dd3-2b0f-647b-22b3-44bda6d49983"; Path=/;
+      Cookie: [vmware_soap_session="52773cd3-35c6-b40a-17f1-fe664a9f08f3"; Path=/;
           HttpOnly; Secure;]
       SOAPAction: ['"urn:vim25/4.1"']
     method: POST
@@ -212,7 +212,7 @@ interactions:
     headers:
       Accept-Encoding: ['gzip, deflate']
       Content-Type: [text/xml; charset=UTF-8]
-      Cookie: [vmware_soap_session="52970dd3-2b0f-647b-22b3-44bda6d49983"; Path=/;
+      Cookie: [vmware_soap_session="52773cd3-35c6-b40a-17f1-fe664a9f08f3"; Path=/;
           HttpOnly; Secure;]
       SOAPAction: ['"urn:vim25/4.1"']
     method: POST

diff --git a/tests/test_connect.py b/tests/test_connect.py
@@ -36,10 +36,15 @@ def test_basic_connection(self):
         si = connect.Connect(host='vcsa',
                              user='my_user',
                              pwd='my_password')
+        cookie = si._stub.cookie
         session_id = si.content.sessionManager.currentSession.key
+        # NOTE (hartsock): The cookie value should never change during
+        # a connected session. That should be verifiable in these tests.
+        self.assertEqual(cookie, si._stub.cookie)
         # NOTE (hartsock): assertIsNotNone does not work in Python 2.6
         self.assertTrue(session_id is not None)
         self.assertEqual('52773cd3-35c6-b40a-17f1-fe664a9f08f3', session_id)
+        self.assertTrue(session_id in cookie)
 
     @vcr.use_cassette('basic_connection_bad_password.yaml',
                       cassette_library_dir=fixtures_path, record_mode='none')