Mercurial > farol
comparison tests/utils.py @ 0:4a9f23230eba
Initial Release
| author | Benoît Allard <benoit.allard@greenbone.net> |
|---|---|
| date | Wed, 24 Sep 2014 10:07:49 +0200 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| -1:000000000000 | 0:4a9f23230eba |
|---|---|
| 1 # -*- encoding: utf-8 -*- | |
| 2 # Description: | |
| 3 # Main test case class with some utilities | |
| 4 # | |
| 5 # Authors: | |
| 6 # Benoît Allard <benoit.allard@greenbone.net> | |
| 7 # | |
| 8 # Copyright: | |
| 9 # Copyright (C) 2014 Greenbone Networks GmbH | |
| 10 # | |
| 11 # This program is free software; you can redistribute it and/or | |
| 12 # modify it under the terms of the GNU General Public License | |
| 13 # as published by the Free Software Foundation; either version 2 | |
| 14 # of the License, or (at your option) any later version. | |
| 15 # | |
| 16 # This program is distributed in the hope that it will be useful, | |
| 17 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 19 # GNU General Public License for more details. | |
| 20 # | |
| 21 # You should have received a copy of the GNU General Public License | |
| 22 # along with this program; if not, write to the Free Software | |
| 23 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. | |
| 24 | |
| 25 import unittest | |
| 26 | |
| 27 from farol import main | |
| 28 from farol.session import CVRFs | |
| 29 | |
| 30 class TestCase(unittest.TestCase): | |
| 31 | |
| 32 def setUp(self): | |
| 33 main.app.testing = True | |
| 34 main.app.secret_key = 'test key' | |
| 35 self.app = main.app.test_client() | |
| 36 | |
| 37 def createDoc(self, title, type_): | |
| 38 rv = self.app.post('/new', data=dict(title=title, type=type_), follow_redirects=True) | |
| 39 self.id = list(CVRFs.keys())[0] | |
| 40 return rv | |
| 41 | |
| 42 def _assertIn(self, data, response, coding='utf-8'): | |
| 43 self.assertIn(data, response.data.decode(coding)) | |
| 44 |