Add Unit Tests for OSC ForegroundColor and BackgroundColor

microsoft · May 22, 2019 · 94ec7f9 · 94ec7f9
1 parent b11e897
commit 94ec7f9
Showing 1 changed file with 58 additions and 0 deletions.
diff --git a/src/host/ut_host/ScreenBufferTests.cpp b/src/host/ut_host/ScreenBufferTests.cpp
@@ -149,6 +149,10 @@ class ScreenBufferTests
 
     TEST_METHOD(SetColorTableThreeDigits);
 
+    TEST_METHOD(SetDefaultForegroundColor);
+
+    TEST_METHOD(SetDefaultBackgroundColor);
+
     TEST_METHOD(DeleteCharsNearEndOfLine);
     TEST_METHOD(DeleteCharsNearEndOfLineSimpleFirstCase);
     TEST_METHOD(DeleteCharsNearEndOfLineSimpleSecondCase);
@@ -2502,6 +2506,60 @@ void ScreenBufferTests::SetColorTableThreeDigits()
 
 }
 
+
+void ScreenBufferTests::SetDefaultForegroundColor()
+{
+    // Setting the default foreground color should work
+
+    CONSOLE_INFORMATION& gci = ServiceLocator::LocateGlobals().getConsoleInformation();
+    gci.LockConsole(); // Lock must be taken to swap buffers.
+    auto unlock = wil::scope_exit([&] { gci.UnlockConsole(); });
+
+    SCREEN_INFORMATION& mainBuffer = gci.GetActiveOutputBuffer();
+    VERIFY_IS_FALSE(mainBuffer._IsAltBuffer());
+    WI_SetFlag(mainBuffer.OutputMode, ENABLE_VIRTUAL_TERMINAL_PROCESSING);
+    VERIFY_IS_TRUE(WI_IsFlagSet(mainBuffer.OutputMode, ENABLE_VIRTUAL_TERMINAL_PROCESSING));
+
+    StateMachine& stateMachine = mainBuffer.GetStateMachine();
+
+    const COLORREF originalColor = gci.GetDefaultForegroundColor();
+    const COLORREF testColor = RGB(0x33, 0x66, 0x99);
+    VERIFY_ARE_NOT_EQUAL(originalColor, testColor);
+
+    std::wstring seq = L"\x1b]10;rgb:33/66/99\x1b\\";
+    stateMachine.ProcessString(seq);
+
+    VERIFY_ARE_EQUAL(testColor, gci.GetDefaultForegroundColor());
+}
+
+
+void ScreenBufferTests::SetDefaultBackgroundColor()
+{
+    // Setting the default foreground color should work
+
+    CONSOLE_INFORMATION& gci = ServiceLocator::LocateGlobals().getConsoleInformation();
+    gci.LockConsole(); // Lock must be taken to swap buffers.
+    auto unlock = wil::scope_exit([&] { gci.UnlockConsole(); });
+
+    SCREEN_INFORMATION& mainBuffer = gci.GetActiveOutputBuffer();
+    VERIFY_IS_FALSE(mainBuffer._IsAltBuffer());
+    WI_SetFlag(mainBuffer.OutputMode, ENABLE_VIRTUAL_TERMINAL_PROCESSING);
+    VERIFY_IS_TRUE(WI_IsFlagSet(mainBuffer.OutputMode, ENABLE_VIRTUAL_TERMINAL_PROCESSING));
+
+    StateMachine& stateMachine = mainBuffer.GetStateMachine();
+
+    const COLORREF originalColor = gci.GetDefaultBackgroundColor();
+    const COLORREF testColor = RGB(0x33, 0x66, 0x99);
+    VERIFY_ARE_NOT_EQUAL(originalColor, testColor);
+
+    std::wstring seq = L"\x1b]11;rgb:33/66/99\x1b\\";
+    stateMachine.ProcessString(seq);
+
+    VERIFY_ARE_EQUAL(testColor, gci.GetDefaultBackgroundColor());
+}
+
+
+
 void ScreenBufferTests::DeleteCharsNearEndOfLine()
 {
     // Created for MSFT:19888564.