|
14 | 14 | package hugolib |
15 | 15 |
|
16 | 16 | import ( |
| 17 | + "flag" |
17 | 18 | "fmt" |
18 | 19 | "math/rand" |
19 | 20 | "path/filepath" |
@@ -64,20 +65,53 @@ func (s siteBuildingBenchmarkConfig) String() string { |
64 | 65 | } |
65 | 66 |
|
66 | 67 | func BenchmarkSiteBuilding(b *testing.B) { |
| 68 | + var ( |
| 69 | + // The below represents the full matrix of benchmarks. Big! |
| 70 | + allFrontmatters = []string{"YAML", "TOML"} |
| 71 | + allNumRootSections = []int{1, 5} |
| 72 | + allNumTags = []int{0, 1, 10, 20, 50, 100, 500, 1000, 5000} |
| 73 | + allTagsPerPage = []int{0, 1, 5, 20, 50, 80} |
| 74 | + allNumPages = []int{1, 10, 100, 500, 1000, 5000, 10000} |
| 75 | + allDoRender = []bool{false, true} |
| 76 | + allDoShortCodes = []bool{false, true} |
| 77 | + ) |
| 78 | + |
| 79 | + var runDefault bool |
| 80 | + |
| 81 | + visitor := func(a *flag.Flag) { |
| 82 | + if a.Name == "test.bench" && len(a.Value.String()) < 40 { |
| 83 | + // The full suite is too big, so fall back to some smaller default if no |
| 84 | + // restriction is set. |
| 85 | + runDefault = true |
| 86 | + } |
| 87 | + } |
| 88 | + |
| 89 | + flag.Visit(visitor) |
| 90 | + |
| 91 | + if runDefault { |
| 92 | + allFrontmatters = allFrontmatters[1:] |
| 93 | + allNumRootSections = allNumRootSections[0:2] |
| 94 | + allNumTags = allNumTags[0:2] |
| 95 | + allTagsPerPage = allTagsPerPage[2:3] |
| 96 | + allNumPages = allNumPages[2:5] |
| 97 | + allDoRender = allDoRender[1:2] |
| 98 | + allDoShortCodes = allDoShortCodes[1:2] |
| 99 | + } |
| 100 | + |
67 | 101 | var conf siteBuildingBenchmarkConfig |
68 | | - for _, frontmatter := range []string{"YAML", "TOML"} { |
| 102 | + for _, frontmatter := range allFrontmatters { |
69 | 103 | conf.Frontmatter = frontmatter |
70 | | - for _, rootSections := range []int{1, 5} { |
| 104 | + for _, rootSections := range allNumRootSections { |
71 | 105 | conf.RootSections = rootSections |
72 | | - for _, numTags := range []int{0, 1, 10, 20, 50, 100, 500, 1000, 5000} { |
| 106 | + for _, numTags := range allNumTags { |
73 | 107 | conf.NumTags = numTags |
74 | | - for _, tagsPerPage := range []int{0, 1, 5, 20, 50, 80} { |
| 108 | + for _, tagsPerPage := range allTagsPerPage { |
75 | 109 | conf.TagsPerPage = tagsPerPage |
76 | | - for _, numPages := range []int{1, 10, 100, 500, 1000, 5000, 10000} { |
| 110 | + for _, numPages := range allNumPages { |
77 | 111 | conf.NumPages = numPages |
78 | | - for _, render := range []bool{false, true} { |
| 112 | + for _, render := range allDoRender { |
79 | 113 | conf.Render = render |
80 | | - for _, shortcodes := range []bool{false, true} { |
| 114 | + for _, shortcodes := range allDoShortCodes { |
81 | 115 | conf.Shortcodes = shortcodes |
82 | 116 | doBenchMarkSiteBuilding(conf, b) |
83 | 117 | } |
@@ -207,8 +241,9 @@ category = "categories" |
207 | 241 | // Maybe consider reusing the Source fs |
208 | 242 | mf := afero.NewMemMapFs() |
209 | 243 | th, h := newTestSitesFromConfig(b, mf, siteConfig, |
210 | | - "layouts/_default/single.html", `Single HTML|{{ .Title }}|{{ .Content }}`, |
| 244 | + "layouts/_default/single.html", `Single HTML|{{ .Title }}|{{ .Content }}|{{ partial "myPartial" . }} `, |
211 | 245 | "layouts/_default/list.html", `List HTML|{{ .Title }}|{{ .Content }}`, |
| 246 | + "layouts/partials/myPartial.html", `Partial: {{ "Hello **world**!" | markdownify }}`, |
212 | 247 | "layouts/shortcodes/myShortcode.html", `<p>MyShortcode</p>`) |
213 | 248 |
|
214 | 249 | fs := th.Fs |
|
0 commit comments