From 3e297af79cfa2a6f5de3fa89540f2e4450eff218 Mon Sep 17 00:00:00 2001
From: A_D <aunderscored@gmail.com>
Date: Wed, 11 Aug 2021 02:17:13 +0200
Subject: [PATCH] added killswitch tests
---
tests/killswitch.py/test_killswitch.py | 49 ++++++++++++++++++++++++++
1 file changed, 49 insertions(+)
create mode 100644 tests/killswitch.py/test_killswitch.py
diff --git a/tests/killswitch.py/test_killswitch.py b/tests/killswitch.py/test_killswitch.py
new file mode 100644
index 00000000..dc53941e
--- /dev/null
+++ b/tests/killswitch.py/test_killswitch.py
@@ -0,0 +1,49 @@
+"""Tests of killswitch behaviour."""
+from typing import Optional
+
+import pytest
+import semantic_version
+
+import killswitch
+
+TEST_SET = killswitch.KillSwitchSet([
+ killswitch.KillSwitches(
+ version=semantic_version.SimpleSpec('1.0.0'),
+ kills={
+ 'no-actions': killswitch.SingleKill('no-actions', 'test'),
+ 'delete-action': killswitch.SingleKill('delete-action', 'remove stuff', delete_fields=['a', 'b.c']),
+ 'delete-action-l': killswitch.SingleKill('delete-action-l', 'remove stuff', delete_fields=['2', '0']),
+ 'set-action': killswitch.SingleKill('set-action', 'set stuff', set_fields={'a': False, 'b.c': True}),
+ 'redact-action': killswitch.SingleKill('redact-action', 'redact stuff', redact_fields=['a', 'b.c'])
+ }
+ )
+])
+
+
+@pytest.mark.parametrize(
+ ('input', 'kill', 'should_pass', 'result', 'version'),
+ [
+ ([], 'doesnt-exist', True, None, '1.0.0'),
+ # should fail, attempts to use 'a' to index a list
+ (['a', 'b', 'c'], 'delete-action', False, ['a', 'b', 'c'], '1.0.0'),
+ (['a', 'b', 'c'], 'delete-action-l', True, ['b'], '1.0.0'),
+ (set(), 'delete-action-l', False, None, '1.0.0'), # set should be thrown out because it cant be indext
+ (['a', 'b'], 'delete-action-l', True, ['b'], '1.0.0'), # has a missing value, but that's fine for delete
+ (['a', 'b'], 'delete-action-l', True, ['a', 'b'], '1.1.0'), # wrong version
+ ],
+)
+def test_killswitch(
+ input: killswitch.UPDATABLE_DATA, kill: str, should_pass: bool, result: Optional[killswitch.UPDATABLE_DATA],
+ version: str
+) -> None:
+ """Simple killswitch tests."""
+ should_return, res = TEST_SET.check_killswitch(kill, input, version=version)
+
+ assert (not should_return) == should_pass, (
+ f'expected to {"pass" if should_pass else "fail"}, but {"passed" if not should_pass else "failed"}'
+ )
+
+ if result is None:
+ return # we didn't expect any result
+
+ assert res == result