Track completion known to implementation state

[btrtrc.git] / storage / boltpc_test.go

diff --git a/storage/boltpc_test.go b/storage/boltpc_test.go
index 3f5572e711ae97316fe5c5c4a230679975a5b270..53fdbdb590a27a7e26c8c233779e7b10a7ae8acc 100644 (file)
--- a/storage/boltpc_test.go
+++ b/storage/boltpc_test.go
@@ -24,17 +24,17 @@ func TestBoltPieceCompletion(t *testing.T) {
 
        b, err := pc.Get(pk)
        require.NoError(t, err)
-       assert.False(t, b)
+       assert.False(t, b.Ok)
 
        require.NoError(t, pc.Set(pk, false))
 
        b, err = pc.Get(pk)
        require.NoError(t, err)
-       assert.False(t, b)
+       assert.Equal(t, Completion{Complete: false, Ok: true}, b)
 
        require.NoError(t, pc.Set(pk, true))
 
        b, err = pc.Get(pk)
        require.NoError(t, err)
-       assert.True(t, b)
+       assert.Equal(t, Completion{Complete: true, Ok: true}, b)
 }