[APOTESTS]

[reactos.git] / rostests / apitests / win32nt / ntgdi / NtGdiRestoreDC.c

diff --git a/rostests/apitests/win32nt/ntgdi/NtGdiRestoreDC.c b/rostests/apitests/win32nt/ntgdi/NtGdiRestoreDC.c
new file mode 100644 (file)
index 0000000..cb5640d
--- /dev/null
+++ b/rostests/apitests/win32nt/ntgdi/NtGdiRestoreDC.c
@@ -0,0 +1,195 @@
+/*
+ * PROJECT:         ReactOS api tests
+ * LICENSE:         GPL - See COPYING in the top level directory
+ * PURPOSE:         Test for NtGdiRestoreDC
+ * PROGRAMMERS:
+ */
+
+#include <win32nt.h>
+
+static HBRUSH hbrush;
+static HBITMAP hbitmap;
+static HPEN hpen;
+static HFONT hfont;
+static HRGN hrgn, hrgn2;
+
+static
+void
+SetSpecialDCState(HDC hdc)
+{
+    /* Select spcial Objects */
+    SelectObject(hdc, hbrush);
+    SelectObject(hdc, hpen);
+    SelectObject(hdc, hbitmap);
+    SelectObject(hdc, hfont);
+    SelectObject(hdc, hrgn);
+
+    /* Colors */
+    SetDCBrushColor(hdc, RGB(12,34,56));
+    SetDCPenColor(hdc, RGB(23,34,45));
+
+    /* Coordinates */
+    SetMapMode(hdc, MM_ANISOTROPIC);
+    SetGraphicsMode(hdc, GM_ADVANCED);
+    SetWindowOrgEx(hdc, 12, 34, NULL);
+    SetViewportOrgEx(hdc, 56, 78, NULL);
+    SetWindowExtEx(hdc, 123, 456, NULL);
+    SetViewportExtEx(hdc, 234, 567, NULL);
+
+
+
+}
+
+static
+void
+SetSpecialDCState2(HDC hdc)
+{
+    /* Select spcial Objects */
+    SelectObject(hdc, GetStockObject(DC_BRUSH));
+    SelectObject(hdc, GetStockObject(DC_PEN));
+    SelectObject(hdc, GetStockObject(DEFAULT_BITMAP));
+    SelectObject(hdc, GetStockObject(DEFAULT_GUI_FONT));
+    SelectObject(hdc, hrgn2);
+
+    /* Colors */
+    SetDCBrushColor(hdc, RGB(65,43,21));
+    SetDCPenColor(hdc, RGB(54,43,32));
+
+    /* Coordinates */
+    SetMapMode(hdc, MM_ISOTROPIC);
+    SetGraphicsMode(hdc, GM_COMPATIBLE);
+    SetWindowOrgEx(hdc, 43, 21, NULL);
+    SetViewportOrgEx(hdc, 87, 65, NULL);
+    SetWindowExtEx(hdc, 654, 321, NULL);
+    SetViewportExtEx(hdc, 765, 432, NULL);
+
+
+}
+
+static
+void
+Test_IsSpecialState(HDC hdc, BOOL bMemDC)
+{
+    POINT pt;
+    SIZE sz;
+
+    /* Test Objects */
+    TEST(SelectObject(hdc, GetStockObject(DC_BRUSH)) == hbrush);
+    TEST(SelectObject(hdc, GetStockObject(DC_PEN)) == hpen);
+    TEST(SelectObject(hdc, GetStockObject(DEFAULT_GUI_FONT)) == hfont);
+    if (bMemDC)
+    {
+        TEST(SelectObject(hdc, GetStockObject(DEFAULT_BITMAP)) == hbitmap);
+        TEST(SelectObject(hdc, hrgn2) == (PVOID)1);
+    }
+    else
+    {
+        TEST(SelectObject(hdc, GetStockObject(DEFAULT_BITMAP)) == 0);
+        TEST(SelectObject(hdc, hrgn2) == (PVOID)2);
+    }
+
+    /* Test colors */
+    TEST(GetDCBrushColor(hdc) == RGB(12,34,56));
+    TEST(GetDCPenColor(hdc) == RGB(23,34,45));
+
+    /* Test coordinates */
+    TEST(GetMapMode(hdc) == MM_ANISOTROPIC);
+    TEST(GetGraphicsMode(hdc) == GM_ADVANCED);
+    GetWindowOrgEx(hdc, &pt);
+    TEST(pt.x == 12);
+    TEST(pt.y == 34);
+    GetViewportOrgEx(hdc, &pt);
+    TEST(pt.x == 56);
+    TEST(pt.y == 78);
+    GetWindowExtEx(hdc, &sz);
+    TESTX(sz.cx == 123, "sz.cx == %ld\n", sz.cx);
+    TESTX(sz.cy == 456, "sz.cy == %ld\n", sz.cy);
+    GetViewportExtEx(hdc, &sz);
+    TEST(sz.cx == 234);
+    TEST(sz.cy == 567);
+
+
+}
+
+
+static
+void
+Test_SaveRestore(HDC hdc, BOOL bMemDC)
+{
+    SetSpecialDCState(hdc);
+    NtGdiSaveDC(hdc);
+    SetSpecialDCState2(hdc);
+
+    SetLastError(0);
+    TEST(NtGdiRestoreDC(hdc, 2) == 0);
+    TEST(GetLastError() == ERROR_INVALID_PARAMETER);
+
+    SetLastError(0);
+    TEST(NtGdiRestoreDC(hdc, 0) == 0);
+    TEST(GetLastError() == ERROR_INVALID_PARAMETER);
+
+    SetLastError(0);
+    TEST(NtGdiRestoreDC(hdc, -2) == 0);
+    TEST(GetLastError() == ERROR_INVALID_PARAMETER);
+
+    SetLastError(0);
+    TEST(NtGdiRestoreDC(hdc, 1) == 1);
+    TEST(GetLastError() == 0);
+
+    Test_IsSpecialState(hdc, bMemDC);
+}
+
+
+START_TEST(NtGdiRestoreDC)
+{
+    HDC hdc;
+
+    hdc = CreateCompatibleDC(0);
+    ASSERT(GdiIsHandleValid(hdc));
+
+    SetLastError(0);
+    TEST(NtGdiRestoreDC(0, -10) == 0);
+    TEST(GetLastError() == ERROR_INVALID_HANDLE);
+
+    SetLastError(0);
+    TEST(NtGdiRestoreDC(hdc, 0) == 0);
+    TEST(GetLastError() == ERROR_INVALID_PARAMETER);
+
+    SetLastError(0);
+    TEST(NtGdiRestoreDC(hdc, 1) == 0);
+    TEST(GetLastError() == ERROR_INVALID_PARAMETER);
+
+    /* Initialize objects */
+    hbrush = CreateSolidBrush(12345);
+    ASSERT(GdiIsHandleValid(hbrush));
+    hpen = CreatePen(PS_SOLID, 4, RGB(10,12,32));
+    ASSERT(GdiIsHandleValid(hpen));
+    hbitmap = CreateBitmap(10, 10, 1, 1, NULL);
+    ASSERT(GdiIsHandleValid(hbitmap));
+    hfont = CreateFont(10, 0, 0, 0, FW_NORMAL, 0, 0, 0, ANSI_CHARSET,
+                OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS, DEFAULT_QUALITY,
+                DEFAULT_PITCH, "Arial");
+    ASSERT(GdiIsHandleValid(hfont));
+    hrgn = CreateRectRgn(12, 14, 14, 17);
+    ASSERT(GdiIsHandleValid(hrgn));
+    hrgn2 = CreateRectRgn(1, 1, 2, 2);
+    ASSERT(GdiIsHandleValid(hrgn2));
+
+    /* Test mem dc */
+    Test_SaveRestore(hdc, TRUE);
+    DeleteDC(hdc);
+
+    /* Test screen DC */
+    hdc = GetDC(0);
+    ASSERT(GdiIsHandleValid(hdc));
+    Test_SaveRestore(hdc, FALSE);
+    ReleaseDC(0, hdc);
+
+    /* Test info dc */
+    hdc = CreateICW(L"DISPLAY", NULL, NULL, NULL);
+    ASSERT(GdiIsHandleValid(hdc));
+    Test_SaveRestore(hdc, FALSE);
+    DeleteDC(hdc);
+
+}
+