fix unversioned test

This commit is contained in:
Daniel Smith
2015-07-17 11:58:50 -07:00
parent 1be091da07
commit 86ecc4f966
2 changed files with 6 additions and 2 deletions

View File

@@ -54,6 +54,10 @@ Documentation for other releases can be found at
<!-- END STRIP_FOR_RELEASE -->
`
func makeUnversionedWarning(fileName string) string {
return fmt.Sprintf(unversionedWarningFmt, fileName)
}
// inserts/updates a warning for unversioned docs
func updateUnversionedWarning(file string, markdown []byte) ([]byte, error) {
lines := splitLines(markdown)
@@ -64,5 +68,5 @@ func updateUnversionedWarning(file string, markdown []byte) ([]byte, error) {
if !hasMacroBlock(lines, beginUnversionedWarning, endUnversionedWarning) {
lines = append([]string{beginUnversionedWarning, endUnversionedWarning}, lines...)
}
return updateMacroBlock(lines, beginUnversionedWarning, endUnversionedWarning, fmt.Sprintf(unversionedWarningFmt, file))
return updateMacroBlock(lines, beginUnversionedWarning, endUnversionedWarning, makeUnversionedWarning(file))
}

View File

@@ -23,7 +23,7 @@ import (
)
func TestUnversionedWarning(t *testing.T) {
warningBlock := beginUnversionedWarning + "\n" + unversionedWarning + "\n" + endUnversionedWarning + "\n"
warningBlock := beginUnversionedWarning + "\n" + makeUnversionedWarning("filename.md") + "\n" + endUnversionedWarning + "\n"
var cases = []struct {
in string
out string