Merge pull request #58 from NoahTheDuke/ftdetect-tests

Add ftdetect tests

tests/Cargo.toml

@@ -3,11 +3,19 @@ name    = "tests"
 version = "0.0.0"
 authors = ["Casey Rodarmor <[email protected]>", "Noah Bogart <[email protected]"]
 edition = "2021"
+default-run = "tests"
+
+[[bin]]
+name = "test-ftdetect"
+path = "src/test-ftdetect.rs"
 
 [dependencies]
-regex     = "~1.9"
-scraper   = "0.17.1"
-clap      = { version = "4", features = ["derive"] }
-tempfile  = "3.8.0"
-ctrlc     = "3.4.1"
+regex        = "~1.9"
+scraper      = "0.17.1"
+clap         = { version = "4", features = ["derive"] }
+tempfile     = "3.8.0"
+ctrlc        = "3.4.1"
 wait-timeout = "0.2.0"
+serde        = { version = "1.0.190", features = ["derive"] }
+serde_json   = "1.0.108"
+rand         = "0.8.5"

tests/batch_ftdetect_res.vim

@@ -0,0 +1,3 @@
+bufdo  redir >> $OUTPUT | echo @% | set ft? | redir END
+
+qa

tests/cases/ftdetect.json

@@ -0,0 +1,24 @@
+[
+  {
+    "filename": "justfile"
+  },
+  {
+    "filename": "Justfile"
+  },
+  {
+    "filename": "*.just"
+  },
+  {
+    "content": "#!/usr/bin/env -S just -f\n"
+  },
+  {
+    "content": "#!/usr/bin/env just -f\n"
+  },
+  {
+    "content": "#!/usr/local/bin/just -f\n"
+  },
+  {
+    "content": "",
+    "not_justfile": true
+  }
+]

tests/justfile

@@ -13,6 +13,10 @@ watch FILTER='':
     --exec "run {{FILTER}}" \
     --watch ..
 
+# run ftdetect tests
+ftdetect:
+  cargo run --bin=test-ftdetect
+
 # install development dependencies
 deps:
   cargo install cargo-watch

tests/src/common.rs

@@ -0,0 +1,32 @@
+use std::{
+  fs,
+  os::unix::fs as ufs,
+  sync::{
+    atomic::{AtomicBool, Ordering::Relaxed},
+    Arc,
+  },
+};
+
+pub fn clean_dotvim_symlink() {
+  if fs::metadata(".vim").is_ok() {
+    fs::remove_file(".vim").unwrap();
+  }
+}
+
+pub fn create_dotvim_symlink() {
+  clean_dotvim_symlink();
+  ufs::symlink("../", ".vim").unwrap();
+}
+
+pub fn setup_ctrlc_handler() -> Arc<AtomicBool> {
+  let interrupted = Arc::new(AtomicBool::new(false));
+  let _interrupted = Arc::clone(&interrupted);
+
+  ctrlc::set_handler(move || {
+    _interrupted.store(true, Relaxed);
+    eprintln!("Received Ctrl+C");
+  })
+  .unwrap();
+
+  interrupted
+}

tests/src/main.rs

@@ -1,3 +1,6 @@
+mod common;
+use crate::common::*;
+
 use clap::Parser;
 use regex::Regex;
 use scraper::{Html, Selector};
@@ -6,13 +9,9 @@ use std::{
   ffi::OsStr,
   fs,
   io::{self, ErrorKind},
-  os::unix::fs as ufs,
   path::Path,
   process::{Command, Stdio},
-  sync::{
-    atomic::{AtomicBool, Ordering::Relaxed},
-    Arc,
-  },
+  sync::atomic::Ordering::Relaxed,
   time::Duration,
 };
 use wait_timeout::ChildExt;
@@ -26,23 +25,13 @@ struct Arguments {
 fn _main() -> io::Result<()> {
   let arguments = Arguments::parse();
 
-  let interrupted = Arc::new(AtomicBool::new(false));
-  let _interrupted = Arc::clone(&interrupted);
-
-  ctrlc::set_handler(move || {
-    _interrupted.store(true, Relaxed);
-    eprintln!("Received Ctrl+C");
-  })
-  .unwrap();
+  let interrupted = setup_ctrlc_handler();
 
   let tempdir = tempfile::tempdir().unwrap();
 
   let case_dir = Path::new("cases");
 
-  if fs::metadata(".vim").is_ok() {
-    fs::remove_file(".vim").unwrap();
-  }
-  ufs::symlink("../", ".vim").unwrap();
+  create_dotvim_symlink();
 
   let mut cases = 0;
   let mut passed = 0;
@@ -177,9 +166,7 @@ fn main() -> io::Result<()> {
   let real_main = _main();
 
   // cleanup
-  if fs::metadata(".vim").is_ok() {
-    fs::remove_file(".vim").unwrap();
-  }
+  clean_dotvim_symlink();
 
   real_main
 }