// VulcanizeDB // Copyright © 2018 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 . package vat_init_test import ( . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" "github.com/vulcanize/vulcanizedb/pkg/transformers/shared" "github.com/vulcanize/vulcanizedb/pkg/transformers/shared/constants" "github.com/vulcanize/vulcanizedb/pkg/transformers/test_data/shared_behaviors" "strconv" "github.com/vulcanize/vulcanizedb/pkg/datastore/postgres" "github.com/vulcanize/vulcanizedb/pkg/datastore/postgres/repositories" "github.com/vulcanize/vulcanizedb/pkg/fakes" "github.com/vulcanize/vulcanizedb/pkg/transformers/test_data" "github.com/vulcanize/vulcanizedb/pkg/transformers/vat_init" "github.com/vulcanize/vulcanizedb/test_config" ) var _ = Describe("Vat init repository", func() { var ( db *postgres.DB repository vat_init.VatInitRepository ) BeforeEach(func() { db = test_config.NewTestDB(test_config.NewTestNode()) test_config.CleanTestDB(db) repository = vat_init.VatInitRepository{} repository.SetDB(db) }) Describe("Create", func() { modelWithDifferentLogIdx := test_data.VatInitModel modelWithDifferentLogIdx.LogIndex++ inputs := shared_behaviors.CreateBehaviorInputs{ CheckedHeaderColumnName: constants.VatInitChecked, LogEventTableName: "maker.vat_init", TestModel: test_data.VatInitModel, ModelWithDifferentLogIdx: modelWithDifferentLogIdx, Repository: &repository, } shared_behaviors.SharedRepositoryCreateBehaviors(&inputs) It("persists vat init records", func() { headerRepository := repositories.NewHeaderRepository(db) headerID, err := headerRepository.CreateOrUpdateHeader(fakes.FakeHeader) Expect(err).NotTo(HaveOccurred()) err = repository.Create(headerID, []interface{}{test_data.VatInitModel}) Expect(err).NotTo(HaveOccurred()) var dbVatInit vat_init.VatInitModel err = db.Get(&dbVatInit, `SELECT ilk, log_idx, tx_idx, raw_log FROM maker.vat_init WHERE header_id = $1`, headerID) Expect(err).NotTo(HaveOccurred()) ilkID, err := shared.GetOrCreateIlk(test_data.VatInitModel.Ilk, db) Expect(err).NotTo(HaveOccurred()) Expect(dbVatInit.Ilk).To(Equal(strconv.Itoa(ilkID))) Expect(dbVatInit.LogIndex).To(Equal(test_data.VatInitModel.LogIndex)) Expect(dbVatInit.TransactionIndex).To(Equal(test_data.VatInitModel.TransactionIndex)) Expect(dbVatInit.Raw).To(MatchJSON(test_data.VatInitModel.Raw)) }) }) Describe("MarkHeaderChecked", func() { inputs := shared_behaviors.MarkedHeaderCheckedBehaviorInputs{ CheckedHeaderColumnName: constants.VatInitChecked, Repository: &repository, } shared_behaviors.SharedRepositoryMarkHeaderCheckedBehaviors(&inputs) }) })