mirror of
https://github.com/caddyserver/caddy.git
synced 2024-11-26 10:13:39 +08:00
162 lines
3.5 KiB
Go
162 lines
3.5 KiB
Go
package markdown
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/mholt/caddy/middleware"
|
|
"github.com/russross/blackfriday"
|
|
)
|
|
|
|
func TestMarkdown(t *testing.T) {
|
|
templates := make(map[string]string)
|
|
templates[DefaultTemplate] = "testdata/markdown_tpl.html"
|
|
md := Markdown{
|
|
Root: "./testdata",
|
|
FileSys: http.Dir("./testdata"),
|
|
Configs: []Config{
|
|
Config{
|
|
Renderer: blackfriday.HtmlRenderer(0, "", ""),
|
|
PathScope: "/blog",
|
|
Extensions: []string{".md"},
|
|
Styles: []string{},
|
|
Scripts: []string{},
|
|
Templates: templates,
|
|
StaticDir: DefaultStaticDir,
|
|
StaticFiles: make(map[string]string),
|
|
},
|
|
Config{
|
|
Renderer: blackfriday.HtmlRenderer(0, "", ""),
|
|
PathScope: "/log",
|
|
Extensions: []string{".md"},
|
|
Styles: []string{"/resources/css/log.css", "/resources/css/default.css"},
|
|
Scripts: []string{"/resources/js/log.js", "/resources/js/default.js"},
|
|
Templates: make(map[string]string),
|
|
StaticDir: DefaultStaticDir,
|
|
StaticFiles: make(map[string]string),
|
|
},
|
|
},
|
|
IndexFiles: []string{"index.html"},
|
|
Next: middleware.HandlerFunc(func(w http.ResponseWriter, r *http.Request) (int, error) {
|
|
t.Fatalf("Next shouldn't be called")
|
|
return 0, nil
|
|
}),
|
|
}
|
|
|
|
req, err := http.NewRequest("GET", "/blog/test.md", nil)
|
|
if err != nil {
|
|
t.Fatalf("Could not create HTTP request: %v", err)
|
|
}
|
|
|
|
rec := httptest.NewRecorder()
|
|
|
|
md.ServeHTTP(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("Wrong status, expected: %d and got %d", http.StatusOK, rec.Code)
|
|
}
|
|
|
|
respBody := rec.Body.String()
|
|
expectedBody := `<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<title>Markdown test</title>
|
|
</head>
|
|
<body>
|
|
<h1>Header</h1>
|
|
|
|
Welcome to A Caddy website!
|
|
<h2>Welcome on the blog</h2>
|
|
|
|
<p>Body</p>
|
|
|
|
<p><code>go
|
|
func getTrue() bool {
|
|
return true
|
|
}
|
|
</code></p>
|
|
|
|
</body>
|
|
</html>
|
|
`
|
|
if respBody != expectedBody {
|
|
t.Fatalf("Expected body: %v got: %v", expectedBody, respBody)
|
|
}
|
|
|
|
req, err = http.NewRequest("GET", "/log/test.md", nil)
|
|
if err != nil {
|
|
t.Fatalf("Could not create HTTP request: %v", err)
|
|
}
|
|
rec = httptest.NewRecorder()
|
|
|
|
md.ServeHTTP(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("Wrong status, expected: %d and got %d", http.StatusOK, rec.Code)
|
|
}
|
|
respBody = rec.Body.String()
|
|
expectedBody = `<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<title>Markdown test</title>
|
|
<meta charset="utf-8">
|
|
<link rel="stylesheet" href="/resources/css/log.css">
|
|
<link rel="stylesheet" href="/resources/css/default.css">
|
|
|
|
<script src="/resources/js/log.js"></script>
|
|
<script src="/resources/js/default.js"></script>
|
|
|
|
</head>
|
|
<body>
|
|
<h2>Welcome on the blog</h2>
|
|
|
|
<p>Body</p>
|
|
|
|
<p><code>go
|
|
func getTrue() bool {
|
|
return true
|
|
}
|
|
</code></p>
|
|
|
|
</body>
|
|
</html>`
|
|
|
|
replacer := strings.NewReplacer("\r", "", "\n", "")
|
|
respBody = replacer.Replace(respBody)
|
|
expectedBody = replacer.Replace(expectedBody)
|
|
if respBody != expectedBody {
|
|
t.Fatalf("Expected body: %v got: %v", expectedBody, respBody)
|
|
}
|
|
|
|
expectedLinks := []string{
|
|
"/blog/test.md",
|
|
"/log/test.md",
|
|
}
|
|
|
|
for i, c := range md.Configs {
|
|
if c.Links[0].Url != expectedLinks[i] {
|
|
t.Fatalf("Expected %v got %v", expectedLinks[i], c.Links[0].Url)
|
|
}
|
|
}
|
|
|
|
// attempt to trigger race condition
|
|
var w sync.WaitGroup
|
|
f := func() {
|
|
req, err := http.NewRequest("GET", "/log/test.md", nil)
|
|
if err != nil {
|
|
t.Fatalf("Could not create HTTP request: %v", err)
|
|
}
|
|
rec := httptest.NewRecorder()
|
|
|
|
md.ServeHTTP(rec, req)
|
|
w.Done()
|
|
}
|
|
for i := 0; i < 5; i++ {
|
|
w.Add(1)
|
|
go f()
|
|
}
|
|
w.Wait()
|
|
|
|
}
|