1 files changed, 84 insertions, 0 deletions
diff --git a/roles/openshift_health_checker/test/memory_availability_test.py b/roles/openshift_health_checker/test/memory_availability_test.py
new file mode 100644
index 000000000..a8ba032cc
--- /dev/null
+++ b/roles/openshift_health_checker/test/memory_availability_test.py
@@ -0,0 +1,84 @@
+import pytest
+
+from openshift_checks.memory_availability import MemoryAvailability, OpenShiftCheckException
+
+
+@pytest.mark.parametrize('group_names,is_containerized,is_active', [
+    (['masters'], False, True),
+    # ensure check is skipped on containerized installs
+    (['masters'], True, True),
+    (['nodes'], True, True),
+    (['etcd'], False, True),
+    (['masters', 'nodes'], False, True),
+    (['masters', 'etcd'], False, True),
+    ([], False, False),
+    (['lb'], False, False),
+    (['nfs'], False, False),
+])
+def test_is_active(group_names, is_containerized, is_active):
+    task_vars = dict(
+        group_names=group_names,
+        openshift=dict(common=dict(is_containerized=is_containerized)),
+    )
+    assert MemoryAvailability.is_active(task_vars=task_vars) == is_active
+
+
+@pytest.mark.parametrize("group_name,size_available", [
+    (
+        "masters",
+        17200,
+    ),
+    (
+        "nodes",
+        8200,
+    ),
+    (
+        "etcd",
+        12200,
+    ),
+])
+def test_mem_check_with_recommended_memtotal(group_name, size_available):
+    result = MemoryAvailability(execute_module=NotImplementedError).run(tmp=None, task_vars=dict(
+        group_names=[group_name],
+        ansible_memtotal_mb=size_available,
+    ))
+
+    assert not result.get('failed', False)
+
+
+@pytest.mark.parametrize("group_name,size_available", [
+    (
+        "masters",
+        1,
+    ),
+    (
+        "nodes",
+        2,
+    ),
+    (
+        "etcd",
+        3,
+    ),
+])
+def test_mem_check_with_insufficient_memtotal(group_name, size_available):
+    result = MemoryAvailability(execute_module=NotImplementedError).run(tmp=None, task_vars=dict(
+        group_names=[group_name],
+        ansible_memtotal_mb=size_available,
+    ))
+
+    assert result['failed']
+    assert "below recommended storage" in result['msg']
+
+
+def test_mem_check_with_invalid_groupname():
+    with pytest.raises(OpenShiftCheckException) as excinfo:
+        result = MemoryAvailability(execute_module=NotImplementedError).run(tmp=None, task_vars=dict(
+            openshift=dict(common=dict(
+                service_type='origin',
+                is_containerized=False,
+            )),
+            group_names=["invalid"],
+            ansible_memtotal_mb=1234567,
+        ))
+
+    assert "'invalid'" in str(excinfo.value)