1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
# Copyright 2017 Lars Wirzenius
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# =*= License: GPL-3+ =*=
import os
import unittest
import yarnhelper
class GetNextMatchTests(unittest.TestCase):
def test_raises_error_if_no_next_match(self):
h = yarnhelper.YarnHelper()
h.set_environment({})
with self.assertRaises(yarnhelper.Error):
h.get_next_match()
def test_returns_first_match_if_there(self):
h = yarnhelper.YarnHelper()
h.set_environment({
'MATCH_1': 'first',
})
self.assertEqual(h.get_next_match(), 'first')
def test_returns_second_match_if_there(self):
h = yarnhelper.YarnHelper()
h.set_environment({
'MATCH_1': 'first',
'MATCH_2': 'second',
})
self.assertEqual(h.get_next_match(), 'first')
self.assertEqual(h.get_next_match(), 'second')
def test_raises_error_if_no_more_matches(self):
h = yarnhelper.YarnHelper()
h.set_environment({
'MATCH_1': 'first',
})
self.assertEqual(h.get_next_match(), 'first')
with self.assertRaises(yarnhelper.Error):
h.get_next_match()
class PersistentVariableTests(unittest.TestCase):
def setUp(self):
# We need this so that tearDown works
pass
def tearDown(self):
if os.path.exists(yarnhelper.variables_filename):
os.remove(yarnhelper.variables_filename)
def test_sets_variable_persistently(self):
h = yarnhelper.YarnHelper()
h.set_variable('FOO', 'bar')
h2 = yarnhelper.YarnHelper()
self.assertEqual(h2.get_variable('FOO'), 'bar')
def test_get_returns_default_if_variable_not_set(self):
h = yarnhelper.YarnHelper()
self.assertEqual(h.get_variable('FOO', 'bar'), 'bar')
def test_appends_to_empty_list(self):
h = yarnhelper.YarnHelper()
h.append_to_list('foo', 1)
self.assertEqual(h.get_variable('foo'), [1])
class HttpTests(unittest.TestCase):
def test_constructs_aliased_request(self):
h = yarnhelper.YarnHelper()
server = 'new.example.com'
url = 'http://www.example.com/path'
r = h.construct_aliased_http_request(server, 'GET', url)
self.assertEqual(r.url, 'http://new.example.com/path')
self.assertEqual(r.headers['Host'], 'www.example.com')
class AssertionTests(unittest.TestCase):
def test_assertEqual_asserts_equals_correctly(self):
h = yarnhelper.YarnHelper()
self.assertEqual(h.assertEqual(0, 0), None)
def test_assertEqual_raises_error_for_nonequal_values(self):
h = yarnhelper.YarnHelper()
with self.assertRaises(yarnhelper.Error):
h.assertEqual(0, 1)
def test_assertNotEqual_asserts_nonequal_correct(self):
h = yarnhelper.YarnHelper()
self.assertEqual(h.assertNotEqual(0, 1), None)
def test_assertNotEqual_raises_error_for_equal_values(self):
h = yarnhelper.YarnHelper()
with self.assertRaises(yarnhelper.Error):
h.assertNotEqual(0, 0)
def test_assertGreaterThan_raises_error_for_equal_values(self):
h = yarnhelper.YarnHelper()
with self.assertRaises(yarnhelper.Error):
h.assertGreaterThan(0, 0)
def test_assertGreaterThan_raises_error_for_unordered_values(self):
h = yarnhelper.YarnHelper()
with self.assertRaises(yarnhelper.Error):
h.assertGreaterThan(0, 1)
def test_assertIn_asserts_correctly(self):
h = yarnhelper.YarnHelper()
self.assertEqual(h.assertIn('ana', 'banana'), None)
def test_assertIn_raises_error_for_false(self):
h = yarnhelper.YarnHelper()
with self.assertRaises(yarnhelper.Error):
h.assertIn('nope', 'banana')
|
