Fix ansible ftdetect

2 files changed, 23 insertions, 7 deletions

diff --git a/scripts/test b/scripts/test
index 084d084a..88a2e973 100755
--- a/scripts/test
+++ b/scripts/test
@@ -2,10 +2,12 @@
 
 set -e
 
-vim --clean -N -u <(
-  echo "filetype plugin indent on"
-  echo "syntax enable"
-  echo "let &rtp='$PWD,'.&rtp"
-  echo "source scripts/test_filetypes.vim"
-  echo "exec ':q!'"
-)
+vim --clean -N --cmd "
+  filetype plugin indent on
+  syntax enable
+  let &rtp='$PWD,'.&rtp
+  source ftdetect/polyglot.vim
+  source scripts/test_extensions.vim
+  source scripts/test_filetypes.vim
+  exec ':q!'
+"
diff --git a/scripts/test_extensions.vim b/scripts/test_extensions.vim
new file mode 100644
index 00000000..aea52f94
--- /dev/null
+++ b/scripts/test_extensions.vim
@@ -0,0 +1,14 @@
+function! TestExtension(filetype, filename, content)
+  try
+    exec "e " . a:filename
+    exec "if &filetype != '" . a:filetype . "' \nthrow &filetype\nendif"
+  catch
+    echo 'Filename "' . a:filename  . '" does not resolve to extension "' . a:filetype . '"'
+    echo '  instead received: "' . v:exception . '"'
+    exec ':cq!'
+  endtry
+endfunction
+
+call TestExtension('blade', 'test.blade.php', '')
+call TestExtension('yaml.ansible', 'playbook.yml', '')
+call TestExtension('yaml.ansible', 'host_vars/foobar', '')