1 files changed, 57 insertions, 40 deletions
diff --git a/test/functional/terminal/tui_spec.lua b/test/functional/terminal/tui_spec.lua
index defa700f3b..a8a49cd378 100644
--- a/test/functional/terminal/tui_spec.lua
+++ b/test/functional/terminal/tui_spec.lua
@@ -871,9 +871,9 @@ describe('TUI background color', function()
     screen:expect{any='did OptionSet, yay!'}
   end)
 
-  local function assert_bg(color, bg)
+  local function assert_bg(colorspace, color, bg)
     it('handles '..color..' as '..bg, function()
-      feed_data('\027]11;rgb:'..color..'\007')
+      feed_data('\027]11;'..colorspace..':'..color..'\007')
       -- Retry until the terminal response is handled.
       retry(100, nil, function()
         feed_data(':echo &background\n')
@@ -893,42 +893,59 @@ describe('TUI background color', function()
     end)
   end
 
-  assert_bg('0000/0000/0000', 'dark')
-  assert_bg('ffff/ffff/ffff', 'light')
-  assert_bg('000/000/000', 'dark')
-  assert_bg('fff/fff/fff', 'light')
-  assert_bg('00/00/00', 'dark')
-  assert_bg('ff/ff/ff', 'light')
-  assert_bg('0/0/0', 'dark')
-  assert_bg('f/f/f', 'light')
-
-  assert_bg('f/0/0', 'dark')
-  assert_bg('0/f/0', 'light')
-  assert_bg('0/0/f', 'dark')
-
-  assert_bg('1/1/1', 'dark')
-  assert_bg('2/2/2', 'dark')
-  assert_bg('3/3/3', 'dark')
-  assert_bg('4/4/4', 'dark')
-  assert_bg('5/5/5', 'dark')
-  assert_bg('6/6/6', 'dark')
-  assert_bg('7/7/7', 'dark')
-  assert_bg('8/8/8', 'light')
-  assert_bg('9/9/9', 'light')
-  assert_bg('a/a/a', 'light')
-  assert_bg('b/b/b', 'light')
-  assert_bg('c/c/c', 'light')
-  assert_bg('d/d/d', 'light')
-  assert_bg('e/e/e', 'light')
-
-  assert_bg('0/e/0', 'light')
-  assert_bg('0/d/0', 'light')
-  assert_bg('0/c/0', 'dark')
-  assert_bg('0/b/0', 'dark')
-
-  assert_bg('f/0/f', 'dark')
-  assert_bg('f/1/f', 'dark')
-  assert_bg('f/2/f', 'dark')
-  assert_bg('f/3/f', 'light')
-  assert_bg('f/4/f', 'light')
+  assert_bg('rgb', '0000/0000/0000', 'dark')
+  assert_bg('rgb', 'ffff/ffff/ffff', 'light')
+  assert_bg('rgb', '000/000/000', 'dark')
+  assert_bg('rgb', 'fff/fff/fff', 'light')
+  assert_bg('rgb', '00/00/00', 'dark')
+  assert_bg('rgb', 'ff/ff/ff', 'light')
+  assert_bg('rgb', '0/0/0', 'dark')
+  assert_bg('rgb', 'f/f/f', 'light')
+
+  assert_bg('rgb', 'f/0/0', 'dark')
+  assert_bg('rgb', '0/f/0', 'light')
+  assert_bg('rgb', '0/0/f', 'dark')
+
+  assert_bg('rgb', '1/1/1', 'dark')
+  assert_bg('rgb', '2/2/2', 'dark')
+  assert_bg('rgb', '3/3/3', 'dark')
+  assert_bg('rgb', '4/4/4', 'dark')
+  assert_bg('rgb', '5/5/5', 'dark')
+  assert_bg('rgb', '6/6/6', 'dark')
+  assert_bg('rgb', '7/7/7', 'dark')
+  assert_bg('rgb', '8/8/8', 'light')
+  assert_bg('rgb', '9/9/9', 'light')
+  assert_bg('rgb', 'a/a/a', 'light')
+  assert_bg('rgb', 'b/b/b', 'light')
+  assert_bg('rgb', 'c/c/c', 'light')
+  assert_bg('rgb', 'd/d/d', 'light')
+  assert_bg('rgb', 'e/e/e', 'light')
+
+  assert_bg('rgb', '0/e/0', 'light')
+  assert_bg('rgb', '0/d/0', 'light')
+  assert_bg('rgb', '0/c/0', 'dark')
+  assert_bg('rgb', '0/b/0', 'dark')
+
+  assert_bg('rgb', 'f/0/f', 'dark')
+  assert_bg('rgb', 'f/1/f', 'dark')
+  assert_bg('rgb', 'f/2/f', 'dark')
+  assert_bg('rgb', 'f/3/f', 'light')
+  assert_bg('rgb', 'f/4/f', 'light')
+
+  assert_bg('rgba', '0000/0000/0000/0000', 'dark')
+  assert_bg('rgba', '0000/0000/0000/ffff', 'dark')
+  assert_bg('rgba', 'ffff/ffff/ffff/0000', 'light')
+  assert_bg('rgba', 'ffff/ffff/ffff/ffff', 'light')
+  assert_bg('rgba', '000/000/000/000', 'dark')
+  assert_bg('rgba', '000/000/000/fff', 'dark')
+  assert_bg('rgba', 'fff/fff/fff/000', 'light')
+  assert_bg('rgba', 'fff/fff/fff/fff', 'light')
+  assert_bg('rgba', '00/00/00/00', 'dark')
+  assert_bg('rgba', '00/00/00/ff', 'dark')
+  assert_bg('rgba', 'ff/ff/ff/00', 'light')
+  assert_bg('rgba', 'ff/ff/ff/ff', 'light')
+  assert_bg('rgba', '0/0/0/0', 'dark')
+  assert_bg('rgba', '0/0/0/f', 'dark')
+  assert_bg('rgba', 'f/f/f/0', 'light')
+  assert_bg('rgba', 'f/f/f/f', 'light')
 end)