SHGetUnreadMailCountW.cpp File Reference

#include "shelltest.h"

Include dependency graph for SHGetUnreadMailCountW.cpp:

Go to the source code of this file.

Functions
static VOID	SetUnreadMailInfo (PDWORD pdwDisposition)
	START_TEST (SHGetUnreadMailCountW)

Function Documentation

SetUnreadMailInfo()

static VOID SetUnreadMailInfo ( PDWORD  pdwDisposition )

static

Definition at line 10 of file SHGetUnreadMailCountW.cpp.

{
    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);
}

Referenced by START_TEST().

START_TEST()

START_TEST

(

SHGetUnreadMailCountW

)

Definition at line 35 of file SHGetUnreadMailCountW.cpp.

{
    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");
    }
}