22.2-64-g1fcd55d6-0ubuntu1~22.10.1 (patches unapplied)

Imported using git-ubuntu import.

1 files changed, 46 insertions, 0 deletions

diff --git a/tests/unittests/sources/test_bigstep.py b/tests/unittests/sources/test_bigstep.py
new file mode 100644
index 00000000..148cfa0b
--- /dev/null
+++ b/tests/unittests/sources/test_bigstep.py
@@ -0,0 +1,46 @@
+import json
+import os
+
+import httpretty
+import pytest
+
+from cloudinit import helpers
+from cloudinit.sources import DataSourceBigstep as bigstep
+from tests.unittests.helpers import mock
+
+M_PATH = "cloudinit.sources.DataSourceBigstep."
+
+IMDS_URL = "http://bigstep.com"
+METADATA_BODY = json.dumps(
+    {
+        "metadata": "metadata",
+        "vendordata_raw": "vendordata_raw",
+        "userdata_raw": "userdata_raw",
+    }
+)
+
+
+class TestBigstep:
+    @httpretty.activate
+    @pytest.mark.parametrize("custom_paths", [False, True])
+    @mock.patch(M_PATH + "util.load_file", return_value=IMDS_URL)
+    def test_get_data_honor_cloud_dir(self, m_load_file, custom_paths, tmpdir):
+        httpretty.register_uri(httpretty.GET, IMDS_URL, body=METADATA_BODY)
+
+        paths = {}
+        url_file = "/var/lib/cloud/data/seed/bigstep/url"
+        if custom_paths:
+            paths = {
+                "cloud_dir": tmpdir.join("cloud"),
+                "run_dir": tmpdir,
+                "templates_dir": tmpdir,
+            }
+            url_file = os.path.join(
+                paths["cloud_dir"], "data", "seed", "bigstep", "url"
+            )
+
+        ds = bigstep.DataSourceBigstep(
+            sys_cfg={}, distro=mock.Mock(), paths=helpers.Paths(paths)
+        )
+        assert ds._get_data()
+        assert [mock.call(url_file)] == m_load_file.call_args_list