transform: Add missing test case in livereloadinject

* Test for both </body> and </BODY>
* This also cosmetically changes the behaviour, as the case of the end body tag is kept.

diff --git a/transform/livereloadinject.go b/transform/livereloadinject.go
index 789ff9d2248073b519732f7dac521cd18dfa50db..9eab45f78d31e58d4d9fcde103433a7c2bd8ade6 100644 (file)
--- a/transform/livereloadinject.go
+++ b/transform/livereloadinject.go
@@ -15,15 +15,20 @@ package transform
 
 import (
        "bytes"
+       "fmt"
 )
 
 func LiveReloadInject(ct contentTransformer) {
-       match := []byte("</body>")
-       replace := []byte(`<script data-no-instant>document.write('<script src="/livereload.js?mindelay=10"></' + 'script>')</script></body>`)
+       endBodyTag := "</body>"
+       match := []byte(endBodyTag)
+       replaceTemplate := `<script data-no-instant>document.write('<script src="/livereload.js?mindelay=10"></' + 'script>')</script>%s`
+       replace := []byte(fmt.Sprintf(replaceTemplate, endBodyTag))
 
        newcontent := bytes.Replace(ct.Content(), match, replace, -1)
        if len(newcontent) == len(ct.Content()) {
-               match := []byte("</BODY>")
+               endBodyTag = "</BODY>"
+               replace := []byte(fmt.Sprintf(replaceTemplate, endBodyTag))
+               match := []byte(endBodyTag)
                newcontent = bytes.Replace(ct.Content(), match, replace, -1)
        }
 

diff --git a/transform/livereloadinject_test.go b/transform/livereloadinject_test.go
index 0c47cb1093c0a0084ec56b6f301d85c85159b6cf..5aea8689cce70b810a02b7a2893e786e89c2e6c6 100644 (file)
--- a/transform/livereloadinject_test.go
+++ b/transform/livereloadinject_test.go
@@ -15,18 +15,24 @@ package transform
 
 import (
        "bytes"
+       "fmt"
        "github.com/spf13/hugo/helpers"
        "testing"
 )
 
 func TestLiveReloadInject(t *testing.T) {
+       doTestLiveReloadInject(t, "</body>")
+       doTestLiveReloadInject(t, "</BODY>")
+}
+
+func doTestLiveReloadInject(t *testing.T, bodyEndTag string) {
        out := new(bytes.Buffer)
-       in := helpers.StringToReader("</body>")
+       in := helpers.StringToReader(bodyEndTag)
 
        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>`
+       expected := fmt.Sprintf(`<script data-no-instant>document.write('<script src="/livereload.js?mindelay=10"></' + 'script>')</script>%s`, bodyEndTag)
        if string(out.Bytes()) != expected {
                t.Errorf("Expected %s got %s", expected, string(out.Bytes()))
        }