root / snf-cyclades-app / synnefo / logic / test / rapi_pool_tests.py @ 5962d9a6
History | View | Annotate | Download (3.7 kB)
1 | b66e4833 | Christos Stavrakakis | # Copyright 2011-2012 GRNET S.A. All rights reserved.
|
---|---|---|---|
2 | b66e4833 | Christos Stavrakakis | #
|
3 | b66e4833 | Christos Stavrakakis | # Redistribution and use in source and binary forms, with or
|
4 | b66e4833 | Christos Stavrakakis | # without modification, are permitted provided that the following
|
5 | b66e4833 | Christos Stavrakakis | # conditions are met:
|
6 | b66e4833 | Christos Stavrakakis | #
|
7 | b66e4833 | Christos Stavrakakis | # 1. Redistributions of source code must retain the above
|
8 | b66e4833 | Christos Stavrakakis | # copyright notice, this list of conditions and the following
|
9 | b66e4833 | Christos Stavrakakis | # disclaimer.
|
10 | b66e4833 | Christos Stavrakakis | #
|
11 | b66e4833 | Christos Stavrakakis | # 2. Redistributions in binary form must reproduce the above
|
12 | b66e4833 | Christos Stavrakakis | # copyright notice, this list of conditions and the following
|
13 | b66e4833 | Christos Stavrakakis | # disclaimer in the documentation and/or other materials
|
14 | b66e4833 | Christos Stavrakakis | # provided with the distribution.
|
15 | b66e4833 | Christos Stavrakakis | #
|
16 | b66e4833 | Christos Stavrakakis | # THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
|
17 | b66e4833 | Christos Stavrakakis | # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
18 | b66e4833 | Christos Stavrakakis | # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
19 | b66e4833 | Christos Stavrakakis | # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
|
20 | b66e4833 | Christos Stavrakakis | # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
21 | b66e4833 | Christos Stavrakakis | # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
22 | b66e4833 | Christos Stavrakakis | # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
|
23 | b66e4833 | Christos Stavrakakis | # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
|
24 | b66e4833 | Christos Stavrakakis | # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
25 | b66e4833 | Christos Stavrakakis | # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
26 | b66e4833 | Christos Stavrakakis | # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
27 | b66e4833 | Christos Stavrakakis | # POSSIBILITY OF SUCH DAMAGE.
|
28 | b66e4833 | Christos Stavrakakis | #
|
29 | b66e4833 | Christos Stavrakakis | # The views and conclusions contained in the software and
|
30 | b66e4833 | Christos Stavrakakis | # documentation are those of the authors and should not be
|
31 | b66e4833 | Christos Stavrakakis | # interpreted as representing official policies, either expressed
|
32 | b66e4833 | Christos Stavrakakis | # or implied, of GRNET S.A.
|
33 | b66e4833 | Christos Stavrakakis | |
34 | b66e4833 | Christos Stavrakakis | from django.test import TestCase |
35 | b66e4833 | Christos Stavrakakis | |
36 | b66e4833 | Christos Stavrakakis | from synnefo.logic import rapi_pool |
37 | b66e4833 | Christos Stavrakakis | |
38 | b66e4833 | Christos Stavrakakis | from mock import patch |
39 | b66e4833 | Christos Stavrakakis | |
40 | b66e4833 | Christos Stavrakakis | |
41 | b66e4833 | Christos Stavrakakis | @patch('synnefo.logic.rapi_pool.GanetiRapiClient', spec=True) |
42 | b66e4833 | Christos Stavrakakis | class GanetiRapiPoolTest(TestCase): |
43 | b66e4833 | Christos Stavrakakis | def test_new_client(self, rclient): |
44 | b66e4833 | Christos Stavrakakis | cl = rapi_pool.get_rapi_client(1, 'amxixa', 'cluster0', '5080', 'user', |
45 | b66e4833 | Christos Stavrakakis | 'pass')
|
46 | b66e4833 | Christos Stavrakakis | rclient.assert_called_once_with("cluster0", "5080", "user", "pass") |
47 | b66e4833 | Christos Stavrakakis | self.assertTrue('amxixa' in rapi_pool._pools) |
48 | b66e4833 | Christos Stavrakakis | self.assertTrue(cl._pool is rapi_pool._pools[rapi_pool._hashes[1]]) |
49 | b66e4833 | Christos Stavrakakis | |
50 | b66e4833 | Christos Stavrakakis | def test_invalid_get(self, rclient): |
51 | b66e4833 | Christos Stavrakakis | self.assertRaises(ValueError, rapi_pool.get_rapi_client, 1, 'amxixa', |
52 | cc92b70f | Christos Stavrakakis | None, '5080', 'user', 'pass') |
53 | b66e4833 | Christos Stavrakakis | self.assertRaises(ValueError, rapi_pool.get_rapi_client, 1, 'amxixa', |
54 | cc92b70f | Christos Stavrakakis | 'Foo', None, 'user', 'pass') |
55 | b66e4833 | Christos Stavrakakis | |
56 | b66e4833 | Christos Stavrakakis | def test_get_from_pool(self, rclient): |
57 | b66e4833 | Christos Stavrakakis | cl = rapi_pool.get_rapi_client(1, 'dummyhash', 'cluster1', '5080', |
58 | b66e4833 | Christos Stavrakakis | 'user', 'pass') |
59 | b66e4833 | Christos Stavrakakis | rclient.assert_called_once_with("cluster1", "5080", "user", "pass") |
60 | b66e4833 | Christos Stavrakakis | rapi_pool.put_rapi_client(cl) |
61 | b66e4833 | Christos Stavrakakis | rclient.reset_mock() |
62 | b66e4833 | Christos Stavrakakis | cl2 = rapi_pool.get_rapi_client(1, 'dummyhash', 'cluster1', '5080', |
63 | b66e4833 | Christos Stavrakakis | 'user', 'pass') |
64 | b66e4833 | Christos Stavrakakis | self.assertTrue(cl is cl2) |
65 | b66e4833 | Christos Stavrakakis | self.assertFalse(rclient.mock_calls)
|
66 | b66e4833 | Christos Stavrakakis | |
67 | b66e4833 | Christos Stavrakakis | def test_changed_credentials(self, rclient): |
68 | b66e4833 | Christos Stavrakakis | cl = rapi_pool.get_rapi_client(1, 'dummyhash2', 'cluster2', '5080', |
69 | b66e4833 | Christos Stavrakakis | 'user', 'pass') |
70 | b66e4833 | Christos Stavrakakis | rclient.assert_called_once_with("cluster2", "5080", "user", "pass") |
71 | b66e4833 | Christos Stavrakakis | rapi_pool.put_rapi_client(cl) |
72 | b66e4833 | Christos Stavrakakis | rclient.reset_mock() |
73 | b66e4833 | Christos Stavrakakis | rapi_pool.get_rapi_client(1, 'dummyhash3', 'cluster2', '5080', |
74 | cc92b70f | Christos Stavrakakis | 'user', 'new_pass') |
75 | b66e4833 | Christos Stavrakakis | rclient.assert_called_once_with("cluster2", "5080", "user", "new_pass") |
76 | b66e4833 | Christos Stavrakakis | self.assertFalse('dummyhash2' in rapi_pool._pools) |
77 | b66e4833 | Christos Stavrakakis | |
78 | b66e4833 | Christos Stavrakakis | def test_no_pool(self, rclient): |
79 | b66e4833 | Christos Stavrakakis | cl = rapi_pool.get_rapi_client(1, 'dummyhash2', 'cluster2', '5080', |
80 | b66e4833 | Christos Stavrakakis | 'user', 'pass') |
81 | b66e4833 | Christos Stavrakakis | cl._pool = None
|
82 | b66e4833 | Christos Stavrakakis | rapi_pool.put_rapi_client(cl) |
83 | b66e4833 | Christos Stavrakakis | self.assertTrue(cl not in rapi_pool._pools.values()) |