aboutsummaryrefslogblamecommitdiffstats
path: root/common/rlp_test.go
blob: 16a3553d753bf390b3e89168467dbe97675ab02a (plain) (tree)
1
2
3
4
5
6
7
8
9
              


               




                  









                                                                         
                                         
                           


































                                                                                   










                                                                             
















                                                                              
                                                            






                                                  
                                                               



































                                                              


                                                                       









                                                                    

                                   



                                                  

 





                                                                   
package common

import (
    "bytes"
    "math/big"
    "reflect"
    "testing"
)

func TestNonInterfaceSlice(t *testing.T) {
    vala := []string{"value1", "value2", "value3"}
    valb := []interface{}{"value1", "value2", "value3"}
    resa := Encode(vala)
    resb := Encode(valb)
    if !bytes.Equal(resa, resb) {
        t.Errorf("expected []string & []interface{} to be equal")
    }
}

func TestRlpValueEncoding(t *testing.T) {
    val := EmptyValue()
    val.AppendList().Append(1).Append(2).Append(3)
    val.Append("4").AppendList().Append(5)

    res := val.Encode()
    exp := Encode([]interface{}{[]interface{}{1, 2, 3}, "4", []interface{}{5}})
    if bytes.Compare(res, exp) != 0 {
        t.Errorf("expected %q, got %q", res, exp)
    }
}

func TestValueSlice(t *testing.T) {
    val := []interface{}{
        "value1",
        "valeu2",
        "value3",
    }

    value := NewValue(val)
    splitVal := value.SliceFrom(1)

    if splitVal.Len() != 2 {
        t.Error("SliceFrom: Expected len", 2, "got", splitVal.Len())
    }

    splitVal = value.SliceTo(2)
    if splitVal.Len() != 2 {
        t.Error("SliceTo: Expected len", 2, "got", splitVal.Len())
    }

    splitVal = value.SliceFromTo(1, 3)
    if splitVal.Len() != 2 {
        t.Error("SliceFromTo: Expected len", 2, "got", splitVal.Len())
    }
}

func TestLargeData(t *testing.T) {
    data := make([]byte, 100000)
    enc := Encode(data)
    value := NewValue(enc)
    value.Decode()

    if value.Len() != len(data) {
        t.Error("Expected data to be", len(data), "got", value.Len())
    }
}

func TestValue(t *testing.T) {
    value := NewValueFromBytes([]byte("\xcd\x83dog\x83god\x83cat\x01"))
    if value.Get(0).Str() != "dog" {
        t.Errorf("expected '%v', got '%v'", value.Get(0).Str(), "dog")
    }

    if value.Get(3).Uint() != 1 {
        t.Errorf("expected '%v', got '%v'", value.Get(3).Uint(), 1)
    }
}

func TestEncode(t *testing.T) {
    strRes := "\x83dog"
    bytes := Encode("dog")

    str := string(bytes)
    if str != strRes {
        t.Errorf("Expected %q, got %q", strRes, str)
    }

    sliceRes := "\xcc\x83dog\x83god\x83cat"
    strs := []interface{}{"dog", "god", "cat"}
    bytes = Encode(strs)
    slice := string(bytes)
    if slice != sliceRes {
        t.Error("Expected %q, got %q", sliceRes, slice)
    }

    intRes := "\x82\x04\x00"
    bytes = Encode(1024)
    if string(bytes) != intRes {
        t.Errorf("Expected %q, got %q", intRes, bytes)
    }
}

func TestDecode(t *testing.T) {
    single := []byte("\x01")
    b, _ := Decode(single, 0)

    if b.(uint8) != 1 {
        t.Errorf("Expected 1, got %q", b)
    }

    str := []byte("\x83dog")
    b, _ = Decode(str, 0)
    if bytes.Compare(b.([]byte), []byte("dog")) != 0 {
        t.Errorf("Expected dog, got %q", b)
    }

    slice := []byte("\xcc\x83dog\x83god\x83cat")
    res := []interface{}{"dog", "god", "cat"}
    b, _ = Decode(slice, 0)
    if reflect.DeepEqual(b, res) {
        t.Errorf("Expected %q, got %q", res, b)
    }
}

func TestEncodeDecodeBigInt(t *testing.T) {
    bigInt := big.NewInt(1391787038)
    encoded := Encode(bigInt)

    value := NewValueFromBytes(encoded)
    if value.BigInt().Cmp(bigInt) != 0 {
        t.Errorf("Expected %v, got %v", bigInt, value.BigInt())
    }
}

func TestEncodeDecodeBytes(t *testing.T) {
    b := NewValue([]interface{}{[]byte{1, 2, 3, 4, 5}, byte(6)})
    val := NewValueFromBytes(b.Encode())
    if !b.Cmp(val) {
        t.Errorf("Expected %v, got %v", val, b)
    }
}

func TestEncodeZero(t *testing.T) {
    b := NewValue(0).Encode()
    exp := []byte{0xc0}
    if bytes.Compare(b, exp) == 0 {
        t.Error("Expected", exp, "got", b)
    }
}

func BenchmarkEncodeDecode(b *testing.B) {
    for i := 0; i < b.N; i++ {
        bytes := Encode([]interface{}{"dog", "god", "cat"})
        Decode(bytes, 0)
    }
}