IndexedDict class for storing the installer steps

1 files changed, 39 insertions, 0 deletions

diff --git a/tests/pyanaconda_test/indexed_dict_test.py b/tests/pyanaconda_test/indexed_dict_test.py
new file mode 100644
index 000000000..c56a9df78
--- /dev/null
+++ b/tests/pyanaconda_test/indexed_dict_test.py
@@ -0,0 +1,39 @@
+import mock
+from pyanaconda.indexed_dict import IndexedDict
+
+class IndexedDictTest(mock.TestCase):
+    def instantiation_test(self):
+        d = IndexedDict()
+        self.assertIsInstance(d, IndexedDict)
+
+    def append_test(self):
+        d = IndexedDict()
+        stored_data = [1, 2, 3]
+        d["some_step"] = stored_data
+        self.assertIs(d["some_step"], stored_data)
+
+    def cant_append_test(self):
+        def assign_int(indexed_dict):
+            indexed_dict[3] = [1, 2, 3]
+        d = IndexedDict()
+        self.assertRaises(TypeError, d.__setitem__, 3, [1, 2, 3])
+
+    def referencing_test(self):
+        d = IndexedDict()
+        d["first"] = 10
+        d["second"] = 20
+        d["third"] = 30
+        self.assertEqual(d[0], 10)
+        self.assertEqual(d[1], 20)
+        self.assertEqual(d[2], 30)
+        self.assertRaises(IndexError, d.__getitem__, 3)
+
+    def index_test(self):
+        d = IndexedDict()
+        d["first"] = 10
+        d["second"] = 20
+        d["third"] = 30
+
+        self.assertEqual(d.index("first"), 0)
+        self.assertEqual(d.index("second"), 1)
+        self.assertEqual(d.index("third"), 2)
\ No newline at end of file