add simple tests and simple working solution

6 files changed, 63 insertions, 24 deletions

diff --git a/src/pymdvar/__init__.py b/pymdvar/__init__.py
index b64a3b6..b64a3b6 100644
--- a/src/pymdvar/__init__.py
+++ b/pymdvar/__init__.py
diff --git a/pymdvar/pymdvar.py b/pymdvar/pymdvar.py
new file mode 100644
index 0000000..1173f20
--- /dev/null
+++ b/pymdvar/pymdvar.py
@@ -0,0 +1,31 @@
+from markdown.extensions import Extension
+from markdown.inlinepatterns import Pattern
+
+VARIABLE_RE = r'(\$\{)(\w+)(\})'
+
+
+class VariablePattern(Pattern):
+    def __init__(self, pattern, variables, md=None):
+        print(type(variables))
+        self.variables = variables
+        super().__init__(pattern, md)
+
+    def handleMatch(self, m):
+        variable = m.group(3)
+        value = ''
+        if variable in self.variables:
+            value = self.variables[variable]
+        return value
+
+
+class VariableExtension(Extension):
+    def __init__(self, **kwargs):
+        self.config = {
+            'enable_env': [False, 'Enable Environment variables parsing.'],
+            'variables': [dict(), 'Dictionary holding variables to be used.']
+        }
+        super().__init__(**kwargs)
+
+    def extendMarkdown(self, md):
+        variable = VariablePattern(VARIABLE_RE, variables=self.getConfig('variables'))
+        md.inlinePatterns.register(variable, 'variable', 75)
\ No newline at end of file
diff --git a/setup.cfg b/setup.cfg
index 640e3e0..e44d6c2 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -24,10 +24,5 @@ packages =
 universal = 0
 
 [options]
-package_dir =
-	= src
 packages = find:
 include_package_data=True
-
-[options.packages.find]
-where = src
diff --git a/src/pymdvar/pymdvar.py b/src/pymdvar/pymdvar.py
deleted file mode 100644
index aa62cfd..0000000
--- a/src/pymdvar/pymdvar.py
+++ /dev/null
@@ -1,19 +0,0 @@
-from markdown.extensions import Extension
-from markdown.inlinepatterns import Pattern
-
-VARIABLE_RE = r'\$\{(\w+)\}'
-
-
-class VariablePattern(Pattern):
-    def handleMatch(self, m):
-        variable = m.group(2)
-        return variable
-
-
-def makeExtension(*args, **kwargs):
-    return VariableExtension(*args, **kwargs)
-
-
-class VariableExtension(Extension):
-    def extendMarkdown(self, md):
-        md.inlinePatterns.register(VariablePattern(VARIABLE_RE), 'var', 175)
\ No newline at end of file
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/__init__.py
diff --git a/tests/test_pymdvar.py b/tests/test_pymdvar.py
new file mode 100644
index 0000000..2e305ec
--- /dev/null
+++ b/tests/test_pymdvar.py
@@ -0,0 +1,32 @@
+import pytest
+from markdown import markdown
+from pymdvar import VariableExtension
+
+
+def test_empty_input():
+    in_str = ''
+    out_str = markdown(in_str, extensions=[VariableExtension()])
+    print(in_str)
+    print(out_str)
+    assert in_str == out_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>')
+])
+def test_non_replacements(in_str, exp_str):
+    out_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>')
+])
+def test_simple_replacements(in_str, exp_str):
+    out_str = markdown(in_str, extensions=[VariableExtension(variables={'test':'value'})])
+    assert out_str == exp_str
+