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
|
#!/usr/bin/env python
import os
import tempfile
import unittest2
import mock
import mitogen.core
import mitogen.parent
import mitogen.master
import mitogen.utils
from mitogen.core import b
import testlib
def func0(router):
return router
@mitogen.utils.with_router
def func(router):
return router
class RunWithRouterTest(testlib.TestCase):
# test_shutdown_on_exception
# test_shutdown_on_success
def test_run_with_broker(self):
router = mitogen.utils.run_with_router(func0)
self.assertIsInstance(router, mitogen.master.Router)
self.assertFalse(testlib.threading__thread_is_alive(router.broker._thread))
class WithRouterTest(testlib.TestCase):
def test_with_broker(self):
router = func()
self.assertIsInstance(router, mitogen.master.Router)
self.assertFalse(testlib.threading__thread_is_alive(router.broker._thread))
class Dict(dict): pass
class List(list): pass
class Tuple(tuple): pass
class Unicode(mitogen.core.UnicodeType): pass
class Bytes(mitogen.core.BytesType): pass
class CastTest(testlib.TestCase):
def test_dict(self):
self.assertEqual(type(mitogen.utils.cast({})), dict)
self.assertEqual(type(mitogen.utils.cast(Dict())), dict)
def test_nested_dict(self):
specimen = mitogen.utils.cast(Dict({'k': Dict({'k2': 'v2'})}))
self.assertEqual(type(specimen), dict)
self.assertEqual(type(specimen['k']), dict)
def test_list(self):
self.assertEqual(type(mitogen.utils.cast([])), list)
self.assertEqual(type(mitogen.utils.cast(List())), list)
def test_nested_list(self):
specimen = mitogen.utils.cast(List((0, 1, List((None,)))))
self.assertEqual(type(specimen), list)
self.assertEqual(type(specimen[2]), list)
def test_tuple(self):
self.assertEqual(type(mitogen.utils.cast(())), list)
self.assertEqual(type(mitogen.utils.cast(Tuple())), list)
def test_nested_tuple(self):
specimen = mitogen.utils.cast(Tuple((0, 1, Tuple((None,)))))
self.assertEqual(type(specimen), list)
self.assertEqual(type(specimen[2]), list)
def assertUnchanged(self, v):
self.assertIs(mitogen.utils.cast(v), v)
def test_passthrough(self):
self.assertUnchanged(0)
self.assertUnchanged(0.0)
self.assertUnchanged(float('inf'))
self.assertUnchanged(True)
self.assertUnchanged(False)
self.assertUnchanged(None)
def test_unicode(self):
self.assertEqual(type(mitogen.utils.cast(u'')), mitogen.core.UnicodeType)
self.assertEqual(type(mitogen.utils.cast(Unicode())), mitogen.core.UnicodeType)
def test_bytes(self):
self.assertEqual(type(mitogen.utils.cast(b(''))), mitogen.core.BytesType)
self.assertEqual(type(mitogen.utils.cast(Bytes())), mitogen.core.BytesType)
def test_unknown(self):
self.assertRaises(TypeError, mitogen.utils.cast, set())
self.assertRaises(TypeError, mitogen.utils.cast, 4j)
if __name__ == '__main__':
unittest2.main()
|
