diff options
author | Luis Pabon <lpabon@redhat.com> | 2013-08-29 11:07:03 -0400 |
---|---|---|
committer | Luis Pabon <lpabon@redhat.com> | 2013-08-30 07:51:50 -0700 |
commit | 4cf6bdb4acbec618640d819fe749a1a4430989e0 (patch) | |
tree | 611317eb3ddb03a5a18e6301b945fb92795c75f1 /test/unit/proxy/controllers/test_obj.py | |
parent | 8737f9e92a5c9dbf0d75e0702c7ab48f287f2761 (diff) |
OpenStack Swift proxy controller unit tests
These are a copy of the OpenStack Swift proxy controller
unit tests as of version their 1.9.1.
Change-Id: Ib05bc2a37dbb29d729346d78ea8f6de83b82375b
Signed-off-by: Luis Pabon <lpabon@redhat.com>
Reviewed-on: http://review.gluster.org/5735
Reviewed-by: Peter Portante <pportant@redhat.com>
Tested-by: Peter Portante <pportant@redhat.com>
Diffstat (limited to 'test/unit/proxy/controllers/test_obj.py')
-rwxr-xr-x | test/unit/proxy/controllers/test_obj.py | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/test/unit/proxy/controllers/test_obj.py b/test/unit/proxy/controllers/test_obj.py new file mode 100755 index 0000000..e4af789 --- /dev/null +++ b/test/unit/proxy/controllers/test_obj.py @@ -0,0 +1,64 @@ +#!/usr/bin/env python +# Copyright (c) 2010-2012 OpenStack, LLC. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or +# implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +import unittest + +from swift.proxy import server as proxy_server +from test.unit import FakeRing, FakeMemcache + + +class TestObjControllerWriteAffinity(unittest.TestCase): + def setUp(self): + self.app = proxy_server.Application( + None, FakeMemcache(), account_ring=FakeRing(), + container_ring=FakeRing(), object_ring=FakeRing(max_more_nodes=9)) + self.app.request_node_count = lambda ring: 10000000 + self.app.sort_nodes = lambda l: l # stop shuffling the primary nodes + + def test_iter_nodes_local_first_noops_when_no_affinity(self): + controller = proxy_server.ObjectController(self.app, 'a', 'c', 'o') + self.app.write_affinity_is_local_fn = None + + all_nodes = self.app.object_ring.get_part_nodes(1) + all_nodes.extend(self.app.object_ring.get_more_nodes(1)) + + local_first_nodes = list(controller.iter_nodes_local_first( + self.app.object_ring, 1)) + + self.maxDiff = None + + self.assertEqual(all_nodes, local_first_nodes) + + def test_iter_nodes_local_first_moves_locals_first(self): + controller = proxy_server.ObjectController(self.app, 'a', 'c', 'o') + self.app.write_affinity_is_local_fn = (lambda node: node['region'] == 1) + self.app.write_affinity_node_count = lambda ring: 4 + + all_nodes = self.app.object_ring.get_part_nodes(1) + all_nodes.extend(self.app.object_ring.get_more_nodes(1)) + + local_first_nodes = list(controller.iter_nodes_local_first( + self.app.object_ring, 1)) + + # the local nodes move up in the ordering + self.assertEqual([1, 1, 1, 1], + [node['region'] for node in local_first_nodes[:4]]) + # we don't skip any nodes + self.assertEqual(sorted(all_nodes), sorted(local_first_nodes)) + + +if __name__ == '__main__': + unittest.main() |