mirror of
https://github.com/nxshock/zkv.git
synced 2024-11-27 11:21:02 +05:00
Move record tests to separate file
This commit is contained in:
parent
822504e8a0
commit
832d12c43b
35
record_test.go
Normal file
35
record_test.go
Normal file
@ -0,0 +1,35 @@
|
||||
package zkv
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestRecord(t *testing.T) {
|
||||
buf := new(bytes.Buffer)
|
||||
|
||||
var records []Record
|
||||
|
||||
for i := 0; i < 10; i++ {
|
||||
record, err := newRecord(RecordTypeSet, i, i)
|
||||
assert.NoError(t, err)
|
||||
|
||||
records = append(records, *record)
|
||||
|
||||
b, err := record.Marshal()
|
||||
assert.NoError(t, err)
|
||||
|
||||
_, err = buf.Write(b)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
for i := 0; i < 10; i++ {
|
||||
_, record, err := readRecord(buf)
|
||||
assert.NoError(t, err)
|
||||
|
||||
assert.Equal(t, record.KeyHash, records[i].KeyHash)
|
||||
assert.Equal(t, record.ValueBytes, records[i].ValueBytes)
|
||||
}
|
||||
}
|
28
zkv_test.go
28
zkv_test.go
@ -1,40 +1,12 @@
|
||||
package zkv
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestRecord(t *testing.T) {
|
||||
buf := new(bytes.Buffer)
|
||||
|
||||
var records []Record
|
||||
|
||||
for i := 0; i < 10; i++ {
|
||||
record, err := newRecord(RecordTypeSet, i, i)
|
||||
assert.NoError(t, err)
|
||||
|
||||
records = append(records, *record)
|
||||
|
||||
b, err := record.Marshal()
|
||||
assert.NoError(t, err)
|
||||
|
||||
_, err = buf.Write(b)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
for i := 0; i < 10; i++ {
|
||||
_, record, err := readRecord(buf)
|
||||
assert.NoError(t, err)
|
||||
|
||||
assert.Equal(t, record.KeyHash, records[i].KeyHash)
|
||||
assert.Equal(t, record.ValueBytes, records[i].ValueBytes)
|
||||
}
|
||||
}
|
||||
|
||||
func TestReadWriteBasic(t *testing.T) {
|
||||
const filePath = "TestReadWriteBasic.zkv"
|
||||
const recordCount = 100
|
||||
|
Loading…
Reference in New Issue
Block a user