// Copyright 2018 Vulcanize // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package vow_flog_test import ( . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" "github.com/vulcanize/vulcanizedb/pkg/transformers/shared/constants" "github.com/vulcanize/vulcanizedb/pkg/transformers/test_data/shared_behaviors" "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/vow_flog" "github.com/vulcanize/vulcanizedb/test_config" ) var _ = Describe("Vow flog repository", func() { var ( db *postgres.DB repository vow_flog.VowFlogRepository ) BeforeEach(func() { db = test_config.NewTestDB(test_config.NewTestNode()) test_config.CleanTestDB(db) repository = vow_flog.VowFlogRepository{} repository.SetDB(db) }) Describe("Create", func() { modelWithDifferentLogIdx := test_data.VowFlogModel modelWithDifferentLogIdx.LogIndex++ inputs := shared_behaviors.CreateBehaviorInputs{ CheckedHeaderColumnName: constants.VowFlogChecked, LogEventTableName: "maker.vow_flog", TestModel: test_data.VowFlogModel, ModelWithDifferentLogIdx: modelWithDifferentLogIdx, Repository: &repository, } shared_behaviors.SharedRepositoryCreateBehaviors(&inputs) It("adds a vow flog event", func() { headerRepository := repositories.NewHeaderRepository(db) headerID, err := headerRepository.CreateOrUpdateHeader(fakes.FakeHeader) Expect(err).NotTo(HaveOccurred()) err = repository.Create(headerID, []interface{}{test_data.VowFlogModel}) Expect(err).NotTo(HaveOccurred()) var dbFlog vow_flog.VowFlogModel err = db.Get(&dbFlog, `SELECT era, log_idx, tx_idx, raw_log FROM maker.vow_flog WHERE header_id = $1`, headerID) Expect(err).NotTo(HaveOccurred()) Expect(dbFlog.Era).To(Equal(test_data.VowFlogModel.Era)) Expect(dbFlog.LogIndex).To(Equal(test_data.VowFlogModel.LogIndex)) Expect(dbFlog.TransactionIndex).To(Equal(test_data.VowFlogModel.TransactionIndex)) Expect(dbFlog.Raw).To(MatchJSON(test_data.VowFlogModel.Raw)) }) }) Describe("MarkHeaderChecked", func() { inputs := shared_behaviors.MarkedHeaderCheckedBehaviorInputs{ CheckedHeaderColumnName: constants.VowFlogChecked, Repository: &repository, } shared_behaviors.SharedRepositoryMarkHeaderCheckedBehaviors(&inputs) }) })