summaryrefslogtreecommitdiffstats
path: root/transform/livereloadinject_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'transform/livereloadinject_test.go')
-rw-r--r--transform/livereloadinject_test.go20
1 files changed, 20 insertions, 0 deletions
diff --git a/transform/livereloadinject_test.go b/transform/livereloadinject_test.go
new file mode 100644
index 000000000..605f65954
--- /dev/null
+++ b/transform/livereloadinject_test.go
@@ -0,0 +1,20 @@
+package transform
+
+import (
+ "bytes"
+ "github.com/spf13/hugo/helpers"
+ "testing"
+)
+
+func TestLiveReloadInject(t *testing.T) {
+ out := new(bytes.Buffer)
+ in := helpers.StringToReader("</body>")
+
+ tr := NewChain(LiveReloadInject)
+ tr.Apply(out, in, []byte("path"))
+
+ expected := `<script data-no-instant>document.write('<script src="/livereload.js?mindelay=10"></' + 'script>')</script></body>`
+ if string(out.Bytes()) != expected {
+ t.Errorf("Expected %s got %s", expected, string(out.Bytes()))
+ }
+}