github.com/hyperledger/burrow@v0.34.5-0.20220512172541-77f09336001d/vent/sqlsol/block_data_test.go (about)

     1  package sqlsol_test
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/hyperledger/burrow/vent/sqlsol"
     7  	"github.com/hyperledger/burrow/vent/types"
     8  	"github.com/stretchr/testify/require"
     9  )
    10  
    11  func TestSetRow(t *testing.T) {
    12  	t.Run("successfully sets a new data row", func(t *testing.T) {
    13  		values := make(map[string]interface{})
    14  		values["c1"] = "v1"
    15  		values["c2"] = "v2"
    16  
    17  		blockData := sqlsol.NewBlockData(44)
    18  		blockData.AddRow("TEST_TABLE", types.EventDataRow{Action: types.ActionUpsert, RowData: values})
    19  
    20  		rows, err := blockData.GetRows("TEST_TABLE")
    21  		require.NoError(t, err)
    22  		require.Equal(t, 1, len(rows))
    23  		require.Equal(t, "v1", rows[0].RowData["c1"])
    24  		require.Equal(t, "v2", rows[0].RowData["c2"])
    25  	})
    26  }
    27  
    28  func TestGetBlockData(t *testing.T) {
    29  	t.Run("successfully gets block data", func(t *testing.T) {
    30  		blockData := sqlsol.NewBlockData(2)
    31  		blk := blockData.Data
    32  		require.EqualValues(t, 2, blk.BlockHeight)
    33  	})
    34  }