Question

我有以下代码会生成一些字符串输出：

package formatter

import (
    "bytes"
    "log"
    "text/template"

    "github.com/foo/bar/internal/mapper"
)

// map of template functions that enable us to identify the final item within a
// collection being iterated over.
var fns = template.FuncMap{
    "plus1": func(x int) int {
        return x + 1
    },
}

// Dot renders our results in dot format for use with graphviz
func Dot(results []mapper.Page) string {
    dotTmpl := `digraph sitemap { {{range .}}
  "{{.URL}}"
    -> { {{$n := len .Anchors}}{{range  $i, $v := .Anchors}}
      "{{.}}"{{if eq (plus1 $i) $n}}{{else}},{{end}}{{end}}
    } {{end}}
}`

    tmpl, err := template.New("digraph").Funcs(fns).Parse(dotTmpl)
    if err != nil {
        log.Fatal(err)
    }

    var output bytes.Buffer
    if err := tmpl.Execute(&output, results); err != nil {
        log.Fatal(err)
    }

    return output.String()
}

它生成如下输出：

digraph sitemap {
  "http://www.example.com/"
    -> {
      "http://www.example.com/foo",
      "http://www.example.com/bar",
      "http://www.example.com/baz"
    }
}

以下是对此功能的测试...

package formatter

import (
    "testing"

    "github.com/foo/bar/internal/mapper"
)

func TestDot(t *testing.T) {
    input := []mapper.Page{
        mapper.Page{
            URL: "http://www.example.com/",
            Anchors: []string{
                "http://www.example.com/foo",
                "http://www.example.com/bar",
                "http://www.example.com/baz",
            },
            Links: []string{
                "http://www.example.com/foo.css",
                "http://www.example.com/bar.css",
                "http://www.example.com/baz.css",
            },
            Scripts: []string{
                "http://www.example.com/foo.js",
                "http://www.example.com/bar.js",
                "http://www.example.com/baz.js",
            },
        },
    }

    output := `digraph sitemap {
      "http://www.example.com/"
          -> {
              "http://www.example.com/foo",
              "http://www.example.com/bar",
              "http://www.example.com/baz"
      }
    }`

    actual := Dot(input)

    if actual != output {
        t.Errorf("expected: %s\ngot: %s", output, actual)
    }
}

哪个失败并出现以下错误（与输出的格式间隔有关）...

--- FAIL: TestDot (0.00s)
    format_test.go:43: expected: digraph sitemap {
                  "http://www.example.com/"
                          -> {
                                  "http://www.example.com/foo",
                                  "http://www.example.com/bar",
                                  "http://www.example.com/baz"
              }
                }
        got: digraph sitemap { 
          "http://www.example.com/"
            -> { 
              "http://www.example.com/foo",
              "http://www.example.com/bar",
              "http://www.example.com/baz"
            } 
        }

我尝试调整测试output变量，以使间隔与实际代码实际输出的对齐。那没用。

我还尝试在输出变量和实际输出内容上都使用strings.Replace()，但奇怪的是，我的函数的输出（即使它通过strings.Replace传递，也会仍然是多行的（因此测试将失败）？

任何人都知道如何通过代码验证使输出保持一致？

谢谢。

更新

我尝试了@icza建议的方法，尽管测试的输出看起来更像预期的那样，但它仍然未能通过测试：

=== RUN   TestDot
--- FAIL: TestDot (0.00s)
    format_test.go:65: expected: digraph sitemap {
          "http://www.example.com/"
            -> {
              "http://www.example.com/foo",
              "http://www.example.com/bar",
              "http://www.example.com/baz"
            }
        }
        got: digraph sitemap { 
          "http://www.example.com/"
            -> { 
              "http://www.example.com/foo",
              "http://www.example.com/bar",
              "http://www.example.com/baz"
            }
        }

Answer 1

最简单的解决方案是在指定预期输出时在测试中使用相同的缩进（与模板中使用的相同）。

您有：

output := `digraph sitemap {
  "http://www.example.com/"
      -> {
          "http://www.example.com/foo",
          "http://www.example.com/bar",
          "http://www.example.com/baz"
  }
}`

将其更改为：

    output := `digraph sitemap {
  "http://www.example.com/"
    -> {
      "http://www.example.com/foo",
      "http://www.example.com/bar",
      "http://www.example.com/baz"
    }
}`

请注意，例如，最后一行没有缩进。使用原始字符串文字时，每个字符（包括缩进字符）都是原样文字的一部分。

创建正确的，不缩进的原始字符串文字的步骤

毕竟，这完全是一个非编码问题，而是编辑器自动格式化和定义原始字符串文字的问题。一种正确的解决方法是首先编写一个空的原始字符串文字，在其中添加空行，并清除编辑器插入的自动缩进：

    output := `
`

遇到这种情况时，请在结束点号之前复制并粘贴正确的输入，例如：

    output := `
digraph sitemap {
  "http://www.example.com/"
    -> {
      "http://www.example.com/foo",
      "http://www.example.com/bar",
      "http://www.example.com/baz"
    }
}`

最后，从原始字符串文字的第一行中删除换行符，您将获得正确的原始字符串文字：

    output := `digraph sitemap {
  "http://www.example.com/"
    -> {
      "http://www.example.com/foo",
      "http://www.example.com/bar",
      "http://www.example.com/baz"
    }
}`

有了这个，运行gofmt或自动格式化编辑器就不会再麻烦了。

更新：

我检查了您更新后的测试结果，得到的结果中，第一行digraph sitemap {后有一个空格，第三行-> {后有一个空格，但是您不要将这些添加到您的预期输出中。也可以将这些添加到您的预期输出中，或者从模板中删除那些空格！比较字符串时，它们是按字节比较的，每个字符（包括空格）都很重要。

要从模板中删除这些多余的空格，请执行以下操作：

    dotTmpl := `digraph sitemap { {{- range .}}
  "{{.URL}}"
    -> { {{- $n := len .Anchors}}{{range  $i, $v := .Anchors}}
      "{{.}}"{{if eq (plus1 $i) $n}}{{else}},{{end}}{{end}}
    } {{end}}
}`

请注意{{-的使用。这是对trim spaces around template actions的补充，它已添加到Go 1.6中。

Answer 2

如果要忽略格式，可以使用strings.Fields。

output := strings.Fields(`digraph sitemap {
  "http://www.example.com/"
      -> {
          "http://www.example.com/foo",
          "http://www.example.com/bar",
          "http://www.example.com/baz"
  }
}`)

actual := strings.Fields(Dot(input))

if !equal(output,actual) {
    // ...
}

其中equal是一个比较两个切片的简单函数。

Answer 3

问题是有多余的空间。在{之后的格式化文本中，这似乎是您的问题。您可以通过将格式字符串更改为此

来解决此问题

`digraph sitemap {{{range .}}
  "{{.URL}}"
    -> {{{$n := len .Anchors}}{{range  $i, $v := .Anchors}}
      "{{.}}"{{if eq (plus1 $i) $n}}{{else}},{{end}}{{end}}
    }{{end}}
}`

如何获取golang以测试多行输出匹配

更新

3 个答案:

创建正确的，不缩进的原始字符串文字的步骤