Add unit tests for create_secret_data

Author: cderici

tests/unit/test_secrets.py

@@ -0,0 +1,59 @@
+# Copyright 2023 Canonical Ltd.
+# Licensed under the Apache V2, see LICENCE file for details.
+
+from pathlib import Path
+import tempfile
+import unittest
+import pytest
+
+from juju.secrets import create_secret_data
+
+
+class TestCreateSecretData(unittest.TestCase):
+    def test_bad_key(self):
+        with pytest.raises(ValueError):
+            create_secret_data(["foo"])
+        with pytest.raises(ValueError):
+            create_secret_data(["=bar"])
+
+    def test_good_key_values(self):
+        self.assertDictEqual(create_secret_data(["foo=bar", "hello=world", "goodbye#base64=world"]),
+                             {
+                                 "foo": "YmFy",
+                                 "hello": "d29ybGQ=",
+                                 "goodbye": "world"})
+
+    def test_key_content_too_large(self):
+        with pytest.raises(ValueError):
+            create_secret_data(["foo=" + ('a' * 8 * 1024)])
+
+    def test_total_content_too_large(self):
+        args = []
+        content = 'a' * 4 * 1024
+        for i in range(20):
+            args.append(f'key{i}={content}')
+        with pytest.raises(ValueError):
+            create_secret_data(args)
+
+    def test_secret_key_from_file(self):
+        content = """
+          -----BEGIN CERTIFICATE-----
+          MIIFYjCCA0qgAwIBAgIQKaPND9YggIG6+jOcgmpk3DANBgkqhkiG9w0BAQsFADAz
+          MRwwGgYDVQQKExNsaW51eGNvbnRhaW5lcnMub3JnMRMwEQYDVQQDDAp0aW1AZWx3
+          -----END CERTIFICATE-----"""[1:]
+
+        with tempfile.TemporaryDirectory() as temp_dir:
+            dir_path = Path(temp_dir)
+            file_path = dir_path / "secret-data.bin"
+
+            with open(file_path, "w") as file:
+                file.write(content)
+
+            data = create_secret_data(["key1=value1", f"key2#file={file_path}"])
+
+            self.assertDictEqual(data, {
+                "key1": "dmFsdWUx",
+                "key2": (
+                    'ICAgICAgICAgIC0tLS0tQkVHSU4gQ0VSVElGSUNBVEUtLS0tLQogICAgICAgICAgTUlJRllqQ0NBMHFnQXdJQkFnSVFLYVBORDlZZ2dJRzYrak9jZ21wazNEQU5CZ2txaGtpRzl3MEJBUXNGQURBegogICAgICAgICAgTVJ3d0dnWURWUVFLRXhOc2FXNTFlR052Ym5SaGFXNWxjbk11YjNKbk1STXdFUVlEVlFRRERBcDBhVzFBWld4MwogICAgICAgICAgLS0tLS1FTkQgQ0VSVElGSUNBVEUtLS0tLQ=='
+                ),
+            })