Update Dutil_TraceErrorSource to filter based on the report level.

12 files changed, 125 insertions, 12 deletions

diff --git a/src/test/DUtilUnitTest/DUtilTests.cpp b/src/test/DUtilUnitTest/DUtilTests.cpp
new file mode 100644
index 00000000..55e81d46
--- /dev/null
+++ b/src/test/DUtilUnitTest/DUtilTests.cpp
@@ -0,0 +1,35 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+#include "precomp.h"
+
+using namespace System;
+using namespace Xunit;
+using namespace WixBuildTools::TestSupport;
+
+namespace DutilTests
+{
+    public ref class DUtil
+    {
+    public:
+        [Fact]
+        void DUtilTraceErrorSourceFiltersOnTraceLevel()
+        {
+            DutilInitialize(&DutilTestTraceError);
+
+            CallDutilTraceErrorSource();
+
+            Dutil_TraceSetLevel(REPORT_DEBUG, FALSE);
+
+            Action^ action = gcnew Action(this, &DUtil::CallDutilTraceErrorSource);
+            Assert::Throws<Exception^>(action);
+
+            DutilUninitialize();
+        }
+
+    private:
+        void CallDutilTraceErrorSource()
+        {
+            Dutil_TraceErrorSource(__FILE__, __LINE__, REPORT_DEBUG, DUTIL_SOURCE_EXTERNAL, E_FAIL, "Error message");
+        }
+    };
+}
diff --git a/src/test/DUtilUnitTest/DUtilUnitTest.vcxproj b/src/test/DUtilUnitTest/DUtilUnitTest.vcxproj
index b023d74f..c0974780 100644
--- a/src/test/DUtilUnitTest/DUtilUnitTest.vcxproj
+++ b/src/test/DUtilUnitTest/DUtilUnitTest.vcxproj
@@ -36,6 +36,8 @@
     <ClCompile Include="AssemblyInfo.cpp" />
     <ClCompile Include="DictUtilTest.cpp" />
     <ClCompile Include="DirUtilTests.cpp" />
+    <ClCompile Include="DUtilTests.cpp" />
+    <ClCompile Include="error.cpp" />
     <ClCompile Include="FileUtilTest.cpp" />
     <ClCompile Include="GuidUtilTest.cpp" />
     <ClCompile Include="IniUtilTest.cpp" />
@@ -92,4 +94,4 @@
     <Error Condition="!Exists('..\..\..\packages\xunit.core.2.4.1\build\xunit.core.targets')" Text="$([System.String]::Format('$(ErrorText)', '..\..\..\packages\xunit.core.2.4.1\build\xunit.core.targets'))" />
     <Error Condition="!Exists('..\..\..\packages\xunit.runner.visualstudio.2.4.1\build\net20\xunit.runner.visualstudio.props')" Text="$([System.String]::Format('$(ErrorText)', '..\..\..\packages\xunit.runner.visualstudio.2.4.1\build\net20\xunit.runner.visualstudio.props'))" />
   </Target>
-</Project>
+</Project>
\ No newline at end of file
diff --git a/src/test/DUtilUnitTest/DUtilUnitTest.vcxproj.filters b/src/test/DUtilUnitTest/DUtilUnitTest.vcxproj.filters
index 783e18c7..2456558f 100644
--- a/src/test/DUtilUnitTest/DUtilUnitTest.vcxproj.filters
+++ b/src/test/DUtilUnitTest/DUtilUnitTest.vcxproj.filters
@@ -24,6 +24,12 @@
     <ClCompile Include="DirUtilTests.cpp">
       <Filter>Source Files</Filter>
     </ClCompile>
+    <ClCompile Include="DUtilTests.cpp">
+      <Filter>Source Files</Filter>
+    </ClCompile>
+    <ClCompile Include="error.cpp">
+      <Filter>Source Files</Filter>
+    </ClCompile>
     <ClCompile Include="FileUtilTest.cpp">
       <Filter>Source Files</Filter>
     </ClCompile>
@@ -42,6 +48,9 @@
     <ClCompile Include="PathUtilTest.cpp">
       <Filter>Source Files</Filter>
     </ClCompile>
+    <ClCompile Include="precomp.cpp">
+      <Filter>Source Files</Filter>
+    </ClCompile>
     <ClCompile Include="StrUtilTest.cpp">
       <Filter>Source Files</Filter>
     </ClCompile>
diff --git a/src/test/DUtilUnitTest/DictUtilTest.cpp b/src/test/DUtilUnitTest/DictUtilTest.cpp
index 4e9d3907..4b4777d7 100644
--- a/src/test/DUtilUnitTest/DictUtilTest.cpp
+++ b/src/test/DUtilUnitTest/DictUtilTest.cpp
@@ -22,6 +22,8 @@ namespace DutilTests
         [Fact]
         void DictUtilTest()
         {
+            DutilInitialize(&DutilTestTraceError);
+
             EmbeddedKeyTestHelper(DICT_FLAG_NONE, numIterations);
 
             EmbeddedKeyTestHelper(DICT_FLAG_CASEINSENSITIVE, numIterations);
@@ -29,6 +31,8 @@ namespace DutilTests
             StringListTestHelper(DICT_FLAG_NONE, numIterations);
 
             StringListTestHelper(DICT_FLAG_CASEINSENSITIVE, numIterations);
+
+            DutilUninitialize();
         }
 
     private:
diff --git a/src/test/DUtilUnitTest/FileUtilTest.cpp b/src/test/DUtilUnitTest/FileUtilTest.cpp
index 9bd1d0c0..0087a1d5 100644
--- a/src/test/DUtilUnitTest/FileUtilTest.cpp
+++ b/src/test/DUtilUnitTest/FileUtilTest.cpp
@@ -18,6 +18,8 @@ namespace DutilTests
             LPWSTR sczTempDir = NULL;
             LPWSTR sczFileDir = NULL;
 
+            DutilInitialize(&DutilTestTraceError);
+
             try
             {
                 hr = PathExpand(&sczTempDir, L"%TEMP%\\FileUtilTest\\", PATH_EXPAND_ENVIRONMENT);
@@ -43,6 +45,7 @@ namespace DutilTests
             {
                 ReleaseStr(sczTempDir);
                 ReleaseStr(sczFileDir);
+                DutilUninitialize();
             }
         }
 
diff --git a/src/test/DUtilUnitTest/IniUtilTest.cpp b/src/test/DUtilUnitTest/IniUtilTest.cpp
index 2edd56ab..946f19c5 100644
--- a/src/test/DUtilUnitTest/IniUtilTest.cpp
+++ b/src/test/DUtilUnitTest/IniUtilTest.cpp
@@ -24,6 +24,8 @@ namespace DutilTests
             LPWSTR wzIniContents = L"           PlainValue             =       \t      Blah               \r\n;CommentHere\r\n[Section1]\r\n     ;Another Comment With = Equal Sign\r\nSection1ValueA=Foo\r\n\r\nSection1ValueB=Bar\r\n[Section2]\r\nSection2ValueA=Cha\r\nArray[0]=Arr\r\n";
             LPWSTR wzScriptContents = L"setf ~PlainValue Blah\r\n;CommentHere\r\n\r\nsetf ~Section1\\Section1ValueA Foo\r\n\r\nsetf ~Section1\\Section1ValueB Bar\r\nsetf ~Section2\\Section2ValueA Cha\r\nsetf ~Section2\\Array[0] Arr\r\n";
 
+            DutilInitialize(&DutilTestTraceError);
+
             try
             {
                 hr = PathExpand(&sczTempIniFilePath, L"%TEMP%\\IniUtilTest\\Test.ini", PATH_EXPAND_ENVIRONMENT);
@@ -58,6 +60,7 @@ namespace DutilTests
             {
                 ReleaseStr(sczTempIniFilePath);
                 ReleaseStr(sczTempIniFileDir);
+                DutilUninitialize();
             }
         }
 
diff --git a/src/test/DUtilUnitTest/MemUtilTest.cpp b/src/test/DUtilUnitTest/MemUtilTest.cpp
index 2621da6d..09692bfb 100644
--- a/src/test/DUtilUnitTest/MemUtilTest.cpp
+++ b/src/test/DUtilUnitTest/MemUtilTest.cpp
@@ -27,6 +27,8 @@ namespace DutilTests
             ArrayValue *rgValues = NULL;
             DWORD cValues = 0;
 
+            DutilInitialize(&DutilTestTraceError);
+
             try
             {
                 hr = MemEnsureArraySize(reinterpret_cast<LPVOID*>(&rgValues), cValues + 1, sizeof(ArrayValue), 5);
@@ -101,7 +103,7 @@ namespace DutilTests
             }
 
         LExit:
-            return;
+            DutilUninitialize();
         }
 
         [Fact]
@@ -111,6 +113,8 @@ namespace DutilTests
             ArrayValue *rgValues = NULL;
             DWORD cValues = 0;
 
+            DutilInitialize(&DutilTestTraceError);
+
             try
             {
                 hr = MemInsertIntoArray(reinterpret_cast<LPVOID*>(&rgValues), 0, 1, cValues + 1, sizeof(ArrayValue), 5);
@@ -180,6 +184,7 @@ namespace DutilTests
             finally
             {
                 ReleaseMem(rgValues);
+                DutilUninitialize();
             }
         }
 
@@ -190,6 +195,8 @@ namespace DutilTests
             ArrayValue *rgValues = NULL;
             DWORD cValues = 0;
 
+            DutilInitialize(&DutilTestTraceError);
+
             try
             {
                 hr = MemEnsureArraySize(reinterpret_cast<LPVOID*>(&rgValues), 10, sizeof(ArrayValue), 10);
@@ -252,6 +259,7 @@ namespace DutilTests
             finally
             {
                 ReleaseMem(rgValues);
+                DutilUninitialize();
             }
         }
 
@@ -262,6 +270,8 @@ namespace DutilTests
             ArrayValue *rgValues = NULL;
             DWORD cValues = 0;
 
+            DutilInitialize(&DutilTestTraceError);
+
             try
             {
                 hr = MemEnsureArraySize(reinterpret_cast<LPVOID*>(&rgValues), 10, sizeof(ArrayValue), 10);
@@ -326,6 +336,7 @@ namespace DutilTests
             finally
             {
                 ReleaseMem(rgValues);
+                DutilUninitialize();
             }
         }
 
@@ -336,6 +347,8 @@ namespace DutilTests
             ArrayValue *rgValues = NULL;
             DWORD cValues = 0;
 
+            DutilInitialize(&DutilTestTraceError);
+
             try
             {
                 hr = MemEnsureArraySize(reinterpret_cast<LPVOID*>(&rgValues), 10, sizeof(ArrayValue), 10);
@@ -424,6 +437,7 @@ namespace DutilTests
             finally
             {
                 ReleaseMem(rgValues);
+                DutilUninitialize();
             }
         }
 
diff --git a/src/test/DUtilUnitTest/StrUtilTest.cpp b/src/test/DUtilUnitTest/StrUtilTest.cpp
index 7c35b7c0..94fee280 100644
--- a/src/test/DUtilUnitTest/StrUtilTest.cpp
+++ b/src/test/DUtilUnitTest/StrUtilTest.cpp
@@ -86,6 +86,8 @@ namespace DutilTests
             HRESULT hr = S_OK;
             LPWSTR sczOutput = NULL;
 
+            DutilInitialize(&DutilTestTraceError);
+
             try
             {
                 hr = StrTrimWhitespace(&sczOutput, wzInput);
@@ -103,7 +105,7 @@ namespace DutilTests
             }
 
         LExit:
-            return;
+            DutilUninitialize();
         }
 
         void TestTrimAnsi(LPCSTR szInput, LPCSTR szExpectedResult)
@@ -111,6 +113,8 @@ namespace DutilTests
             HRESULT hr = S_OK;
             LPSTR sczOutput = NULL;
 
+            DutilInitialize(&DutilTestTraceError);
+
             try
             {
                 hr = StrAnsiTrimWhitespace(&sczOutput, szInput);
@@ -128,7 +132,7 @@ namespace DutilTests
             }
 
         LExit:
-            return;
+            DutilUninitialize();
         }
 
         void TestStrAllocStringAnsi(LPCSTR szSource, DWORD cchSource, LPCWSTR wzExpectedResult)
@@ -136,6 +140,8 @@ namespace DutilTests
             HRESULT hr = S_OK;
             LPWSTR sczOutput = NULL;
 
+            DutilInitialize(&DutilTestTraceError);
+
             try
             {
                 hr = StrAllocStringAnsi(&sczOutput, szSource, cchSource, CP_UTF8);
@@ -153,7 +159,7 @@ namespace DutilTests
             }
 
         LExit:
-            return;
+            DutilUninitialize();
         }
 
         void TestStrAnsiAllocString(LPWSTR wzSource, DWORD cchSource, LPCSTR szExpectedResult)
@@ -161,6 +167,8 @@ namespace DutilTests
             HRESULT hr = S_OK;
             LPSTR sczOutput = NULL;
 
+            DutilInitialize(&DutilTestTraceError);
+
             try
             {
                 hr = StrAnsiAllocString(&sczOutput, wzSource, cchSource, CP_UTF8);
@@ -178,7 +186,7 @@ namespace DutilTests
             }
 
         LExit:
-            return;
+            DutilUninitialize();
         }
     };
 }
diff --git a/src/test/DUtilUnitTest/UriUtilTest.cpp b/src/test/DUtilUnitTest/UriUtilTest.cpp
index 220b3ff5..b3bf87a2 100644
--- a/src/test/DUtilUnitTest/UriUtilTest.cpp
+++ b/src/test/DUtilUnitTest/UriUtilTest.cpp
@@ -17,6 +17,8 @@ namespace CfgTests
                 {
                         HRESULT hr = S_OK;
 
+                        DutilInitialize(&DutilTestTraceError);
+
                         LPCWSTR uri = L"https://localhost/";
                         URI_PROTOCOL uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
                         hr = UriProtocol(uri, &uriProtocol);
@@ -90,7 +92,7 @@ namespace CfgTests
                         Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_FTP, (int)uriProtocol);
 
                 LExit:
-                        ;
+                        DutilUninitialize();
                 }
         };
 }
diff --git a/src/test/DUtilUnitTest/error.cpp b/src/test/DUtilUnitTest/error.cpp
new file mode 100644
index 00000000..e51971c3
--- /dev/null
+++ b/src/test/DUtilUnitTest/error.cpp
@@ -0,0 +1,26 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+#include "precomp.h"
+
+const int ERROR_STRING_BUFFER = 1024;
+
+static char szMsg[ERROR_STRING_BUFFER];
+static WCHAR wzMsg[ERROR_STRING_BUFFER];
+
+void CALLBACK DutilTestTraceError(
+    __in_z LPCSTR /*szFile*/,
+    __in int /*iLine*/,
+    __in REPORT_LEVEL /*rl*/,
+    __in UINT source,
+    __in HRESULT hrError,
+    __in_z __format_string LPCSTR szFormat,
+    __in va_list args
+    )
+{
+    if (DUTIL_SOURCE_EXTERNAL == source)
+    {
+        ::StringCchPrintfA(szMsg, countof(szMsg), szFormat, args);
+        MultiByteToWideChar(CP_ACP, 0, szMsg, -1, wzMsg, countof(wzMsg));
+        throw gcnew System::Exception(System::String::Format("hr = 0x{0:X8}, message = {1}", hrError, gcnew System::String(wzMsg)));
+    }
+}
diff --git a/src/test/DUtilUnitTest/error.h b/src/test/DUtilUnitTest/error.h
index a52db56d..b973acaf 100644
--- a/src/test/DUtilUnitTest/error.h
+++ b/src/test/DUtilUnitTest/error.h
@@ -1,8 +1,14 @@
+#pragma once
 // Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
 
-const int ERROR_STRING_BUFFER = 1024;
+#define DUTIL_SOURCE_DEFAULT DUTIL_SOURCE_EXTERNAL
 
-static char szMsg[ERROR_STRING_BUFFER];
-static WCHAR wzMsg[ERROR_STRING_BUFFER];
-
-#define ExitTrace(x, f, ...) { HRESULT hrTemp = x; hr = ::StringCchPrintfA(szMsg, countof(szMsg), f, __VA_ARGS__); MultiByteToWideChar(CP_ACP, 0, szMsg, -1, wzMsg, countof(wzMsg)); throw gcnew System::Exception(System::String::Format("hr = 0x{0:X8}, message = {1}", hrTemp, gcnew System::String(wzMsg))); }
+void CALLBACK DutilTestTraceError(
+    __in_z LPCSTR szFile,
+    __in int iLine,
+    __in REPORT_LEVEL rl,
+    __in UINT source,
+    __in HRESULT hrError,
+    __in_z __format_string LPCSTR szFormat,
+    __in va_list args
+    );
diff --git a/src/test/DUtilUnitTest/precomp.h b/src/test/DUtilUnitTest/precomp.h
index 15dfcd1a..284668fe 100644
--- a/src/test/DUtilUnitTest/precomp.h
+++ b/src/test/DUtilUnitTest/precomp.h
@@ -7,6 +7,7 @@
 #include <ShlObj.h>
 
 // Include error.h before dutil.h
+#include <dutilsources.h>
 #include "error.h"
 #include <dutil.h>