1 files changed, 30 insertions, 0 deletions
diff --git a/util_test.lua b/util_test.lua
index 63393b1..99f34f0 100755
--- a/util_test.lua
+++ b/util_test.lua
@@ -73,4 +73,34 @@ function test_table_merge() -- luacheck: ignore 111
   lunatest.assert_equal(t.foo.bar.nose, 'nose')
 end
 
+function test_style_define() -- luacheck: ignore 111
+  -- Modern vis: vis.ui.style_define present
+  vis.API = 0
+  vis.ui = {
+    style_define = function(self, id, style)
+      return id == 1 and style == 'fore:red'
+    end,
+  }
+  lunatest.assert_true(util.style_define(1, 'fore:red'))
+
+  -- Modern vis: vis.API > 3215
+  vis.API = 3216
+  lunatest.assert_true(util.style_define(1, 'fore:red'))
+
+  -- Older vis: fallback with explicit win
+  vis.API = nil
+  vis.ui = nil
+  local win = {
+    style_define = function(self, id, style)
+      return id == 2 and style == 'fore:blue'
+    end,
+  }
+  lunatest.assert_true(util.style_define(2, 'fore:blue', win))
+
+  -- Older vis: fallback with vis.win
+  vis.win = win
+  lunatest.assert_true(util.style_define(2, 'fore:blue'))
+  vis.win = nil
+end
+
 lunatest.run()