From cb884b11cdcf75a8b5ad82fe16069df6b2c1b1e2 Mon Sep 17 00:00:00 2001 From: kballou Date: Thu, 9 Apr 2015 16:54:02 -0600 Subject: Add check_login function --- pylibchorus/chorus_client.py | 13 +++++++++++++ pylibchorus/tests/chorus_client_tests.py | 28 ++++++++++++++++++++++++++++ 2 files changed, 41 insertions(+) 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']) -- cgit v1.2.1