SHGetUnreadMailCountW.cpp

Go to the documentation of this file.

/*
 * PROJECT:     ReactOS API tests
 * LICENSE:     GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
 * PURPOSE:     Test for SHGetUnreadMailCountW
 * COPYRIGHT:   Copyright 2025 Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
 */
 
#include "shelltest.h"
 
static VOID SetUnreadMailInfo(PDWORD pdwDisposition)
{
    HKEY hKey;
    LSTATUS error = RegCreateKeyExW(
        HKEY_CURRENT_USER, L"Software\\Microsoft\\Windows\\CurrentVersion\\UnreadMail\\example.com",
        0, NULL, 0, KEY_WRITE, NULL, &hKey, pdwDisposition);
    ok_long(error, ERROR_SUCCESS);
 
    DWORD dwCount = 1;
    error = SHSetValueW(hKey, NULL, L"MessageCount", REG_DWORD, &dwCount, sizeof(dwCount));
    ok_long(error, ERROR_SUCCESS);
 
    FILETIME FileTime;
    GetSystemTimeAsFileTime(&FileTime);
    error = SHSetValueW(hKey, NULL, L"TimeStamp", REG_BINARY, &FileTime, sizeof(FileTime));
    ok_long(error, ERROR_SUCCESS);
 
    LPCWSTR pszApp = L"MyMailerApp";
    DWORD cbValue = (lstrlenW(pszApp) + 1) * sizeof(WCHAR);
    error = SHSetValueW(hKey, NULL, L"Application", REG_SZ, pszApp, cbValue);
    ok_long(error, ERROR_SUCCESS);
 
    RegCloseKey(hKey);
}
 
START_TEST(SHGetUnreadMailCountW)
{
    HRESULT hr;
 
    DWORD dwDisposition;
    SetUnreadMailInfo(&dwDisposition);
 
    hr = SHGetUnreadMailCountW(NULL, L"example.com", NULL, NULL, NULL, 0);
    ok_hex(hr, S_OK);
 
    FILETIME FileTime;
    ZeroMemory(&FileTime, sizeof(FileTime));
    hr = SHGetUnreadMailCountW(HKEY_CURRENT_USER, L"example.com", NULL, &FileTime, NULL, 0);
    ok_hex(hr, S_OK);
    ok(FileTime.dwHighDateTime != 0, "FileTime.dwHighDateTime was zero\n");
 
    DWORD dwCount = 0;
    ZeroMemory(&FileTime, sizeof(FileTime));
    hr = SHGetUnreadMailCountW(NULL, NULL, &dwCount, &FileTime, NULL, 0);
    ok_hex(hr, S_OK);
    ok_long(dwCount, 1);
    ok_long(FileTime.dwHighDateTime, 0);
 
    dwCount = 0;
    hr = SHGetUnreadMailCountW(NULL, L"example.com", &dwCount, NULL, NULL, 0);
    ok_hex(hr, S_OK);
    ok_long(dwCount, 1);
 
    hr = SHGetUnreadMailCountW(NULL, NULL, &dwCount, NULL, NULL, 0);
    ok_hex(hr, S_OK);
 
    WCHAR szAppName[MAX_PATH];
    dwCount = 0;
    hr = SHGetUnreadMailCountW(NULL, NULL, &dwCount, NULL, szAppName, _countof(szAppName));
    ok_hex(hr, E_INVALIDARG);
    ok_long(dwCount, 0);
 
    hr = SHGetUnreadMailCountW(NULL, L"example.com", NULL, NULL, szAppName, _countof(szAppName));
    ok_hex(hr, S_OK);
    ok_wstr(szAppName, L"MyMailerApp");
 
    if (dwDisposition == REG_CREATED_NEW_KEY)
    {
        RegDeleteKeyW(HKEY_CURRENT_USER,
                      L"Software\\Microsoft\\Windows\\CurrentVersion\\UnreadMail\\example.com");
    }
}