From 4914b0c74ed669d6631897950ff1660ce91ef918 Mon Sep 17 00:00:00 2001
From: Simeon Simeonov
Date: Thu, 23 Dec 2021 14:41:14 +0100
Subject: Initial commit and release for Christmas with love

---
 tests/test_envtoolkit_instance.py | 64 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 tests/test_envtoolkit_instance.py

(limited to 'tests/test_envtoolkit_instance.py')

diff --git a/tests/test_envtoolkit_instance.py b/tests/test_envtoolkit_instance.py
new file mode 100644
index 0000000..2ffacd2
--- /dev/null
+++ b/tests/test_envtoolkit_instance.py
@@ -0,0 +1,64 @@
+# etoolkit
+# Copyright (C) 2021 Simeon Simeonov
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+"""Tests for etoolkit.EtoolkitInstance"""
+import pytest
+
+import etoolkit
+
+
+def test_instantiation(config_data, password_hash):
+    """Tests for object instatiation"""
+    with pytest.raises(etoolkit.EtoolkitInstanceError) as exc_info:
+        instance = etoolkit.EtoolkitInstance('devv', config_data)
+    assert exc_info.type is etoolkit.EtoolkitInstanceError
+    assert exc_info.value.args[0] == 'Unknown instance "devv"'
+
+    instance = etoolkit.EtoolkitInstance('secret', config_data)
+    assert 'ETOOLKIT_PARENT' not in instance.raw_env_variables
+    assert 'ETOOLKIT_SENSITIVE' not in instance.raw_env_variables
+    assert 'DB_CONNECTION' not in instance.sensitive_env_variables
+    assert 'PASSWORD' in instance.sensitive_env_variables
+    assert instance.name == 'secret'
+    assert instance.master_password_hash == password_hash
+    assert instance.master_password is None
+
+
+def test_get_environ(config_data):
+    """Tests the EtoolkitInstance.get_environ method"""
+    instance = etoolkit.EtoolkitInstance('secret', config_data)
+    with pytest.raises(etoolkit.EtoolkitInstanceError) as exc_info:
+        env = instance.get_environ()
+    assert exc_info.type is etoolkit.EtoolkitInstanceError
+    assert exc_info.value.args[0] == 'Neither password or prompt function set'
+
+    instance.master_password = 'The very secret passwd'  # wrong passwd
+    with pytest.raises(etoolkit.EtoolkitInstanceError) as exc_info:
+        env = instance.get_environ()
+    assert exc_info.type is etoolkit.EtoolkitInstanceError
+    assert exc_info.value.args[0].startswith(
+        'Invalid tag when decrypting: enc-val$'
+    )
+
+    instance.master_password = 'the very secret passwd'  # correct passwd
+    env = instance.get_environ()
+    assert isinstance(env, dict)
+    assert env['PASSWORD'] == 'secret2'
+
+
+def test_get_full_name(config_data):
+    """Tests the EtoolkitInstance.get_full_name method"""
+    instance = etoolkit.EtoolkitInstance('secret', config_data)
+    assert instance.get_full_name('->') == 'default->secret'
-- 
cgit v1.3