fix: missing wal while make

Author: anrs

Makefile

@@ -54,7 +54,8 @@ unit-test:
 	./resources/types/. \
 	./resources/storage/... \
 	./resources/volume/... \
-	./resources/cpumem/...
+	./resources/cpumem/... \
+	./wal/...
 
 lint:
 	golangci-lint run

wal/kv/mocked.go

@@ -92,16 +92,14 @@ func (m *MockedKV) Scan(ctx context.Context, prefix []byte) (<-chan ScanEntry, f
 		defer close(ch)
 
 		m.pool.Range(func(rkey, rvalue interface{}) (next bool) {
-			select {
-			case <-exit:
-				next = false
-			default:
-				next = true
-			}
-
 			var entry MockedScanEntry
 			defer func() {
-				ch <- entry
+				select {
+				case <-exit:
+					next = false
+				case ch <- entry:
+					next = true
+				}
 			}()
 
 			var ok bool

wal/kv/mocked_test.go

@@ -0,0 +1,49 @@
+package kv
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestMockedKV(t *testing.T) {
+	m := &MockedKV{}
+	require.NoError(t, m.Open(context.Background(), "/tmp/wal", 0777, time.Second))
+
+	a := []byte("/a")
+	b := []byte("/b")
+	expValue := []byte("v")
+	ctx := context.Background()
+	require.NoError(t, m.Put(ctx, a, expValue))
+	require.NoError(t, m.Put(ctx, b, expValue))
+	require.NoError(t, m.Put(ctx, []byte("out-of-scan"), expValue))
+
+	ch, abort := m.Scan(ctx, []byte("/"))
+	elem := <-ch
+	ent, ok := elem.(MockedScanEntry)
+	require.True(t, ok)
+	require.NoError(t, ent.Err)
+	abort()
+
+	_, abort = m.Scan(ctx, []byte("/"))
+	abort()
+
+	realValue, err := m.Get(ctx, a)
+	require.NoError(t, err)
+	require.Equal(t, expValue, realValue)
+
+	realValue, err = m.Get(ctx, b)
+	require.NoError(t, err)
+	require.Equal(t, expValue, realValue)
+
+	require.NoError(t, m.Delete(ctx, b))
+	realValue, err = m.Get(ctx, b)
+	require.Error(t, err)
+
+	require.NoError(t, m.Close(ctx))
+
+	realValue, err = m.Get(ctx, a)
+	require.Error(t, err)
+}