ipld-eth-server/pkg/super_node/btc/publisher_test.go

// VulcanizeDB
// Copyright © 2019 Vulcanize

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.

// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package btc_test

import (
	"bytes"

	"github.com/ethereum/go-ethereum/common"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"

	mocks2 "github.com/vulcanize/vulcanizedb/pkg/ipfs/mocks"
	"github.com/vulcanize/vulcanizedb/pkg/super_node/btc"
	"github.com/vulcanize/vulcanizedb/pkg/super_node/btc/mocks"
)

var (
	mockHeaderDagPutter  *mocks2.MappedDagPutter
	mockTrxDagPutter     *mocks2.MappedDagPutter
	mockTrxTrieDagPutter *mocks2.DagPutter
)

var _ = Describe("Publisher", func() {
	BeforeEach(func() {
		mockHeaderDagPutter = new(mocks2.MappedDagPutter)
		mockTrxDagPutter = new(mocks2.MappedDagPutter)
		mockTrxTrieDagPutter = new(mocks2.DagPutter)
	})

	Describe("Publish", func() {
		It("Publishes the passed IPLDPayload objects to IPFS and returns a CIDPayload for indexing", func() {
			by := new(bytes.Buffer)
			err := mocks.MockConvertedPayload.BlockPayload.Header.Serialize(by)
			Expect(err).ToNot(HaveOccurred())
			headerBytes := by.Bytes()
			err = mocks.MockTransactions[0].MsgTx().Serialize(by)
			Expect(err).ToNot(HaveOccurred())
			tx1Bytes := by.Bytes()
			err = mocks.MockTransactions[1].MsgTx().Serialize(by)
			Expect(err).ToNot(HaveOccurred())
			tx2Bytes := by.Bytes()
			err = mocks.MockTransactions[2].MsgTx().Serialize(by)
			Expect(err).ToNot(HaveOccurred())
			tx3Bytes := by.Bytes()
			mockHeaderDagPutter.CIDsToReturn = map[common.Hash]string{
				common.BytesToHash(headerBytes): "mockHeaderCID",
			}
			mockTrxDagPutter.CIDsToReturn = map[common.Hash]string{
				common.BytesToHash(tx1Bytes): "mockTrxCID1",
				common.BytesToHash(tx2Bytes): "mockTrxCID2",
				common.BytesToHash(tx3Bytes): "mockTrxCID3",
			}
			publisher := btc.IPLDPublisher{
				HeaderPutter:          mockHeaderDagPutter,
				TransactionPutter:     mockTrxDagPutter,
				TransactionTriePutter: mockTrxTrieDagPutter,
			}
			payload, err := publisher.Publish(mocks.MockConvertedPayload)
			Expect(err).ToNot(HaveOccurred())
			cidPayload, ok := payload.(*btc.CIDPayload)
			Expect(ok).To(BeTrue())
			Expect(cidPayload).To(Equal(&mocks.MockCIDPayload))
			Expect(cidPayload.HeaderCID).To(Equal(mocks.MockHeaderMetaData))
			Expect(cidPayload.TransactionCIDs).To(Equal(mocks.MockTxsMetaDataPostPublish))
		})
	})
})