Add tests

Shawn Axsom · thomasballinger · commit ce1eb1aa2eb2 · 2015-11-28T14:38:54.000-05:00
diff --git a/bpython/autocomplete.py b/bpython/autocomplete.py
@@ -335,7 +335,7 @@ def list_attributes(self, obj):
                 return dir(obj)
 
 
-class ArrayObjectMembersCompletion(BaseCompletionType):
+class ArrayItemMembersCompletion(BaseCompletionType):
 
     def __init__(self, shown_before_tab=True, mode=SIMPLE):
         self._shown_before_tab = shown_before_tab
@@ -603,7 +603,7 @@ def get_default_completer(mode=SIMPLE):
         MagicMethodCompletion(mode=mode),
         MultilineJediCompletion(mode=mode),
         GlobalCompletion(mode=mode),
-        ArrayObjectMembersCompletion(mode=mode),
+        ArrayItemMembersCompletion(mode=mode),
         CumulativeCompleter((AttrCompletion(mode=mode),
                              ParameterNameCompletion(mode=mode)),
                             mode=mode)
diff --git a/bpython/test/test_autocomplete.py b/bpython/test/test_autocomplete.py
@@ -263,6 +263,22 @@ def __getattr__(self, attr):
                       self.com.matches(4, 'a.__', locals_=locals_))
 
 
+class TestArrayItemCompletion(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.com = autocomplete.ArrayItemMembersCompletion()
+
+    def test_att_matches_found_on_instance(self):
+        self.assertSetEqual(self.com.matches(5, 'a[0].', locals_={'a': [Foo()]}),
+                            set(['a[0].method', 'a[0].a', 'a[0].b']))
+
+    @skip_old_style
+    def test_att_matches_found_on_old_style_instance(self):
+        self.assertSetEqual(self.com.matches(5, 'a[0].',
+                                             locals_={'a': [OldStyleFoo()]}),
+                            set(['a[0].method', 'a[0].a', 'a[0].b']))
+
+
 class TestMagicMethodCompletion(unittest.TestCase):
 
     def test_magic_methods_complete_after_double_underscores(self):
