github.com/braveheart12/insolar-09-08-19@v0.8.7/ledger/storage/jetstorage_test.go (about) 1 /* 2 * Copyright 2019 Insolar Technologies 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package storage 18 19 import ( 20 "testing" 21 22 "github.com/stretchr/testify/require" 23 24 "github.com/insolar/insolar/instrumentation/inslogger" 25 "github.com/insolar/insolar/ledger/storage/jet" 26 ) 27 28 func TestJetStorage_GetJetTree(t *testing.T) { 29 ctx := inslogger.TestContext(t) 30 js := NewJetStorage() 31 32 tree := js.(*jetStorage).getJetTree(ctx, 100) 33 require.Equal(t, "root (level=0 actual=false)\n", tree.String()) 34 } 35 36 func TestJetStorage_UpdateJetTree(t *testing.T) { 37 ctx := inslogger.TestContext(t) 38 js := NewJetStorage() 39 40 js.UpdateJetTree(ctx, 100, true, *jet.NewID(0, nil)) 41 42 tree := js.(*jetStorage).getJetTree(ctx, 100) 43 require.Equal(t, "root (level=0 actual=true)\n", tree.String()) 44 } 45 46 func TestJetStorage_SplitJetTree(t *testing.T) { 47 ctx := inslogger.TestContext(t) 48 js := NewJetStorage() 49 50 l, r, err := js.SplitJetTree(ctx, 100, *jet.NewID(0, nil)) 51 require.NoError(t, err) 52 require.Equal(t, "[JET 1 0]", l.DebugString()) 53 require.Equal(t, "[JET 1 1]", r.DebugString()) 54 55 tree := js.(*jetStorage).getJetTree(ctx, 100) 56 require.Equal(t, "root (level=0 actual=false)\n 0 (level=1 actual=false)\n 1 (level=1 actual=false)\n", tree.String()) 57 } 58 59 func TestJetStorage_CloneJetTree(t *testing.T) { 60 ctx := inslogger.TestContext(t) 61 js := NewJetStorage() 62 63 js.UpdateJetTree(ctx, 100, true, *jet.NewID(0, nil)) 64 65 tree := js.(*jetStorage).getJetTree(ctx, 100) 66 require.Equal(t, "root (level=0 actual=true)\n", tree.String()) 67 68 tree = js.CloneJetTree(ctx, 100, 101) 69 require.Equal(t, "root (level=0 actual=false)\n", tree.String()) 70 71 tree = js.(*jetStorage).getJetTree(ctx, 101) 72 require.Equal(t, "root (level=0 actual=false)\n", tree.String()) 73 74 tree = js.(*jetStorage).getJetTree(ctx, 100) 75 require.Equal(t, "root (level=0 actual=true)\n", tree.String()) 76 } 77 78 func TestJetStorage_DeleteJetTree(t *testing.T) { 79 ctx := inslogger.TestContext(t) 80 js := NewJetStorage() 81 82 _, _, err := js.SplitJetTree(ctx, 100, *jet.NewID(0, nil)) 83 require.NoError(t, err) 84 85 js.DeleteJetTree(ctx, 100) 86 87 tree := js.(*jetStorage).getJetTree(ctx, 100) 88 require.NoError(t, err) 89 require.Equal(t, "root (level=0 actual=false)\n", tree.String()) 90 }