1 files changed, 69 insertions, 0 deletions
diff --git a/platform_tools/android/tests/ordered_set_tests.py b/platform_tools/android/tests/ordered_set_tests.py
new file mode 100644
index 0000000000..248f0f6f6c
--- /dev/null
+++ b/platform_tools/android/tests/ordered_set_tests.py
@@ -0,0 +1,69 @@
+#!/usr/bin/python
+
+# Copyright 2014 Google Inc.
+#
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""
+Test OrderedSet.
+"""
+
+import sys
+import test_variables
+import unittest
+
+sys.path.append(test_variables.GYP_GEN_DIR)
+
+from vars_dict_lib import OrderedSet
+
+
+def create_dummy_var(i):
+  return 'dummy_var' + str(i)
+
+
+class OrderedSetTest(unittest.TestCase):
+
+  def setUp(self):
+    self.__set = OrderedSet()
+
+  def test_methods(self):
+    """
+    Test methods on OrderedSet.
+    """
+    RANGE = 10
+    for i in range(RANGE):
+      dummy_var = create_dummy_var(i)
+      # Add to the list. This should succeed.
+      self.__set.add(dummy_var)
+      self.assertEqual(len(self.__set), i+1)
+      self.assertTrue(dummy_var in self.__set)
+      self.assertEqual(self.__set[i], dummy_var)
+
+      # Now attempt to add it again. This should fail.
+      self.__set.add(dummy_var)
+      self.assertEqual(len(self.__set), i+1)
+      self.assertEqual(self.__set[i], dummy_var)
+
+    # Test iterator.
+    counter = 0
+    for set_member in self.__set:
+      self.assertEqual(create_dummy_var(counter), set_member)
+      counter += 1
+    self.assertEqual(counter, len(self.__set))
+
+    # Now test removal.
+    for i in range(RANGE):
+      dummy_var = create_dummy_var(i)
+      self.__set.remove(dummy_var)
+      self.assertEqual(len(self.__set), RANGE-i-1)
+      self.assertFalse(dummy_var in self.__set)
+
+def main():
+  loader = unittest.TestLoader()
+  suite = loader.loadTestsFromTestCase(OrderedSetTest)
+  unittest.TextTestRunner(verbosity=2).run(suite)
+
+if __name__ == "__main__":
+  main()
+