From 1e95dc530a88524a9061eeca36dfa788c71042d2 Mon Sep 17 00:00:00 2001
From: Henrik Lissner <henrik@lissner.net>
Date: Fri, 6 Oct 2017 02:35:26 +0200
Subject: [PATCH] Add zunit file template & file extension

---
 modules/feature/file-templates/config.el                 | 3 ++-
 modules/feature/file-templates/templates/sh-mode/__zunit | 5 +++++
 modules/lang/sh/config.el                                | 1 +
 3 files changed, 8 insertions(+), 1 deletion(-)
 create mode 100644 modules/feature/file-templates/templates/sh-mode/__zunit

diff --git a/modules/feature/file-templates/config.el b/modules/feature/file-templates/config.el
index fc7ed642d..8c55c349c 100644
--- a/modules/feature/file-templates/config.el
+++ b/modules/feature/file-templates/config.el
@@ -106,5 +106,6 @@
           ;; Slim
           ("/\\(index\\|main\\)\\.slim$"     "__"               slim-mode)
           ;; Shell scripts
-          ("\\.z?sh$"                        "__"               sh-mode))))
+          ("\\.z?sh$"                        "__"               sh-mode)
+          ("\\.zunit$"                       "__zunit"          sh-mode))))
 
diff --git a/modules/feature/file-templates/templates/sh-mode/__zunit b/modules/feature/file-templates/templates/sh-mode/__zunit
new file mode 100644
index 000000000..6abced490
--- /dev/null
+++ b/modules/feature/file-templates/templates/sh-mode/__zunit
@@ -0,0 +1,5 @@
+#!/usr/bin/env zunit
+
+@test '...' {
+    $0
+}
diff --git a/modules/lang/sh/config.el b/modules/lang/sh/config.el
index 0c06ab1f3..f5c93a8bc 100644
--- a/modules/lang/sh/config.el
+++ b/modules/lang/sh/config.el
@@ -14,6 +14,7 @@
 
 (def-package! sh-script ; built-in
   :mode ("\\.zsh$"   . sh-mode)
+  :mode ("\\.zunit$" . sh-mode)
   :mode ("/bspwmrc$" . sh-mode)
   :init
   (add-hook! sh-mode #'(flycheck-mode highlight-numbers-mode))