git.maemo.org Git - ipypbx/blob - src/ipypbx/tests/test_http.py

   1 # Copyright (c) Stas Shtin, 2010
   2
   3 # This file is part of IPyPBX.
   4
   5 # IPyPBX is free software: you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation, either version 3 of the License, or
   8 # (at your option) any later version.
   9
  10 # IPyPBX is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14
  15 # You should have received a copy of the GNU General Public License
  16 # along with IPyPBX.  If not, see <http://www.gnu.org/licenses/>.
  17
  18 import unittest
  19 from ipypbx import http
  20
  21
  22 class TestHttpRequestParser(unittest.TestCase):
  23     """
  24     Tests for ipypbx.http.HttpRequestParser.
  25     """
  26     def setUp(self):
  27         """
  28         Initialize request object.
  29         """
  30         self.request = http.HttpRequestParser()
  31
  32     def test_initial_state(self):
  33         """
  34         Check the default values.
  35         """
  36         self.assertEqual(self.request.method, None)
  37         self.assertEqual(self.request.request_path, None)
  38         self.assertEqual(self.request.http_version, None)
  39         self.assertEqual(self.request.headers, {})
  40         self.assertEqual(self.request.data, {})
  41         self.assertEqual(self.request.result, None)
  42         self.assertEqual(self.request.state, self.request.HTTP_NONE)
  43
  44     def test_handle_none(self):
  45         """
  46         HttpRequestParser.handleNone should just invoke next method and got to
  47         headers parsing.
  48         """
  49         self.request.handle('GET / HTTP/1.1')
  50         self.assertNotEqual(self.request.state, self.request.HTTP_NONE)
  51
  52
  53     def test_handle_request(self):
  54         """
  55         HttpRequestParser.handleRequest should get method, request path and HTTP
  56         version from the request and move to headers parsing.
  57         """
  58         self.request.handle('GET / HTTP/1.1')
  59         self.assertEqual(self.request.method, 'GET')
  60         self.assertEqual(self.request.request_path, '/')
  61         self.assertEqual(self.request.http_version, 'HTTP/1.1')
  62
  63         self.assertEqual(self.request.state, self.request.HTTP_HEADERS)
  64
  65
  66     def test_handle_headers(self):
  67         """
  68         HttpRequestParser.handleNone should parse headers until empty line is
  69         received.
  70         """
  71         # Move to target state.
  72         self.request.handle('GET / HTTP/1.1')
  73
  74         # Handle first header.
  75         self.request.handle('Foo: bar')
  76         self.assertEqual(self.request.state, self.request.HTTP_HEADERS)
  77         self.assertEqual(self.request.headers['Foo'], 'bar')
  78         self.assertEqual(len(self.request.headers), 1)
  79
  80         # Handle next header.
  81         self.request.handle('Qwe: asd')
  82         self.assertEqual(self.request.state, self.request.HTTP_HEADERS)
  83         self.assertEqual(self.request.headers['Qwe'], 'asd')
  84         self.assertEqual(len(self.request.headers), 2)
  85
  86         # Empty line terminates
  87         self.request.handle('')
  88         self.assertNotEqual(self.request.state, self.request.HTTP_HEADERS)
  89         self.assertEqual(self.request.headers['Qwe'], 'asd')
  90         self.assertEqual(len(self.request.headers), 2)
  91
  92     def test_handle_empty(self):
  93         """
  94         HttpRequestParser.handleEmpty just passes to next state.
  95         """
  96         prev_state = self.request.state
  97         self.request.handleEmpty('')
  98         self.assertEqual(self.request.state, prev_state + 1)
  99
 100     def _run_lines(self, *lines):
 101         for line in lines:
 102             self.request.handle(line)
 103
 104     def test_handle_body(self):
 105         self._run_lines(
 106             'POST / HTTP/1.1',
 107             '',
 108             'foo=bar&qwe=asd')
 109         self.assertEqual(self.request.data, {'foo': 'bar', 'qwe': 'asd'})
 110
 111     def test_handle_body_with_missing_values(self):
 112         self._run_lines(
 113             'POST / HTTP/1.1',
 114             '',
 115             'foo=bar&qwe=')
 116         self.assertEqual(self.request.data, {'foo': 'bar', 'qwe': ''})
 117
 118
 119     def test_non_post_fails(self):
 120         self.request.handle('GET / HTTP/1.1')
 121         self.request.handle('')
 122         self.assertRaises(
 123             http.HttpParseError, self.request.handle,
 124             'Only post requests work')
 125
 126