Add tests

Author: dkirov-dd

datadog_checks_base/tests/base/checks/test_agent_check.py

@@ -1316,3 +1316,23 @@ def test_env_var_logic_preset():
         AgentCheck()
         assert os.getenv('OPENSSL_CONF', None) == preset_conf
         assert os.getenv('OPENSSL_MODULES', None) == preset_modules
+
+@pytest.mark.parametrize(
+        "should_profile_value, expected_calls",
+        [
+            (True, 1),
+            (False, 0),
+        ],
+)
+def test_profile_memory(should_profile_value, expected_calls):
+    """
+    Test that profile_memory is called when should_profile_memory is True
+    """
+    check = AgentCheck('test', {}, [{}])
+    check.should_profile_memory = mock.MagicMock(return_value=should_profile_value)
+    check.profile_memory = mock.MagicMock()
+
+    check.run()
+
+    assert check.should_profile_memory.call_count == 1
+    assert check.profile_memory.call_count == expected_calls