Merge remote-tracking branch 'upstream/master'

4 files changed, 672 insertions, 1 deletions

diff --git a/test/lsp/test_other_initialize_message_handling.vader b/test/lsp/test_other_initialize_message_handling.vader
index b6ef852a..f3b53843 100644
--- a/test/lsp/test_other_initialize_message_handling.vader
+++ b/test/lsp/test_other_initialize_message_handling.vader
@@ -23,6 +23,7 @@ Before:
   \   'completion_trigger_characters': [],
   \   'definition': 0,
   \   'symbol_search': 0,
+  \   'code_actions': 0,
   \ },
   \}
 
@@ -102,6 +103,7 @@ Execute(Capabilities should bet set up correctly):
   \   'definition': 1,
   \   'symbol_search': 1,
   \   'rename': 1,
+  \   'code_actions': 1,
   \ },
   \ b:conn.capabilities
   AssertEqual [[1, 'initialized', {}]], g:message_list
@@ -125,7 +127,7 @@ Execute(Disabled capabilities should be recognised correctly):
   \     'referencesProvider': v:false,
   \     'textDocumentSync': 2,
   \     'documentFormattingProvider': v:true,
-  \     'codeActionProvider': v:true,
+  \     'codeActionProvider': v:false,
   \     'signatureHelpProvider': {
   \       'triggerCharacters': ['(', ','],
   \     },
@@ -146,6 +148,7 @@ Execute(Disabled capabilities should be recognised correctly):
   \   'definition': 0,
   \   'symbol_search': 0,
   \   'rename': 0,
+  \   'code_actions': 0,
   \ },
   \ b:conn.capabilities
   AssertEqual [[1, 'initialized', {}]], g:message_list
@@ -197,6 +200,7 @@ Execute(Capabilities should be enabled when send as Dictionaries):
   \   'typeDefinition': 1,
   \   'symbol_search': 1,
   \   'rename': 1,
+  \   'code_actions': 1,
   \ },
   \ b:conn.capabilities
   AssertEqual [[1, 'initialized', {}]], g:message_list
diff --git a/test/test_code_action.vader b/test/test_code_action.vader
index 2e5d1381..7eabb759 100644
--- a/test/test_code_action.vader
+++ b/test/test_code_action.vader
@@ -3,6 +3,9 @@ Before:
 
   let g:ale_enabled = 0
 
+  " Enable fix end-of-line as tests below expect that
+  set fixeol
+
   runtime autoload/ale/code_action.vim
   runtime autoload/ale/util.vim
 
@@ -211,6 +214,7 @@ Execute(End of file can be modified):
   \)
 
   AssertEqual g:test.text + [
+  \  '',
   \  'type A: string',
   \  'type B: number',
   \  '',
diff --git a/test/test_code_action_python.vader b/test/test_code_action_python.vader
new file mode 100644
index 00000000..fd30633d
--- /dev/null
+++ b/test/test_code_action_python.vader
@@ -0,0 +1,59 @@
+Given python(An example Python file):
+  def main():
+      a = 1
+      c = a + 1
+
+Execute():
+  let g:changes = [
+  \ {'end': {'offset': 7, 'line': 1}, 'newText': 'func_qtffgsv', 'start': {'offset': 5, 'line': 1}},
+  \ {'end': {'offset': 9, 'line': 1}, 'newText': '', 'start': {'offset': 8, 'line': 1}},
+  \ {'end': {'offset': 15, 'line': 3}, 'newText': "    return c\n\n\ndef main():\n    c = func_qtffgsvi()\n", 'start': {'offset': 15, 'line': 3}}
+  \]
+
+  call ale#code_action#ApplyChanges(expand('%:p'), g:changes, 0)
+
+Expect(The changes should be applied correctly):
+  def func_qtffgsvi():
+      a = 1
+      c = a + 1
+      return c
+
+
+  def main():
+      c = func_qtffgsvi()
+
+
+Given python(Second python example):
+  import sys
+  import exifread
+
+  def main():
+      with open(sys.argv[1], 'rb') as f:
+          exif = exifread.process_file(f)
+          dt = str(exif['Image DateTime'])
+          date = dt[:10].replace(':', '-')
+
+Execute():
+  let g:changes = [
+  \ {'end': {'offset': 16, 'line': 2}, 'newText': "\n\ndef func_ivlpdpao(f):\n    exif = exifread.process_file(f)\n    dt = str(exif['Image DateTime'])\n    date = dt[:10].replace(':', '-')\n    return date\n", 'start': {'offset': 16, 'line': 2}},
+  \ {'end': {'offset': 32, 'line': 6}, 'newText': 'date = func', 'start': {'offset': 9, 'line': 6}},
+  \ {'end': {'offset': 42, 'line': 8}, 'newText': "ivlpdpao(f)\n", 'start': {'offset': 33, 'line': 6}}
+  \]
+
+  call ale#code_action#ApplyChanges(expand('%:p'), g:changes, 0)
+
+Expect(The changes should be applied correctly):
+  import sys
+  import exifread
+
+
+  def func_ivlpdpao(f):
+      exif = exifread.process_file(f)
+      dt = str(exif['Image DateTime'])
+      date = dt[:10].replace(':', '-')
+      return date
+
+
+  def main():
+      with open(sys.argv[1], 'rb') as f:
+          date = func_ivlpdpao(f)
diff --git a/test/test_codefix.vader b/test/test_codefix.vader
new file mode 100644
index 00000000..63275d7f
--- /dev/null
+++ b/test/test_codefix.vader
@@ -0,0 +1,604 @@
+Before:
+  call ale#test#SetDirectory('/testplugin/test')
+  call ale#test#SetFilename('dummy.txt')
+  Save g:ale_buffer_info
+
+  let g:ale_buffer_info = {}
+
+  let g:old_filename = expand('%:p')
+  let g:Callback = ''
+  let g:expr_list = []
+  let g:message_list = []
+  let g:handle_code_action_called = 0
+  let g:code_actions = []
+  let g:options = {}
+  let g:capability_checked = ''
+  let g:conn_id = v:null
+  let g:InitCallback = v:null
+
+  runtime autoload/ale/lsp_linter.vim
+  runtime autoload/ale/lsp.vim
+  runtime autoload/ale/util.vim
+  runtime autoload/ale/codefix.vim
+  runtime autoload/ale/code_action.vim
+
+  function! ale#lsp_linter#StartLSP(buffer, linter, Callback) abort
+    let g:conn_id = ale#lsp#Register('executable', '/foo/bar', {})
+    call ale#lsp#MarkDocumentAsOpen(g:conn_id, a:buffer)
+
+    if a:linter.lsp is# 'tsserver'
+        call ale#lsp#MarkConnectionAsTsserver(g:conn_id)
+    endif
+
+    let l:details = {
+    \ 'command': 'foobar',
+    \ 'buffer': a:buffer,
+    \ 'connection_id': g:conn_id,
+    \ 'project_root': '/foo/bar',
+    \}
+
+    let g:InitCallback = {-> ale#lsp_linter#OnInit(a:linter, l:details, a:Callback)}
+  endfunction
+
+  function! ale#lsp#HasCapability(conn_id, capability) abort
+    let g:capability_checked = a:capability
+
+    return 1
+  endfunction
+
+  function! ale#lsp#RegisterCallback(conn_id, callback) abort
+    let g:Callback = a:callback
+  endfunction
+
+  function! ale#lsp#Send(conn_id, message) abort
+    call add(g:message_list, a:message)
+
+    return 42
+  endfunction
+
+  function! ale#util#Execute(expr) abort
+    call add(g:expr_list, a:expr)
+  endfunction
+
+  function! ale#code_action#HandleCodeAction(code_action, options) abort
+    let g:handle_code_action_called = 1
+    Assert !get(a:options, 'should_save')
+    call add(g:code_actions, a:code_action)
+  endfunction
+
+  function! ale#util#Input(message, value) abort
+    return '2'
+  endfunction
+
+After:
+  Restore
+
+  if g:conn_id isnot v:null
+    call ale#lsp#RemoveConnectionWithID(g:conn_id)
+  endif
+
+  call ale#test#RestoreDirectory()
+  call ale#linter#Reset()
+
+  unlet! g:capability_checked
+  unlet! g:InitCallback
+  unlet! g:old_filename
+  unlet! g:conn_id
+  unlet! g:Callback
+  unlet! g:message_list
+  unlet! g:expr_list
+  unlet! b:ale_linters
+  unlet! g:options
+  unlet! g:code_actions
+  unlet! g:handle_code_action_called
+
+  runtime autoload/ale/lsp_linter.vim
+  runtime autoload/ale/lsp.vim
+  runtime autoload/ale/util.vim
+  runtime autoload/ale/codefix.vim
+  runtime autoload/ale/code_action.vim
+
+Execute(Failed codefix responses should be handled correctly):
+  call ale#codefix#HandleTSServerResponse(
+  \ 1,
+  \ {'command': 'getCodeFixes', 'request_seq': 3}
+  \)
+  AssertEqual g:handle_code_action_called, 0
+
+
+
+Given typescript(Some typescript file):
+  foo
+  somelongerline
+  bazxyzxyzxyz
+
+Execute(getCodeFixes from tsserver should be handled):
+  call ale#codefix#SetMap({3: {}})
+  call ale#codefix#HandleTSServerResponse(1, {
+    \ 'command': 'getCodeFixes',
+    \ 'request_seq': 3,
+    \ 'success': v:true,
+    \ 'type': 'response',
+    \ 'body': [
+    \   {
+    \     'description': 'Import default "x" from module "./z"',
+    \     'fixName': 'import',
+    \     'changes': [
+    \       {
+    \         'fileName': "/foo/bar/file1.ts",
+    \         'textChanges': [
+    \           {
+    \             'end': {
+    \               'line': 2,
+    \               'offset': 1,
+    \             },
+    \             'newText': 'import x from "./z";^@',
+    \             'start': {
+    \               'line': 2,
+    \               'offset': 1,
+    \             }
+    \           }
+    \         ]
+    \       }
+    \     ]
+    \   }
+    \ ]
+    \})
+
+  AssertEqual g:handle_code_action_called, 1
+  AssertEqual
+  \ [
+  \   {
+  \     'description': 'codefix',
+  \     'changes': [
+  \       {
+  \         'fileName': "/foo/bar/file1.ts",
+  \         'textChanges': [
+  \           {
+  \             'end': {
+  \               'line': 2,
+  \               'offset': 1
+  \             },
+  \             'newText': 'import x from "./z";^@',
+  \             'start': {
+  \               'line': 2,
+  \               'offset': 1
+  \             }
+  \           }
+  \         ]
+  \       }
+  \     ]
+  \   }
+  \ ],
+  \ g:code_actions
+
+Execute(getCodeFixes from tsserver should be handled with user input if there are more than one action):
+  call ale#codefix#SetMap({3: {}})
+  call ale#codefix#HandleTSServerResponse(1, {
+    \ 'command': 'getCodeFixes',
+    \ 'request_seq': 3,
+    \ 'success': v:true,
+    \ 'type': 'response',
+    \ 'body': [
+    \   {
+    \     'description': 'Import default "x" from module "./z"',
+    \     'fixName': 'import',
+    \     'changes': [
+    \       {
+    \         'fileName': "/foo/bar/file1.ts",
+    \         'textChanges': [
+    \           {
+    \             'end': {
+    \               'line': 2,
+    \               'offset': 1,
+    \             },
+    \             'newText': 'import x from "./z";^@',
+    \             'start': {
+    \               'line': 2,
+    \               'offset': 1,
+    \             }
+    \           }
+    \         ]
+    \       }
+    \     ]
+    \   },
+    \   {
+    \     'description': 'Import default "x" from module "./y"',
+    \     'fixName': 'import',
+    \     'changes': [
+    \       {
+    \         'fileName': "/foo/bar/file1.ts",
+    \         'textChanges': [
+    \           {
+    \             'end': {
+    \               'line': 2,
+    \               'offset': 1,
+    \             },
+    \             'newText': 'import x from "./y";^@',
+    \             'start': {
+    \               'line': 2,
+    \               'offset': 1,
+    \             }
+    \           }
+    \         ]
+    \       }
+    \     ]
+    \   }
+    \ ]
+    \})
+
+  AssertEqual g:handle_code_action_called, 1
+  AssertEqual
+  \ [
+  \   {
+  \     'description': 'codefix',
+  \     'changes': [
+  \       {
+  \         'fileName': "/foo/bar/file1.ts",
+  \         'textChanges': [
+  \           {
+  \             'end': {
+  \               'line': 2,
+  \               'offset': 1
+  \             },
+  \             'newText': 'import x from "./y";^@',
+  \             'start': {
+  \               'line': 2,
+  \               'offset': 1
+  \             }
+  \           }
+  \         ]
+  \       }
+  \     ]
+  \   }
+  \ ],
+  \ g:code_actions
+
+Execute(Prints a tsserver error message when getCodeFixes unsuccessful):
+  call ale#codefix#SetMap({3: {}})
+  call ale#codefix#HandleTSServerResponse(1, {
+  \ 'command': 'getCodeFixes',
+  \ 'request_seq': 3,
+  \ 'success': v:false,
+  \ 'message': 'something is wrong',
+  \})
+
+  AssertEqual g:handle_code_action_called, 0
+  AssertEqual ['echom ''Error while getting code fixes. Reason: something is wrong'''], g:expr_list
+
+Execute(Does nothing when where are no code fixes):
+  call ale#codefix#SetMap({3: {}})
+  call ale#codefix#HandleTSServerResponse(1, {
+  \ 'command': 'getCodeFixes',
+  \ 'request_seq': 3,
+  \ 'success': v:true,
+  \ 'body': []
+  \})
+
+  AssertEqual g:handle_code_action_called, 0
+  AssertEqual ['echom ''No code fixes available.'''], g:expr_list
+
+Execute(tsserver codefix requests should be sent):
+  call ale#linter#Reset()
+
+  runtime ale_linters/typescript/tsserver.vim
+  let g:ale_buffer_info = {bufnr(''): {'loclist': [{'lnum': 2, 'col': 5, 'code': 2304}]}}
+  call setpos('.', [bufnr(''), 2, 5, 0])
+
+  " ALECodeAction
+  call ale#codefix#Execute(0)
+
+  " We shouldn't register the callback yet.
+  AssertEqual '''''', string(g:Callback)
+
+  AssertEqual type(function('type')), type(g:InitCallback)
+  call g:InitCallback()
+
+  AssertEqual 'code_actions', g:capability_checked
+  AssertEqual
+  \ 'function(''ale#codefix#HandleTSServerResponse'')',
+  \ string(g:Callback)
+  AssertEqual
+  \ [
+  \   ale#lsp#tsserver_message#Change(bufnr('')),
+  \   [0, 'ts@getCodeFixes', {
+  \     'startLine': 2,
+  \     'startOffset': 5,
+  \     'endLine': 2,
+  \     'endOffset': 6,
+  \     'file': expand('%:p'),
+  \     'errorCodes': [2304],
+  \   }]
+  \ ],
+  \ g:message_list
+
+Execute(tsserver codefix requests should be sent only for error with code):
+  call ale#linter#Reset()
+
+  runtime ale_linters/typescript/tsserver.vim
+  let g:ale_buffer_info = {bufnr(''): {'loclist': [{'lnum': 2, 'col': 5}, {'lnum': 2, 'col': 5, 'code': 2304}]}}
+  call setpos('.', [bufnr(''), 2, 5, 0])
+
+  " ALECodeAction
+  call ale#codefix#Execute(0)
+
+  " We shouldn't register the callback yet.
+  AssertEqual '''''', string(g:Callback)
+
+  AssertEqual type(function('type')), type(g:InitCallback)
+  call g:InitCallback()
+
+  AssertEqual 'code_actions', g:capability_checked
+  AssertEqual
+  \ 'function(''ale#codefix#HandleTSServerResponse'')',
+  \ string(g:Callback)
+  AssertEqual
+  \ [
+  \   ale#lsp#tsserver_message#Change(bufnr('')),
+  \   [0, 'ts@getCodeFixes', {
+  \     'startLine': 2,
+  \     'startOffset': 5,
+  \     'endLine': 2,
+  \     'endOffset': 6,
+  \     'file': expand('%:p'),
+  \     'errorCodes': [2304],
+  \   }]
+  \ ],
+  \ g:message_list
+
+Execute(getApplicableRefactors from tsserver should be handled):
+  call ale#codefix#SetMap({3: {
+  \ 'buffer': expand('%:p'),
+  \ 'line': 1,
+  \ 'column': 2,
+  \ 'end_line': 3,
+  \ 'end_column': 4,
+  \ 'connection_id': 0,
+  \}})
+  call ale#codefix#HandleTSServerResponse(1,
+  \ {'seq': 0, 'request_seq': 3, 'type': 'response', 'success': v:true, 'body': [{'actions': [{'description': 'Extract to constant in enclosing scope', 'name': 'constant_scope_0'}], 'description': 'Extract constant', 'name': 'Extract Symbol'}, {'actions': [{'description': 'Extract to function in module scope', 'name': 'function_scope_1'}], 'description': 'Extract function', 'name': 'Extract Symbol'}], 'command': 'getApplicableRefactors'})
+
+  AssertEqual
+  \ [
+  \   [0, 'ts@getEditsForRefactor', {
+  \     'startLine': 1,
+  \     'startOffset': 2,
+  \     'endLine': 3,
+  \     'endOffset': 5,
+  \     'file': expand('%:p'),
+  \     'refactor': 'Extract Symbol',
+  \     'action': 'function_scope_1',
+  \   }]
+  \ ],
+  \ g:message_list
+
+Execute(getApplicableRefactors should print error on failure):
+  call ale#codefix#SetMap({3: {
+  \ 'buffer': expand('%:p'),
+  \ 'line': 1,
+  \ 'column': 2,
+  \ 'end_line': 3,
+  \ 'end_column': 4,
+  \ 'connection_id': 0,
+  \}})
+  call ale#codefix#HandleTSServerResponse(1,
+  \ {'seq': 0, 'request_seq': 3, 'type': 'response', 'success': v:false, 'message': 'oops', 'command': 'getApplicableRefactors'})
+
+  AssertEqual ['echom ''Error while getting applicable refactors. Reason: oops'''], g:expr_list
+
+Execute(getApplicableRefactors should do nothing if there are no refactors):
+  call ale#codefix#SetMap({3: {
+  \ 'buffer': expand('%:p'),
+  \ 'line': 1,
+  \ 'column': 2,
+  \ 'end_line': 3,
+  \ 'end_column': 4,
+  \ 'connection_id': 0,
+  \}})
+  call ale#codefix#HandleTSServerResponse(1,
+  \ {'seq': 0, 'request_seq': 3, 'type': 'response', 'success': v:true, 'body': [], 'command': 'getApplicableRefactors'})
+
+  AssertEqual ['echom ''No applicable refactors available.'''], g:expr_list
+
+Execute(getEditsForRefactor from tsserver should be handled):
+  call ale#codefix#SetMap({3: {}})
+  call ale#codefix#HandleTSServerResponse(1,
+  \{'seq': 0, 'request_seq': 3, 'type': 'response', 'success': v:true, 'body': {'edits': [{'fileName': '/foo/bar/file.ts', 'textChanges': [{'end': {'offset': 35, 'line': 9}, 'newText': 'newFunction(app);', 'start': {'offset': 3, 'line': 8}}, {'end': {'offset': 4, 'line': 19}, 'newText': '^@function newFunction(app: Router) {^@    app.use(booExpressCsrf());^@    app.use(booExpressRequireHttps);^@}^@', 'start': {'offset': 4, 'line': 19}}]}], 'renameLocation': {'offset': 3, 'line': 8}, 'renameFilename': '/foo/bar/file.ts'}, 'command': 'getEditsForRefactor' }
+  \)
+
+  AssertEqual g:handle_code_action_called, 1
+  AssertEqual
+  \ [
+  \   {
+  \     'description': 'editsForRefactor',
+  \     'changes': [{'fileName': '/foo/bar/file.ts', 'textChanges': [{'end': {'offset': 35, 'line': 9}, 'newText': 'newFunction(app);', 'start': {'offset': 3, 'line': 8}}, {'end': {'offset': 4, 'line': 19}, 'newText': '^@function newFunction(app: Router) {^@    app.use(booExpressCsrf());^@    app.use(booExpressRequireHttps);^@}^@', 'start': {'offset': 4, 'line': 19}}]}],
+  \   }
+  \ ],
+  \ g:code_actions
+
+Execute(getEditsForRefactor should print error on failure):
+  call ale#codefix#SetMap({3: {}})
+  call ale#codefix#HandleTSServerResponse(1,
+  \{'seq': 0, 'request_seq': 3, 'type': 'response', 'success': v:false, 'message': 'oops', 'command': 'getEditsForRefactor' }
+  \)
+
+  AssertEqual ['echom ''Error while getting edits for refactor. Reason: oops'''], g:expr_list
+
+" TODO: I can't figure out how to run ALECodeAction on range
+" in test function. Therefore I can't write properly working
+" test. If somebody knows how to do that help is appreciated.
+"
+" Execute(tsserver getApplicableRefactors requests should be sent):
+"   call ale#linter#Reset()
+"
+"   runtime ale_linters/typescript/tsserver.vim
+"   let g:ale_buffer_info = {bufnr(''): {'loclist': []}}
+"   call setpos('.', [bufnr(''), 2, 5, 0])
+"   normal "v$"
+"
+"   execute "ALECodeAction"
+"
+"   " We shouldn't register the callback yet.
+"   AssertEqual '''''', string(g:Callback)
+"
+"   AssertEqual type(function('type')), type(g:InitCallback)
+"   call g:InitCallback()
+"
+"   AssertEqual 'code_actions', g:capability_checked
+"   AssertEqual
+"   \ 'function(''ale#codefix#HandleTSServerResponse'')',
+"   \ string(g:Callback)
+"   AssertEqual
+"   \ [
+"   \   ale#lsp#tsserver_message#Change(bufnr('')),
+"   \   [0, 'ts@getApplicableRefactors', {
+"   \     'startLine': 2,
+"   \     'startOffset': 5,
+"   \     'endLine': 2,
+"   \     'endOffset': 15,
+"   \     'file': expand('%:p'),
+"   \   }]
+"   \ ],
+"   \ g:message_list
+
+Execute(Failed LSP responses should be handled correctly):
+  call ale#codefix#HandleLSPResponse(
+  \ 1,
+  \ {'method': 'workspace/applyEdit', 'request_seq': 3}
+  \)
+  AssertEqual g:handle_code_action_called, 0
+
+Given python(Some python file):
+  def main():
+      a = 1
+      b = a + 2
+
+Execute("workspace/applyEdit" from LSP should be handled):
+  call ale#codefix#SetMap({3: {}})
+  call ale#codefix#HandleLSPResponse(1,
+  \ {'id': 0, 'jsonrpc': '2.0', 'method': 'workspace/applyEdit', 'params': {'edit': {'changes': {'file:///foo/bar/file.ts': [{'range': {'end': {'character': 27, 'line': 7}, 'start': {'character': 27, 'line': 7}}, 'newText': ', Config'}, {'range': {'end': {'character': 12, 'line': 96}, 'start': {'character': 2, 'line': 94}}, 'newText': 'await newFunction(redis, imageKey, cover, config);'}, {'range': {'end': {'character': 2, 'line': 99}, 'start': {'character': 2, 'line': 99}}, 'newText': '^@async function newFunction(redis: IRedis, imageKey: string, cover: Buffer, config: Config) {^@    try {^@        await redis.set(imageKey, cover, ''ex'', parseInt(config.coverKeyTTL, 10));^@    }^@    catch { }^@}^@'}]}}}})
+
+  AssertEqual g:handle_code_action_called, 1
+  AssertEqual
+  \ [{'description': 'applyEdit', 'changes': [{'fileName': '/foo/bar/file.ts', 'textChanges': [{'end': {'offset': 28, 'line': 8}, 'newText': ', Config', 'start': {'offset': 28, 'line': 8}}, {'end': {'offset': 13, 'line': 97}, 'newText': 'await newFunction(redis, imageKey, cover, config);', 'start': {'offset': 3, 'line': 95}}, {'end': {'offset': 3, 'line': 100}, 'newText': '^@async function newFunction(redis: IRedis, imageKey: string, cover: Buffer, config: Config) {^@    try {^@        await redis.set(imageKey, cover, ''ex'', parseInt(config.coverKeyTTL, 10));^@    }^@    catch { }^@}^@', 'start': {'offset': 3, 'line': 100}}]}]}],
+  \ g:code_actions
+
+Execute(Code Actions from LSP should be handled with user input if there are more than one action):
+  call ale#codefix#SetMap({2: {}})
+  call ale#codefix#HandleLSPResponse(1,
+  \ {'id': 2, 'jsonrpc': '2.0', 'result': [{'title': 'fake for testing'}, {'arguments': [{'documentChanges': [{'edits': [{'range': {'end': {'character': 31, 'line': 2}, 'start': {'character': 31, 'line': 2}}, 'newText': ', createVideo'}], 'textDocument': {'uri': 'file:///foo/bar/file.ts', 'version': 1}}]}], 'title': 'Add ''createVideo'' to existing import declaration from "./video"', 'command': '_typescript.applyWorkspaceEdit'}]})
+
+  AssertEqual g:handle_code_action_called, 1
+  AssertEqual
+  \ [{'description': 'codeaction', 'changes': [{'fileName': '/foo/bar/file.ts', 'textChanges': [{'end': {'offset': 32, 'line': 3}, 'newText': ', createVideo', 'start': {'offset': 32, 'line': 3}}]}]}],
+  \ g:code_actions
+
+Execute(Code Actions from LSP should be handled when returned with documentChanges):
+  call ale#codefix#SetMap({2: {}})
+  call ale#codefix#HandleLSPResponse(1,
+  \ {'id': 2, 'jsonrpc': '2.0', 'result': [{'diagnostics': v:null, 'edit': {'changes': v:null, 'documentChanges': [{'edits': [{'range': {'end': {'character': 4, 'line': 2}, 'start': {'character': 4, 'line': 1}}, 'newText': ''}, {'range': {'end': {'character': 9, 'line': 2}, 'start': {'character': 8, 'line': 2}}, 'newText': '(1)'}], 'textDocument': {'uri': 'file:///foo/bar/test.py', 'version': v:null}}]}, 'kind': 'refactor.inline', 'title': 'Inline variable', 'command': v:null}, {'diagnostics': v:null, 'edit': {'changes': v:null, 'documentChanges': [{'edits': [{'range': {'end': {'character': 0, 'line': 0}, 'start': {'character': 0, 'line': 0}}, 'newText': 'def func_bomdjnxh():^@    a = 1return a^@^@^@'}, {'range': {'end': {'character': 9, 'line': 1}, 'start': {'character': 8, 'line': 1}}, 'newText': 'func_bomdjnxh()^@'}], 'textDocument': {'uri': 'file:///foo/bar/test.py', 'version': v:null}}]}, 'kind': 'refactor.extract', 'title': 'Extract expression into function ''func_bomdjnxh''', 'command': v:null}]})
+
+  AssertEqual g:handle_code_action_called, 1
+  AssertEqual
+  \ [{'description': 'codeaction', 'changes': [{'fileName': '/foo/bar/test.py', 'textChanges': [{'end': {'offset': 1, 'line': 1}, 'newText': 'def func_bomdjnxh():^@    a = 1return a^@^@^@', 'start': {'offset': 1, 'line': 1}}, {'end': {'offset': 10, 'line': 2}, 'newText': 'func_bomdjnxh()^@', 'start': {'offset': 9, 'line': 2}}]}]}],
+  \ g:code_actions
+
+Execute(LSP Code Actions handles command responses):
+  call ale#codefix#SetMap({3: {
+  \ 'connection_id': 0,
+  \}})
+  call ale#codefix#HandleLSPResponse(1,
+  \ {'id': 3, 'jsonrpc': '2.0', 'result': [{'kind': 'refactor', 'title': 'Extract to inner function in function ''getVideo''', 'command': {'arguments': [{'file': '/foo/bar/file.ts', 'endOffset': 0, 'action': 'function_scope_0', 'startOffset': 1, 'startLine': 65, 'refactor': 'Extract Symbol', 'endLine': 68}], 'title': 'Extract to inner function in function ''getVideo''', 'command': '_typescript.applyRefactoring'}}, {'kind': 'refactor', 'title': 'Extract to function in module scope', 'command': {'arguments': [{'file': '/foo/bar/file.ts', 'endOffset': 0, 'action': 'function_scope_1', 'startOffset': 1, 'startLine': 65, 'refactor': 'Extract Symbol', 'endLine': 68}], 'title': 'Extract to function in module scope', 'command': '_typescript.applyRefactoring'}}]})
+
+  AssertEqual
+  \ [[0, 'workspace/executeCommand', {'arguments': [{'file': '/foo/bar/file.ts', 'action': 'function_scope_1', 'endOffset': 0, 'refactor': 'Extract Symbol', 'endLine': 68, 'startLine': 65, 'startOffset': 1}], 'command': '_typescript.applyRefactoring'}]],
+  \ g:message_list
+
+
+Execute(Prints message when LSP code action returns no results):
+  call ale#codefix#SetMap({3: {}})
+  call ale#codefix#HandleLSPResponse(1,
+  \ {'id': 3, 'jsonrpc': '2.0', 'result': []})
+
+  AssertEqual g:handle_code_action_called, 0
+  AssertEqual ['echom ''No code actions received from server'''], g:expr_list
+
+Execute(LSP code action requests should be sent):
+  call ale#linter#Reset()
+
+  runtime ale_linters/python/jedils.vim
+  let g:ale_buffer_info = {bufnr(''): {'loclist': [{'lnum': 2, 'col': 5, 'end_lnum': 2, 'end_col': 6, 'code': 2304, 'text': 'oops'}]}}
+  call setpos('.', [bufnr(''), 2, 5, 0])
+
+  " ALECodeAction
+  call ale#codefix#Execute(0)
+
+  " We shouldn't register the callback yet.
+  AssertEqual '''''', string(g:Callback)
+
+  AssertEqual type(function('type')), type(g:InitCallback)
+  call g:InitCallback()
+
+  AssertEqual 'code_actions', g:capability_checked
+  AssertEqual
+  \ 'function(''ale#codefix#HandleLSPResponse'')',
+  \ string(g:Callback)
+  AssertEqual
+  \ [
+  \   [1, 'workspace/didChangeConfiguration', {'settings': {'python': {}}}],
+  \   [1, 'textDocument/didChange', {
+  \     'contentChanges': [{'text': "def main():\n    a = 1\n    b = a + 2\n"}],
+  \     'textDocument': {
+  \         'uri': ale#path#ToURI(expand('%:p')),
+  \         'version': g:ale_lsp_next_version_id - 1,
+  \     },
+  \   }],
+  \   [0, 'textDocument/codeAction', {
+  \     'context': {
+  \       'diagnostics': [{'range': {'end': {'character': 5, 'line': 1}, 'start': {'character': 4, 'line': 1}}, 'code': 2304, 'message': 'oops'}]
+  \     },
+  \     'range': {'end': {'character': 5, 'line': 1}, 'start': {'character': 4, 'line': 1}},
+  \     'textDocument': {'uri': ale#path#ToURI(expand('%:p'))}
+  \   }]
+  \ ],
+  \ g:message_list
+
+Execute(LSP code action requests should be sent only for error with code):
+  call ale#linter#Reset()
+
+  runtime ale_linters/python/jedils.vim
+  let g:ale_buffer_info = {bufnr(''): {'loclist': [{'lnum': 2, 'col': 5, 'end_lnum': 2, 'end_col': 6, 'code': 2304, 'text': 'oops'}]}}
+  call setpos('.', [bufnr(''), 2, 5, 0])
+
+  " ALECodeAction
+  call ale#codefix#Execute(0)
+
+  " We shouldn't register the callback yet.
+  AssertEqual '''''', string(g:Callback)
+
+  AssertEqual type(function('type')), type(g:InitCallback)
+  call g:InitCallback()
+
+  AssertEqual 'code_actions', g:capability_checked
+  AssertEqual
+  \ 'function(''ale#codefix#HandleLSPResponse'')',
+  \ string(g:Callback)
+  AssertEqual
+  \ [
+  \   [1, 'workspace/didChangeConfiguration', {'settings': {'python': {}}}],
+  \   [1, 'textDocument/didChange', {
+  \     'contentChanges': [{'text': "def main():\n    a = 1\n    b = a + 2\n"}],
+  \     'textDocument': {
+  \         'uri': ale#path#ToURI(expand('%:p')),
+  \         'version': g:ale_lsp_next_version_id - 1,
+  \     },
+  \   }],
+  \   [0, 'textDocument/codeAction', {
+  \     'context': {
+  \       'diagnostics': [{'range': {'end': {'character': 5, 'line': 1}, 'start': {'character': 4, 'line': 1}}, 'code': 2304, 'message': 'oops'}]
+  \     },
+  \     'range': {'end': {'character': 5, 'line': 1}, 'start': {'character': 4, 'line': 1}},
+  \     'textDocument': {'uri': ale#path#ToURI(expand('%:p'))}
+  \   }]
+  \ ],
+  \ g:message_list