docdb/cmd/envtool/envtool_test.go

// Copyright 2021 DocDB Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package main

import (
	"bytes"
	"os"
	"testing"

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

	"github.com/hanzoai/docdb/internal/util/testutil"
)

func TestPrintDiagnosticData(t *testing.T) {
	t.Parallel()

	assert.NotPanics(t, func() {
		var buf bytes.Buffer
		l := testutil.Logger(t)
		err := printDiagnosticData(&buf, nil, l)
		require.NoError(t, err)
	})
}

func TestShellMkDirRmDir(t *testing.T) {
	t.Parallel()

	t.Run("Absent", func(t *testing.T) {
		err := shellRmDir("absent")
		assert.NoError(t, err)
	})

	paths := []string{"ab/c", "ab"}

	err := shellMkDir(paths...)
	assert.NoError(t, err)

	for _, path := range paths {
		assert.DirExists(t, path)
	}

	err = shellRmDir(paths...)
	assert.NoError(t, err)

	for _, path := range paths {
		assert.NoDirExists(t, path)
	}
}

func TestShellRead(t *testing.T) {
	t.Parallel()

	f, err := os.CreateTemp("", "test_read")
	assert.NoError(t, err)

	s := "test string in a file"
	_, err = f.Write([]byte(s))
	assert.NoError(t, err)

	var output bytes.Buffer
	err = shellRead(&output, f.Name())
	assert.NoError(t, err)
	assert.Equal(t, s, output.String())
}

func TestPackageVersion(t *testing.T) {
	t.Parallel()

	f, err := os.CreateTemp("", "test_print_version")
	assert.NoError(t, err)

	s := "v1.0.0"
	_, err = f.Write([]byte(s))
	assert.NoError(t, err)

	var output bytes.Buffer
	err = packageVersion(&output, f.Name())
	assert.NoError(t, err)
	assert.Equal(t, "1.0.0", output.String())
}