storage_test.go

package firecore

import (
	"path/filepath"
	"testing"

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

func Test_searchBlockNum(t *testing.T) {
	tests := []struct {
		name          string
		startBlockNum uint64
		lastBlockNum  *uint64
		expect        int
		expectErr     bool
	}{
		{"golden path", 1_690_600, uptr(208_853_300), 208_853_300, false},
		{"no block file found", 1_690_600, nil, 1_690_600, false},
		{"block file greater then start block", 0, uptr(100), 100, false},
		{"block file less then start block", 200, uptr(100), 200, false},
		{"golden path 2", 0, uptr(17821900), 17821900, false},
	}
	for _, tt := range tests {
		t.Run("", func(t *testing.T) {
			dstoreOpt := 0
			v, err := searchBlockNum(tt.startBlockNum, func(i uint64) (bool, error) {
				dstoreOpt++
				if tt.lastBlockNum == nil {
					return false, nil
				}
				if i > *tt.lastBlockNum {
					return false, nil
				}
				return true, nil
			})
			if tt.expectErr {
				require.Error(t, err)
			} else {
				require.NoError(t, err)
				assert.Equal(t, tt.expect, int(v))
			}
		})
	}
}

func uptr(v uint64) *uint64 {
	return &v
}

func TestGetTmpDir(t *testing.T) {
	dataDir := t.TempDir()
	testSetViper(t, "common-tmp-dir", "{data-dir}/value")

	dir, err := GetTmpDir(dataDir)
	assert.NoError(t, err)
	assert.Equal(t, filepath.Join(dataDir, "value"), dir)

	dir2, err2 := GetTmpDir(dataDir)
	assert.NoError(t, err2)
	assert.Equal(t, filepath.Join(dataDir, "value"), dir2)
}

func testSetViper(t *testing.T, key string, value string) {
	current := viper.Get(key)
	t.Cleanup(func() {
		viper.Set(key, current)
	})
	viper.Set(key, value)
}