storage_test.go

package mocha

import (
	"sync"
	"testing"

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

func TestInMemoryStorage(t *testing.T) {
	st := newStorage()
	st.Save(&Mock{ID: 1, Name: "mock_1", Enabled: true, mu: &sync.Mutex{}, Priority: 0})
	st.Save(&Mock{ID: 2, Name: "mock_2", Enabled: true, mu: &sync.Mutex{}, Priority: 1})
	st.Save(&Mock{ID: 3, Name: "mock_3", Enabled: true, mu: &sync.Mutex{}, Priority: 2})

	m := st.FetchAll()[0]
	assert.Equal(t, m.ID, 1)
	assert.Equal(t, m.Name, "mock_1")

	m.Disable()

	mocks := st.FetchEligible()
	assert.Len(t, mocks, 2)

	mocks = st.FetchAll()
	assert.Len(t, mocks, 3)

	st.Delete(2)
	mocks = st.FetchAll()
	assert.Len(t, mocks, 2)

	mocks = st.FetchEligible()
	assert.Len(t, mocks, 1)
	assert.Equal(t, 3, mocks[0].ID)

	st.Flush()

	mocks = st.FetchAll()

	assert.Len(t, mocks, 0)
}