48 lines
978 B
Go
Executable File
48 lines
978 B
Go
Executable File
package server
|
|
|
|
import (
|
|
"local/notes-server/config"
|
|
"net/http/httptest"
|
|
"regexp"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestHead(t *testing.T) {
|
|
config.Head = "A"
|
|
w := httptest.NewRecorder()
|
|
head(w, nil)
|
|
if s := strings.TrimSpace(string(w.Body.Bytes())); s != config.Head {
|
|
t.Fatal(s)
|
|
}
|
|
}
|
|
|
|
func TestFoot(t *testing.T) {
|
|
config.Foot = "A"
|
|
w := httptest.NewRecorder()
|
|
foot(w, nil)
|
|
if s := strings.TrimSpace(string(w.Body.Bytes())); s != config.Foot {
|
|
t.Fatal(s)
|
|
}
|
|
}
|
|
|
|
func TestBlock(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
block("hi", w)
|
|
s := strings.ReplaceAll(strings.TrimSpace(string(w.Body.Bytes())), "\n", ".")
|
|
if ok, err := regexp.MatchString("<div>.*hi.*<.div>", s); err != nil {
|
|
t.Fatal(err, s)
|
|
} else if !ok {
|
|
t.Fatal(ok, s)
|
|
}
|
|
}
|
|
|
|
func TestH(t *testing.T) {
|
|
s := strings.ReplaceAll(strings.TrimSpace(h2("hi")), "\n", ".")
|
|
if ok, err := regexp.MatchString("<h2>.*hi.*<.h2>", s); err != nil {
|
|
t.Fatal(err, s)
|
|
} else if !ok {
|
|
t.Fatal(ok, s)
|
|
}
|
|
}
|