1 files changed, 37 insertions, 10 deletions
diff --git a/tests/integration_tests/modules/test_write_files.py b/tests/integration_tests/modules/test_write_files.py
index 15832ae3..1eb7e945 100644
--- a/tests/integration_tests/modules/test_write_files.py
+++ b/tests/integration_tests/modules/test_write_files.py
@@ -7,8 +7,8 @@ and then checks if those files were created during boot.
 ``tests/cloud_tests/testcases/modules/write_files.yaml``.)"""
 
 import base64
-import pytest
 
+import pytest
 
 ASCII_TEXT = "ASCII text"
 B64_CONTENT = base64.b64encode(ASCII_TEXT.encode("utf-8"))
@@ -21,6 +21,9 @@ B64_CONTENT = base64.b64encode(ASCII_TEXT.encode("utf-8"))
 #
 USER_DATA = """\
 #cloud-config
+users:
+-   default
+-   name: myuser
 write_files:
 -   encoding: b64
     content: {}
@@ -41,26 +44,50 @@ write_files:
         H4sIAIDb/U8C/1NW1E/KzNMvzuBKTc7IV8hIzcnJVyjPL8pJ4QIA6N+MVxsAAAA=
     path: /root/file_gzip
     permissions: '0755'
-""".format(B64_CONTENT.decode("ascii"))
+-   path: '/home/testuser/my-file'
+    content: |
+      echo 'hello world!'
+    defer: true
+    owner: 'myuser'
+    permissions: '0644'
+""".format(
+    B64_CONTENT.decode("ascii")
+)
 
 
 @pytest.mark.ci
 @pytest.mark.user_data(USER_DATA)
 class TestWriteFiles:
-
     @pytest.mark.parametrize(
-        "cmd,expected_out", (
+        "cmd,expected_out",
+        (
             ("file /root/file_b64", ASCII_TEXT),
             ("md5sum </root/file_binary", "3801184b97bb8c6e63fa0e1eae2920d7"),
-            ("sha256sum </root/file_binary", (
+            (
+                "sha256sum </root/file_binary",
                 "2c791c4037ea5bd7e928d6a87380f8ba"
-                "7a803cd83d5e4f269e28f5090f0f2c9a"
-            )),
-            ("file /root/file_gzip",
-             "POSIX shell script, ASCII text executable"),
+                "7a803cd83d5e4f269e28f5090f0f2c9a",
+            ),
+            (
+                "file /root/file_gzip",
+                "POSIX shell script, ASCII text executable",
+            ),
             ("file /root/file_text", ASCII_TEXT),
-        )
+        ),
     )
     def test_write_files(self, cmd, expected_out, class_client):
         out = class_client.execute(cmd)
         assert expected_out in out
+
+    def test_write_files_deferred(self, class_client):
+        """Test that write files deferred works as expected.
+
+        Users get created after write_files module runs, so ensure that
+        with `defer: true`, the file gets written with correct ownership.
+        """
+        out = class_client.read_from_file("/home/testuser/my-file")
+        assert "echo 'hello world!'" == out
+        assert (
+            class_client.execute('stat -c "%U %a" /home/testuser/my-file')
+            == "myuser 644"
+        )