diff options
Diffstat (limited to 'tests/test_pymdvar.py')
| -rw-r--r-- | tests/test_pymdvar.py | 61 |
1 files changed, 52 insertions, 9 deletions
diff --git a/tests/test_pymdvar.py b/tests/test_pymdvar.py index 91a97d0..a4c4439 100644 --- a/tests/test_pymdvar.py +++ b/tests/test_pymdvar.py @@ -1,3 +1,4 @@ +import os import pytest from markdown import markdown from pymdvar import VariableExtension @@ -5,31 +6,73 @@ from pymdvar import VariableExtension def test_empty_input(): in_str: str = '' + exp_str: str = markdown(in_str) out_str: str = markdown(in_str, extensions=[VariableExtension()]) - assert in_str == out_str + assert out_str == exp_str @pytest.mark.parametrize('in_str, exp_str', [ ('foo bar', '<p>foo bar</p>'), ('foo *test* bar', '<p>foo <em>test</em> bar</p>'), - ('foo **test** bar', '<p>foo <strong>test</strong> bar</p>'), + ('foo *PYMDVAR_TEST_1* bar', '<p>foo <em>PYMDVAR_TEST_1</em> bar</p>'), ('foo $test bar', '<p>foo $test bar</p>'), - ('foo *${test* bar', '<p>foo <em>${test</em> bar</p>'), ('foo **$test}** bar', '<p>foo <strong>$test}</strong> bar</p>'), + ('foo **$PYMDVAR_TEST_2}** bar', '<p>foo <strong>$PYMDVAR_TEST_2}</strong> bar</p>'), + ('foo [link](${test/a.html) bar', '<p>foo <a href="${test/a.html">link</a> bar</p>'), + ('foo  bar', '<p>foo <img alt="image" src="$test}/a.jpg" /> bar</p>') +]) +def test_no_replacements_config(in_str: str, exp_str: str, single_var_dict: dict[str, str]): + out_str: str = markdown(in_str, extensions=[VariableExtension(variables=single_var_dict, enable_env=True)]) + assert out_str == exp_str + + +@pytest.mark.parametrize('in_str, exp_str', [ + ('foo *${test}* bar', '<p>foo <em></em> bar</p>'), + ('foo *${PYMDVAR_TEST_1}* bar', '<p>foo <em></em> bar</p>'), + ('foo [link](${test}/a.html) bar', '<p>foo <a href="/a.html">link</a> bar</p>'), ]) -def test_non_replacements(in_str, exp_str): +def test_replacements_no_config(in_str: str, exp_str: str): out_str: str = markdown(in_str, extensions=[VariableExtension()]) assert out_str == exp_str @pytest.mark.parametrize('in_str, exp_str', [ - ('foo ${test} bar', '<p>foo value bar</p>'), ('foo *${test}* bar', '<p>foo <em>value</em> bar</p>'), - ('foo **${test}** bar', '<p>foo <strong>value</strong> bar</p>'), + ('foo *${PYMDVAR_TEST_1}* bar', '<p>foo <em>1</em> bar</p>'), + ('foo ${value} bar', '<p>foo bar</p>'), + ('foo **${PYMDVAR_TEST_2}** bar', '<p>foo <strong></strong> bar</p>'), ('foo [link](${test}/a.html) bar', '<p>foo <a href="value/a.html">link</a> bar</p>'), - ('foo  bar', '<p>foo <img alt="image" src="value/a.jpg" /> bar</p>'), + ('foo  bar', '<p>foo <img alt="image" src="1/a.jpg" /> bar</p>'), + ('foo [link](${PYMDVAR_TEST_2}/a.html) bar', '<p>foo <a href="/a.html">link</a> bar</p>') ]) -def test_simple_replacements(in_str, exp_str): - out_str: str = markdown(in_str, extensions=[VariableExtension(variables={'test':'value'})]) +def test_simple_replacements(in_str: str, exp_str: str, single_var_dict: dict[str, str]): + out_str: str = markdown(in_str, extensions=[VariableExtension(variables=single_var_dict, enable_env=True)]) assert out_str == exp_str + +@pytest.mark.parametrize('in_str, exp_str', [ + ('foo *${test}* bar', '<p>foo <em>value</em> bar</p>'), + ('foo *${TEST}* bar', '<p>foo <em>VALUE</em> bar</p>'), + ('foo **${SOMETHING_ELSE}** bar', '<p>foo <strong>something_else</strong> bar</p>') +]) +def test_multivar_replacements(in_str: str, exp_str: str, multi_var_dict: dict[str, str]): + out_str: str = markdown(in_str, extensions=[VariableExtension(variables=multi_var_dict)]) + assert out_str == exp_str + + +@pytest.mark.parametrize('in_str, exp_str', [ + ('foo *${test}* and **${TEST}** bar', '<p>foo <em>value</em> and <strong>VALUE</strong> bar</p>'), + ('foo  bar', '<p>foo <img alt="image" src="/a.jpg" /> bar</p>') +]) +def test_text_multivar_replacements(in_str: str, exp_str: str, multi_var_dict: dict[str, str]): + out_str: str = markdown(in_str, extensions=[VariableExtension(variables=multi_var_dict)]) + assert out_str == exp_str + + +@pytest.mark.parametrize('in_str, exp_str', [ + ('foo *${test}* and **${TEST}** bar', '<p>foo <em>value</em> and <strong>VALUE</strong> bar</p>'), + ('foo  bar', '<p>foo <img alt="image" src="/a.jpg" /> bar</p>') +]) +def test_extension_text_mode(in_str: str, exp_str: str, multi_var_dict: dict[str, str]): + out_str: str = markdown(in_str, extensions=['pymdvar'], extension_configs={'pymdvar': {'variables': multi_var_dict}}) + assert out_str == exp_str |