main_test.go

package main

import (
	"bytes"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestConvert(t *testing.T) {
	examples := []struct {
		name     string
		input    string
		expected string
		errors   bool
	}{
		{
			name: "valid toml",
			input: `
mytoml.a = 42.0
`,
			expected: `[mytoml]
a = 42.0
`,
		},
		{
			name:   "invalid toml",
			input:  `[what`,
			errors: true,
		},
	}

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