Add check_login function

2 files changed, 41 insertions, 0 deletions

diff --git a/pylibchorus/chorus_client.py b/pylibchorus/chorus_client.py
index 5c7119f..ff90e77 100644
--- a/pylibchorus/chorus_client.py
+++ b/pylibchorus/chorus_client.py
@@ -69,6 +69,19 @@ def _login_(username, password):
         'method': 'POST',
     }
 
+def _check_login_(_, cookies):
+    '''Create request data for check login check'''
+    return {
+        'data': None,
+        'params': None,
+        'headers': {
+            'content-type': CONTENT_TYPE,
+        },
+        'cookies': cookies,
+        'url': '/sessions',
+        'method': 'GET',
+    }
+
 def _logout_(sid, cookies):
     '''Create request data for ChorusSession'''
     return {
diff --git a/pylibchorus/tests/chorus_client_tests.py b/pylibchorus/tests/chorus_client_tests.py
index e2eb6e1..8802606 100644
--- a/pylibchorus/tests/chorus_client_tests.py
+++ b/pylibchorus/tests/chorus_client_tests.py
@@ -4,6 +4,7 @@
 import logging
 from pylibchorus.chorus_client import _login_
 from pylibchorus.chorus_client import _logout_
+from pylibchorus.chorus_client import _check_login_
 import unittest
 
 LOG = logging.getLogger(__name__)
@@ -72,3 +73,30 @@ class ChorusSessionTests(unittest.TestCase):
         self.assertEquals(cookies, actual['cookies'])
         self.assertEquals('/sessions', actual['url'])
         self.assertEquals('DELETE', actual['method'])
+
+    #pylint: disable=C0103
+    def test_check_login_returns_request_data(self):
+        '''Test _check_login_ returns correct request data'''
+        sid = 'foobar'
+        cookies = {'session_id': sid}
+        actual = _check_login_(sid, cookies)
+        self.assertIsNotNone(actual)
+        self.assertIn('data', actual)
+        self.assertIn('headers', actual)
+        self.assertIn('params', actual)
+        self.assertIn('cookies', actual)
+        self.assertIn('url', actual)
+        self.assertIn('method', actual)
+        self.assertIsNone(actual['data'])
+        self.assertIsNotNone(actual['headers'])
+        self.assertIsNone(actual['params'])
+        self.assertIsNotNone(actual['cookies'])
+        self.assertIsNotNone(actual['url'])
+        self.assertIsNotNone(actual['method'])
+        headers = actual['headers']
+        self.assertIn('content-type', headers)
+        self.assertEquals('application/x-www-form-urlencoded',
+                          headers['content-type'])
+        self.assertEquals(cookies, actual['cookies'])
+        self.assertEquals('/sessions', actual['url'])
+        self.assertEquals('GET', actual['method'])