Make all doc fragments, module utils, and plugin utils private (#1144)

* Make all doc fragments, module utils, and plugin utils private. * Remove some unused and no longer needed imports. This hopefully also fixes the CI issues, which do not happen locally for me... * Fix formatting. * Try to make CI happy, again. * Fix imports. * Lint.
2026-07-29 11:55:04 +00:00 · 2025-10-07 07:32:33 +02:00
parent bb39e67c8f
commit 117271579e
106 changed files with 480 additions and 365 deletions
@@ -0,0 +1,80 @@
+# Copyright 2022 Red Hat | Ansible
+# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from __future__ import absolute_import, division, print_function
+
+
+__metaclass__ = type
+
+import pytest
+from ansible_collections.community.docker.plugins.module_utils._copy import (
+    _stream_generator_to_fileobj,
+)
+
+
+def _simple_generator(sequence):
+    yield from sequence
+
+
+@pytest.mark.parametrize(
+    "chunks, read_sizes",
+    [
+        (
+            [
+                (1, b"1"),
+                (1, b"2"),
+                (1, b"3"),
+                (1, b"4"),
+            ],
+            [
+                1,
+                2,
+                3,
+            ],
+        ),
+        (
+            [
+                (1, b"123"),
+                (1, b"456"),
+                (1, b"789"),
+            ],
+            [
+                1,
+                4,
+                2,
+                2,
+                2,
+            ],
+        ),
+        (
+            [
+                (10 * 1024 * 1024, b"0"),
+                (10 * 1024 * 1024, b"1"),
+            ],
+            [
+                1024 * 1024 - 5,
+                5 * 1024 * 1024 - 3,
+                10 * 1024 * 1024 - 2,
+                2 * 1024 * 1024 - 1,
+                2 * 1024 * 1024 + 5 + 3 + 2 + 1,
+            ],
+        ),
+    ],
+)
+def test__stream_generator_to_fileobj(chunks, read_sizes):
+    chunks = [count * data for count, data in chunks]
+    stream = _simple_generator(chunks)
+    expected = b"".join(chunks)
+
+    buffer = b""
+    totally_read = 0
+    f = _stream_generator_to_fileobj(stream)
+    for read_size in read_sizes:
+        chunk = f.read(read_size)
+        assert len(chunk) == min(read_size, len(expected) - len(buffer))
+        buffer += chunk
+        totally_read += read_size
+
+    assert buffer == expected[: len(buffer)]
+    assert min(totally_read, len(expected)) == len(buffer)