Commit 94f1f518 authored by Bryan Boreham's avatar Bryan Boreham
Browse files

Re-use existing code for loading report from file

parent e6eaf11f
Showing with 4 additions and 14 deletions
+4 -14
......@@ -3,11 +3,8 @@ package render_test
import (
"context"
"flag"
"io/ioutil"
"testing"
"github.com/ugorji/go/codec"
"github.com/weaveworks/scope/render"
"github.com/weaveworks/scope/report"
"github.com/weaveworks/scope/test/fixture"
......@@ -52,23 +49,16 @@ func benchmarkRender(b *testing.B, r render.Renderer) {
b.StopTimer()
render.ResetCache()
b.StartTimer()
benchmarkRenderResult = r.Render(context.Background(), report)
benchmarkRenderResult = r.Render(context.Background(), *report)
if len(benchmarkRenderResult.Nodes) == 0 {
b.Errorf("Rendered topology contained no nodes")
}
}
}
func loadReport() (report.Report, error) {
func loadReport() (*report.Report, error) {
if *benchReportFile == "" {
return fixture.Report, nil
}
b, err := ioutil.ReadFile(*benchReportFile)
if err != nil {
return rpt, err
return &fixture.Report, nil
}
rpt := report.MakeReport()
err = codec.NewDecoderBytes(b, &codec.JsonHandle{}).Decode(&rpt)
return rpt, err
return report.MakeFromFile(context.Background(), *benchReportFile)
}
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment