Unittesting repl.py

mmaker · mmaker · commit a4aca49d7168 · 2010-07-07T09:14:38.000+02:00
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -161,7 +161,7 @@ def append(self, line):
     def first(self):
         """Move back to the beginning of the history."""
         if not self.is_at_end:
-            self.index = len(self.entries) - 1
+            self.index = len(self.entries)
         return self.entries[-self.index]
 
     def back(self):
diff --git a/bpython/test/test_repl.py b/bpython/test/test_repl.py
@@ -0,0 +1,60 @@
+#!/usr/bin/env python
+import unittest
+from bpython import repl
+
+
+class TestHistory(unittest.TestCase):
+    def setUp(self):
+        self.history = repl.History('#%d' % x for x in range(1000))
+
+    def test_is_at_start(self):
+        self.history.first()
+
+        self.assertNotEqual(self.history.index, 0)
+        self.assertTrue(self.history.is_at_end)
+        self.history.forward()
+        self.assertFalse(self.history.is_at_end)
+
+    def test_is_at_end(self):
+        self.history.last()
+
+        self.assertEqual(self.history.index, 0)
+        self.assertTrue(self.history.is_at_start)
+        self.assertFalse(self.history.is_at_end)
+
+    def test_first(self):
+        self.history.first()
+
+        self.assertFalse(self.history.is_at_start)
+        self.assertTrue(self.history.is_at_end)
+
+    def test_last(self):
+        self.history.last()
+
+        self.assertTrue(self.history.is_at_start)
+        self.assertFalse(self.history.is_at_end)
+
+    def test_back(self):
+        self.assertEqual(self.history.back(), '#999')
+        self.assertNotEqual(self.history.back(), '#999')
+        self.assertEqual(self.history.back(), '#997')
+        for x in range(997):
+            self.history.back()
+        self.assertEqual(self.history.back(), '#0')
+
+    def test_forward(self):
+        self.history.first()
+
+        self.assertEqual(self.history.forward(), '#1')
+        self.assertNotEqual(self.history.forward(), '#1')
+        self.assertEqual(self.history.forward(), '#3')
+        #  1000 == entries   4 == len(range(1, 3) ===> '#1000' (so +1)
+        for x in range(1000 - 4 - 1):
+            self.history.forward()
+        self.assertEqual(self.history.forward(), '#999')
+
+
+
+
+if __name__ == '__main__':
+    unittest.main()
