| 64 | } |
| 65 | |
| 66 | func TestWriteOKOmitsStatsWhenFlagOff(t *testing.T) { |
| 67 | oldWriter, oldStats, oldSDKStats := writer, statsFlag, sdkStats |
| 68 | defer func() { writer, statsFlag, sdkStats = oldWriter, oldStats, oldSDKStats }() |
| 69 | |
| 70 | var buf bytes.Buffer |
| 71 | writer = output.New(output.Options{Format: output.FormatJSON, Stdout: &buf}) |
| 72 | sdkStats = &statsHooks{} |
| 73 | statsFlag = false |
| 74 | |
| 75 | if err := writeOK(map[string]string{"hello": "world"}); err != nil { |
| 76 | t.Fatalf("writeOK: %v", err) |
| 77 | } |
| 78 | |
| 79 | var resp output.Response |
| 80 | if err := json.Unmarshal(buf.Bytes(), &resp); err != nil { |
| 81 | t.Fatalf("unmarshal: %v", err) |
| 82 | } |
| 83 | |
| 84 | if resp.Meta != nil { |
| 85 | if _, ok := resp.Meta["stats"]; ok { |
| 86 | t.Error("expected meta.stats to be absent when --stats is off") |
| 87 | } |
| 88 | } |
| 89 | } |