61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
from __future__ import annotations
|
|
import typing as t
|
|
import unittest
|
|
|
|
from tutor import hooks
|
|
|
|
|
|
class PluginFiltersTests(unittest.TestCase):
|
|
def tearDown(self) -> None:
|
|
super().tearDown()
|
|
hooks.filters.clear_all(context="tests")
|
|
|
|
def run(self, result: t.Any = None) -> t.Any:
|
|
with hooks.contexts.enter("tests"):
|
|
return super().run(result=result)
|
|
|
|
def test_add(self) -> None:
|
|
@hooks.filters.add("tests:count-sheeps")
|
|
def filter1(value: int) -> int:
|
|
return value + 1
|
|
|
|
value = hooks.filters.apply("tests:count-sheeps", 0)
|
|
self.assertEqual(1, value)
|
|
|
|
def test_add_items(self) -> None:
|
|
@hooks.filters.add("tests:add-sheeps")
|
|
def filter1(sheeps: list[int]) -> list[int]:
|
|
return sheeps + [0]
|
|
|
|
hooks.filters.add_item("tests:add-sheeps", 1)
|
|
hooks.filters.add_item("tests:add-sheeps", 2)
|
|
hooks.filters.add_items("tests:add-sheeps", [3, 4])
|
|
|
|
sheeps: list[int] = hooks.filters.apply("tests:add-sheeps", [])
|
|
self.assertEqual([0, 1, 2, 3, 4], sheeps)
|
|
|
|
def test_filter_callbacks(self) -> None:
|
|
callback = hooks.filters.FilterCallback(lambda _: 1)
|
|
self.assertTrue(callback.is_in_context(None))
|
|
self.assertFalse(callback.is_in_context("customcontext"))
|
|
self.assertEqual(1, callback.apply(0))
|
|
|
|
def test_filter_context(self) -> None:
|
|
with hooks.contexts.enter("testcontext"):
|
|
hooks.filters.add_item("test:sheeps", 1)
|
|
hooks.filters.add_item("test:sheeps", 2)
|
|
|
|
self.assertEqual([1, 2], hooks.filters.apply("test:sheeps", []))
|
|
self.assertEqual(
|
|
[1], hooks.filters.apply_from_context("testcontext", "test:sheeps", [])
|
|
)
|
|
|
|
def test_clear_context(self) -> None:
|
|
with hooks.contexts.enter("testcontext"):
|
|
hooks.filters.add_item("test:sheeps", 1)
|
|
hooks.filters.add_item("test:sheeps", 2)
|
|
|
|
self.assertEqual([1, 2], hooks.filters.apply("test:sheeps", []))
|
|
hooks.filters.clear("test:sheeps", context="testcontext")
|
|
self.assertEqual([2], hooks.filters.apply("test:sheeps", []))
|