ec2_metadata_facts: unit-test coverage

Partial unit-test coverage of the `ec2_metadata_facts` module.

tests/unit/plugins/modules/test_ec2_metadata_facts.py

@@ -0,0 +1,70 @@
+# This file is part of Ansible
+# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
+
+from unittest.mock import Mock, MagicMock
+from unittest.mock import patch
+from unittest.mock import call
+
+import io
+import pytest
+
+from ansible_collections.amazon.aws.plugins.modules import ec2_metadata_facts
+
+module_name = "ansible_collections.amazon.aws.plugins.modules.ec2_metadata_facts"
+
+
+class FailJson(Exception):
+    pass
+
+
+@pytest.fixture()
+def ec2_instance():
+    module = MagicMock()
+    return ec2_metadata_facts.Ec2Metadata(module)
+
+
+@patch(module_name + ".fetch_url")
+def test__fetch_401(m_fetch_url, ec2_instance):
+    ec2_instance.module.fail_json.side_effect = FailJson()
+    m_fetch_url.return_value = (None, {"status": 401, "msg": "Oops"})
+    with pytest.raises(FailJson):
+        ec2_instance._fetch("http://169.254.169.254/latest/meta-data/")
+
+
+@patch(module_name + ".fetch_url")
+def test__fetch_200(m_fetch_url, ec2_instance):
+    m_fetch_url.return_value = (io.StringIO("my-value"), {"status": 200})
+    assert (
+        ec2_instance._fetch("http://169.254.169.254/latest/meta-data/ami-id")
+        == "my-value"
+    )
+
+    m_fetch_url.return_value = (io.StringIO("1"), {"status": 200})
+    assert ec2_instance._fetch("http://169.254.169.254/latest/meta-data/ami-id") == "1"
+
+
+@patch(module_name + ".fetch_url")
+def test_fetch(m_fetch_url, ec2_instance):
+    raw_list = "ami-id\n"
+    m_fetch_url.side_effect = [
+        (io.StringIO(raw_list), {"status": 200}),
+        (io.StringIO("my-value"), {"status": 200}),
+    ]
+    ec2_instance.fetch("http://169.254.169.254/latest/meta-data/")
+    assert ec2_instance._data == {
+        "http://169.254.169.254/latest/meta-data/ami-id": "my-value"
+    }
+
+
+@patch(module_name + ".fetch_url")
+def test_fetch_recusive(m_fetch_url, ec2_instance):
+    raw_list = "whatever/\n"
+    m_fetch_url.side_effect = [
+        (io.StringIO(raw_list), {"status": 200}),
+        (io.StringIO("my-key"), {"status": 200}),
+        (io.StringIO("my-value"), {"status": 200}),
+    ]
+    ec2_instance.fetch("http://169.254.169.254/latest/meta-data/")
+    assert ec2_instance._data == {
+        "http://169.254.169.254/latest/meta-data/whatever/my-key": "my-value"
+    }