write_markdown_test.go

package main

import (
	"bytes"
	"testing"
)

func TestWriteMarkdown(t *testing.T) {
	expected := `Network policy viewer
=====================

|**Field**                  |**Value**|
|:--------------------------|:--------|
|percentageIsolated         |20       |
|percentageNamespaceCoverage|30       |

`

	var buffer bytes.Buffer
	writeMarkdown(20, 30, &buffer)
	s := buffer.String()
	if s != expected {
		t.Errorf("Markdown output faulty\nExpected:\n%s\nGot:\n%s\n", expected, s)
	}
}

func TestMarkdownHeading(t *testing.T) {
	expectedH1 := `Foobar
======

`
	expectedH3 := `### Foobar

`

	var bufferH1 bytes.Buffer
	heading("Foobar", 1, &bufferH1)
	if bufferH1.String() != expectedH1 {
		t.Errorf("Markdown heading faulty\nExpected:\n%s\nGot:\n%s\n", expectedH1, bufferH1.String())
	}

	var bufferH3 bytes.Buffer
	heading("Foobar", 3, &bufferH3)
	if bufferH3.String() != expectedH3 {
		t.Errorf("Expected:\n%s\nGot:\n%s\n", expectedH3, bufferH3.String())
	}
}