added killswitch tests

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