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