1 files changed, 130 insertions, 0 deletions
diff --git a/test/python/test_deoplete_source.py b/test/python/test_deoplete_source.py
new file mode 100644
index 00000000..960abe3a
--- /dev/null
+++ b/test/python/test_deoplete_source.py
@@ -0,0 +1,130 @@
+import unittest
+import imp
+
+ale_module = imp.load_source(
+    'deoplete.sources.ale',
+    '/testplugin/rplugin/python3/deoplete/sources/ale.py',
+)
+
+
+class VimMock(object):
+    def __init__(self, call_list, call_results):
+        self.__call_list = call_list
+        self.__call_results = call_results
+
+    def call(self, function, *args):
+        self.__call_list.append((function, args))
+
+        return self.__call_results.get(function, 0)
+
+
+class DeopleteSourceTest(unittest.TestCase):
+    def setUp(self):
+        super(DeopleteSourceTest, self).setUp()
+
+        self.call_list = []
+        self.call_results = {}
+        self.source = ale_module.Source('vim')
+        self.source.vim = VimMock(self.call_list, self.call_results)
+
+    def test_attributes(self):
+        """
+        Check all of the attributes we set.
+        """
+        attributes = dict(
+            (key, getattr(self.source, key))
+            for key in
+            dir(self.source)
+            if not key.startswith('__')
+            and key != 'vim'
+            and not hasattr(getattr(self.source, key), '__self__')
+        )
+
+        self.assertEqual(attributes, {
+            'is_bytepos': True,
+            'mark': '[L]',
+            'min_pattern_length': 1,
+            'name': 'ale',
+            'rank': 100,
+        })
+
+    def test_completion_position(self):
+        self.call_results['ale#completion#GetCompletionPosition'] = 2
+
+        self.assertEqual(self.source.get_completion_position(), 2)
+        self.assertEqual(self.call_list, [
+            ('ale#completion#GetCompletionPosition', ()),
+        ])
+
+    def test_request_completion_results(self):
+        context = {'is_async': False}
+
+        self.assertEqual(self.source.gather_candidates(context), [])
+        self.assertEqual(context, {'is_async': True})
+        self.assertEqual(self.call_list, [
+            ('ale#completion#GetCompletions', ('deoplete',)),
+        ])
+
+    def test_refresh_completion_results(self):
+        context = {'is_async': False}
+
+        self.assertEqual(self.source.gather_candidates(context), [])
+        self.assertEqual(context, {'is_async': True})
+        self.assertEqual(self.call_list, [
+            ('ale#completion#GetCompletions', ('deoplete',)),
+        ])
+
+        context = {'is_async': True, 'is_refresh': True}
+
+        self.assertEqual(self.source.gather_candidates(context), [])
+        self.assertEqual(context, {'is_async': True, 'is_refresh': True})
+        self.assertEqual(self.call_list, [
+            ('ale#completion#GetCompletions', ('deoplete',)),
+            ('ale#completion#GetCompletions', ('deoplete',)),
+        ])
+
+    def test_poll_no_result(self):
+        context = {'is_async': True}
+        self.call_results['ale#completion#GetCompletionResult'] = None
+
+        self.assertEqual(self.source.gather_candidates(context), [])
+        self.assertEqual(context, {'is_async': True})
+        self.assertEqual(self.call_list, [
+            ('ale#completion#GetCompletionResult', ()),
+        ])
+
+    def test_poll_empty_result_ready(self):
+        context = {'is_async': True}
+        self.call_results['ale#completion#GetCompletionResult'] = []
+
+        self.assertEqual(self.source.gather_candidates(context), [])
+        self.assertEqual(context, {'is_async': False})
+        self.assertEqual(self.call_list, [
+            ('ale#completion#GetCompletionResult', ()),
+        ])
+
+    def test_poll_non_empty_result_ready(self):
+        context = {'is_async': True}
+        self.call_results['ale#completion#GetCompletionResult'] = [
+            {
+                'word': 'foobar',
+                'kind': 'v',
+                'icase': 1,
+                'menu': '',
+                'info': '',
+            },
+        ]
+
+        self.assertEqual(self.source.gather_candidates(context), [
+            {
+                'word': 'foobar',
+                'kind': 'v',
+                'icase': 1,
+                'menu': '',
+                'info': '',
+            },
+        ])
+        self.assertEqual(context, {'is_async': False})
+        self.assertEqual(self.call_list, [
+            ('ale#completion#GetCompletionResult', ()),
+        ])