Add tests for replaceShortcuts

1 files changed, 28 insertions, 0 deletions

diff --git a/lib/templating/__tests__/replaceShortcuts.js b/lib/templating/__tests__/replaceShortcuts.js
new file mode 100644
index 0000000..bd94f29
--- /dev/null
+++ b/lib/templating/__tests__/replaceShortcuts.js
@@ -0,0 +1,28 @@
+var TemplateEngine = require('../../models/templateEngine');
+var TemplateBlock = require('../../models/templateBlock');
+var replaceShortcuts = require('../replaceShortcuts');
+
+describe('replaceShortcuts', function() {
+    var engine = TemplateEngine.create({
+        blocks:[
+            TemplateBlock.create('math', {
+                shortcuts: {
+                    start: '$$',
+                    end: '$$',
+                    parsers: ['markdown']
+                }
+            })
+        ]
+    });
+
+    it('should correctly replace inline matches by block', function() {
+        var content = replaceShortcuts(engine, 'test.md', 'Hello $$a = b$$');
+        expect(content).toBe('Hello {% math %}a = b{% endmath %}');
+    });
+
+    it('should correctly replace block matches', function() {
+        var content = replaceShortcuts(engine, 'test.md', 'Hello\n$$\na = b\n$$\n');
+        expect(content).toBe('Hello\n{% math %}\na = b\n{% endmath %}\n');
+    });
+});
+