cid/cid_test.go

package cid

import (
	"bytes"
	"testing"

	mh "github.com/jbenet/go-multihash"
)

func assertEqual(t *testing.T, a, b *Cid) {
	if a.Type != b.Type {
		t.Fatal("mismatch on type")
	}

	if a.Version != b.Version {
		t.Fatal("mismatch on version")
	}

	if !bytes.Equal(a.Hash, b.Hash) {
		t.Fatal("multihash mismatch")
	}
}

func TestBasicMarshaling(t *testing.T) {
	h, err := mh.Sum([]byte("TEST"), mh.SHA3, 4)
	if err != nil {
		t.Fatal(err)
	}

	cid := &Cid{
		Type:    7,
		Version: 1,
		Hash:    h,
	}

	data, err := cid.Bytes()
	if err != nil {
		t.Fatal(err)
	}

	out, err := Cast(data)
	if err != nil {
		t.Fatal(err)
	}

	assertEqual(t, cid, out)

	s := cid.String()
	out2, err := Decode(s)
	if err != nil {
		t.Fatal(err)
	}

	assertEqual(t, cid, out2)
}

func TestV0Handling(t *testing.T) {
	old := "QmdfTbBqBPQ7VNxZEYEj14VmRuZBkqFbiwReogJgS1zR1n"

	cid, err := Decode(old)
	if err != nil {
		t.Fatal(err)
	}

	if cid.Version != 0 {
		t.Fatal("should have gotten version 0 cid")
	}

	if cid.Hash.B58String() != old {
		t.Fatal("marshaling roundtrip failed")
	}

	if cid.String() != old {
		t.Fatal("marshaling roundtrip failed")
	}
}

func TestV0ErrorCases(t *testing.T) {
	badb58 := "QmdfTbBqBPQ7VNxZEYEj14VmRuZBkqFbiwReogJgS1zIII"
	_, err := Decode(badb58)
	if err == nil {
		t.Fatal("should have failed to decode that ref")
	}
}

func TestBadVersion(t *testing.T) {
	c := &Cid{
		Version: 17,
	}

	if c.String() != UnsupportedVersionString {
		t.Fatal("expected unsup string")
	}

	_, err := c.Bytes()
	if err == nil {
		t.Fatal("shouldnt have succeeded in calling bytes")
	}
}