From dec9c0f8273acf792b42bcff26663575054d23d2 Mon Sep 17 00:00:00 2001
From: micbou <contact@micbou.com>
Date: Fri, 7 Jul 2017 13:59:57 +0200
Subject: [PATCH] Add test with omnifunc moving cursor position

---
 python/ycm/tests/omni_completer_test.py | 37 +++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/python/ycm/tests/omni_completer_test.py b/python/ycm/tests/omni_completer_test.py
index a892813a..e8e02b5a 100644
--- a/python/ycm/tests/omni_completer_test.py
+++ b/python/ycm/tests/omni_completer_test.py
@@ -30,6 +30,7 @@ from ycm.tests.test_utils import ( ExpectedFailure, MockVimBuffers,
                                    MockVimModule, ToBytesOnPY2, VimBuffer )
 MockVimModule()
 
+from ycm import vimsupport
 from ycm.tests import YouCompleteMeInstance
 
 
@@ -621,3 +622,39 @@ def OmniCompleter_GetCompletions_Cache_ObjectListObject_Unicode_test( ycm ):
         'completion_start_column': 13
       } )
     )
+
+
+@YouCompleteMeInstance( { 'cache_omnifunc': 1 } )
+def OmniCompleter_GetCompletions_RestoreCursorPositionAfterOmnifuncCall_test(
+  ycm ):
+
+  # This omnifunc moves the cursor to the test definition like
+  # ccomplete#Complete would.
+  def Omnifunc( findstart, base ):
+    if findstart:
+      return 5
+    vimsupport.SetCurrentLineAndColumn( 0, 0 )
+    return [ 'length' ]
+
+  current_buffer = VimBuffer( 'buffer',
+                              contents = [ 'String test',
+                                           '',
+                                           'test.' ],
+                              filetype = 'java',
+                              omnifunc = Omnifunc )
+
+  with MockVimBuffers( [ current_buffer ], current_buffer, ( 3, 5 ) ):
+    # Make sure there is an omnifunc set up.
+    ycm.OnFileReadyToParse()
+    ycm.SendCompletionRequest()
+    assert_that(
+      vimsupport.CurrentLineAndColumn(),
+      contains( 2, 5 )
+    )
+    assert_that(
+      ycm.GetCompletionResponse(),
+      has_entries( {
+        'completions': ToBytesOnPY2( [ 'length' ] ),
+        'completion_start_column': 6
+      } )
+    )