fix mocks

DataDog · Dec 25, 2018 · 69a057c · 69a057c
1 parent d3c36bf
commit 69a057c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/datadog_checks_tests_helper/datadog_test_libs/win/pdh_mocks.py b/datadog_checks_tests_helper/datadog_test_libs/win/pdh_mocks.py
@@ -31,10 +31,12 @@ def pdh_mocks_fixture():
     pdhaddcounter = mock.patch('win32pdh.AddCounter', mock_AddCounter)
     pdhgetformattedcountervalue = mock.patch('win32pdh.GetFormattedCounterValue', mock_GetFormattedCounterValue)
     pdhcollectquerydata = mock.patch('win32pdh.CollectQueryData', mock_CollectQueryData)
+    pdhvalidatepath = mock.patch('win32pdh.ValidatePath', mock_ValidatePath)
 
     yield regqueryvalueex.start(), pdhlookupbyindex.start(), \
         pdhenumobjectitems.start(),  pdhmakecounterpath.start(), \
-        pdhaddcounter.start(), pdhgetformattedcountervalue.start(), pdhcollectquerydata.start()
+        pdhaddcounter.start(), pdhgetformattedcountervalue.start(), pdhcollectquerydata.start(), \
+        pdhvalidatepath.start()
 
     regqueryvalueex.stop()
     pdhlookupbyindex.stop()
@@ -43,6 +45,7 @@ def pdh_mocks_fixture():
     pdhaddcounter.stop()
     pdhgetformattedcountervalue.stop()
     pdhcollectquerydata.stop()
+    pdhvalidatepath.stop()
 
 
 @pytest.fixture
@@ -216,3 +219,7 @@ def mock_QueryValueEx(*args, **kwargs):
 
 def mock_QueryValueExWithRaise(*args, **kwargs):
     raise WindowsError
+
+
+def mock_ValidatePath(p):
+    return 0