107 lines
4.5 KiB
Python
107 lines
4.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright(C) 2014 Julia Leven
|
|
#
|
|
# This file is part of weboob.
|
|
#
|
|
# weboob is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# weboob is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
|
|
import collections
|
|
import warnings
|
|
from unittest import TestCase
|
|
|
|
import lxml.html
|
|
|
|
from weboob.browser import URL
|
|
from weboob.browser.pages import Form, FormSubmitWarning
|
|
|
|
|
|
# Mock that allows to represent a Page
|
|
class MyMockPage():
|
|
url = URL("http://httpbin.org")
|
|
|
|
|
|
# Class that tests different methods from the class URL
|
|
class FormTest(TestCase):
|
|
|
|
# Initialization of the objects needed by the tests
|
|
def setUp(self):
|
|
self.page = MyMockPage()
|
|
self.el = lxml.html.fromstring(
|
|
"""<form method='GET'>
|
|
<input type ='text' name='nom' value='Dupont'/>
|
|
<input type ='text' name='prenom' value=''/>
|
|
<select name='mySelect'>
|
|
<option value='item1'>item1</option>
|
|
<option selected='true' value='item2'>item2</option>
|
|
</select>
|
|
<select name='mySelectNotSelected'>
|
|
<option value='item1'>item1</option>
|
|
<option value='item2'>item2</option>
|
|
</select>
|
|
<input type='submit' name='submitForm' />
|
|
</form>""")
|
|
self.elMoreSubmit = lxml.html.fromstring(
|
|
"""<form method='GET'>
|
|
<input type ='text' name='nom' value='Dupont'/>
|
|
<input type ='text' name='prenom' value=''/>
|
|
<select name='mySelect'>
|
|
<option value='item1'>item1</option>
|
|
<option selected='true' value='item2'>item2</option>
|
|
</select>
|
|
<select name='mySelectNotSelected'>
|
|
<option value='item1'>item1</option>
|
|
<option value='item2'>item2</option>
|
|
</select>
|
|
<input type='submit' name='submitForm'/>
|
|
<input type='submit' name='submit2'/>
|
|
</form>""")
|
|
|
|
# Checks that the dictionary is correctly initialised
|
|
def test_init_nominal_case(self):
|
|
form = Form(self.page, self.el, None)
|
|
self.assertDictEqual(form, collections.OrderedDict([
|
|
('nom', 'Dupont'), ('prenom', ''), ('mySelect', 'item2'),
|
|
('mySelectNotSelected', 'item1'), ('submitForm', u'')]))
|
|
|
|
# Checks that submit fields are not added to the dictionary when the
|
|
# attribute submit_el is set to False
|
|
def test_no_submit(self):
|
|
formNoSubmit = Form(self.page, self.el, False)
|
|
self.assertDictEqual(formNoSubmit, collections.OrderedDict([
|
|
('nom', 'Dupont'), ('prenom', ''), ('mySelect', 'item2'),
|
|
('mySelectNotSelected', 'item1')]))
|
|
|
|
# Checks that the right warning is issued when there are several submit
|
|
# fields
|
|
def test_warning_more_submit(self):
|
|
with warnings.catch_warnings(record=True) as w:
|
|
warnings.simplefilter('always')
|
|
Form(self.page, self.elMoreSubmit)
|
|
warningMsg = "Form has more than one submit input, you" + \
|
|
" should chose the correct one"
|
|
assert len(w) == 1
|
|
assert issubclass(w[-1].category, FormSubmitWarning)
|
|
assert warningMsg in str(w[-1].message)
|
|
|
|
# Checks that a warning is raised when the submit passed as a parameter
|
|
# does not exist in the form
|
|
def test_warning_submit_not_find(self):
|
|
with warnings.catch_warnings(record=True) as w:
|
|
warnings.simplefilter('always')
|
|
Form(self.page, self.el, lxml.html.fromstring(
|
|
"<input type='submit' name='submitNotFind' />"))
|
|
warningMsg = "Form had a submit element provided, but" + \
|
|
" it was not found"
|
|
assert len(w) == 1
|
|
assert issubclass(w[-1].category, FormSubmitWarning)
|
|
assert warningMsg in str(w[-1].message)
|