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()