2012-12-17 13:38:33 -05:00
package main
import (
2020-05-09 16:05:51 -04:00
"bytes"
"io/ioutil"
"log"
"os"
"path/filepath"
2012-12-17 13:38:33 -05:00
"testing"
2017-10-23 12:23:46 -04:00
"github.com/stretchr/testify/assert"
2012-12-17 13:38:33 -05:00
)
2020-05-09 16:05:51 -04:00
func TestLoadTemplates ( t * testing . T ) {
data := struct {
TestString string
} {
TestString : "Testing" ,
}
templates := loadTemplates ( "" )
assert . NotEqual ( t , templates , nil )
var defaultSignin bytes . Buffer
templates . ExecuteTemplate ( & defaultSignin , "sign_in.html" , data )
assert . Equal ( t , "\n<!DOCTYPE html>" , defaultSignin . String ( ) [ 0 : 16 ] )
var defaultError bytes . Buffer
templates . ExecuteTemplate ( & defaultError , "error.html" , data )
assert . Equal ( t , "\n<!DOCTYPE html>" , defaultError . String ( ) [ 0 : 16 ] )
dir , err := ioutil . TempDir ( "" , "templatetest" )
if err != nil {
log . Fatal ( err )
}
defer os . RemoveAll ( dir )
templateHTML := ` {{ .TestString }} {{ .TestString | ToLower }} {{ .TestString | ToUpper }} `
signInFile := filepath . Join ( dir , "sign_in.html" )
if err := ioutil . WriteFile ( signInFile , [ ] byte ( templateHTML ) , 0666 ) ; err != nil {
log . Fatal ( err )
}
errorFile := filepath . Join ( dir , "error.html" )
if err := ioutil . WriteFile ( errorFile , [ ] byte ( templateHTML ) , 0666 ) ; err != nil {
log . Fatal ( err )
}
templates = loadTemplates ( dir )
assert . NotEqual ( t , templates , nil )
var sitpl bytes . Buffer
templates . ExecuteTemplate ( & sitpl , "sign_in.html" , data )
assert . Equal ( t , "Testing testing TESTING" , sitpl . String ( ) )
var errtpl bytes . Buffer
templates . ExecuteTemplate ( & errtpl , "error.html" , data )
assert . Equal ( t , "Testing testing TESTING" , errtpl . String ( ) )
}
2012-12-17 13:38:33 -05:00
func TestTemplatesCompile ( t * testing . T ) {
templates := getTemplates ( )
assert . NotEqual ( t , templates , nil )
}
