go-toml/cmd/tomljson/main_test.go

package main

import (
	"bytes"
	"errors"
	"io"
	"strings"
	"testing"

	"git.ostiwe.com/ostiwe/go-toml/v2/internal/assert"
)

func TestConvert(t *testing.T) {
	examples := []struct {
		name     string
		input    io.Reader
		expected string
		errors   bool
	}{
		{
			name: "valid toml",
			input: strings.NewReader(`
[mytoml]
a = 42`),
			expected: `{
  "mytoml": {
    "a": 42
  }
}
`,
		},
		{
			name:   "invalid toml",
			input:  strings.NewReader(`bad = []]`),
			errors: true,
		},
		{
			name:   "bad reader",
			input:  &badReader{},
			errors: true,
		},
	}

	for _, e := range examples {
		b := new(bytes.Buffer)
		err := convert(e.input, b)
		if e.errors {
			assert.Error(t, err)
		} else {
			assert.NoError(t, err)
			assert.Equal(t, e.expected, b.String())
		}
	}
}

type badReader struct{}

func (r *badReader) Read([]byte) (int, error) {
	return 0, errors.New("reader failed on purpose")
}