refactor(SwitchBundle): 重构h5子游戏拆分上传工具

python/test_vault_options.py

@@ -0,0 +1,66 @@
+import unittest
+
+from vault_options import parse_vault_options, read_switch_bundle
+
+
+class _FakeVaultV2(object):
+    def __init__(self):
+        self.calls = []
+
+    def read_secret_version(self, **kwargs):
+        self.calls.append(kwargs)
+        return {'data': {'data': {'switchBundle': '1'}}}
+
+
+class _FakeClient(object):
+    def __init__(self):
+        self.v2 = _FakeVaultV2()
+        self.secrets = self
+        self.kv = self
+
+
+class ParseVaultOptionsTests(unittest.TestCase):
+    def test_uses_default_path_and_mount_point(self):
+        options = parse_vault_options(['switchBundle.py', 'project', '123!@#'])
+
+        self.assertEqual(options, ('default', 'aws'))
+
+    def test_reads_path_and_mount_point_after_existing_arguments(self):
+        options = parse_vault_options(
+            ['switchBundle.py', 'project', '123!@#', 'release', 'production']
+        )
+
+        self.assertEqual(options, ('release', 'production'))
+
+    def test_uses_default_mount_point_when_only_path_is_provided(self):
+        options = parse_vault_options(
+            ['switchBundle.py', 'project', '123!@#', 'release']
+        )
+
+        self.assertEqual(options, ('release', 'aws'))
+
+    def test_passes_path_and_mount_point_to_vault(self):
+        client = _FakeClient()
+
+        read_switch_bundle(client, 'release', 'production')
+
+        self.assertEqual(
+            client.v2.calls,
+            [{
+                'path': 'release',
+                'mount_point': 'production',
+                'raise_on_deleted_version': True,
+            }],
+        )
+
+    def test_uses_default_vault_settings_when_not_provided(self):
+        client = _FakeClient()
+
+        read_switch_bundle(client)
+
+        self.assertEqual(client.v2.calls[0]['path'], 'default')
+        self.assertEqual(client.v2.calls[0]['mount_point'], 'aws')
+
+
+if __name__ == '__main__':
+    unittest.main()