ipld-eth-server/pkg/transformers/shared/repository_utility_test.go

// 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 shared_test

import (
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	"github.com/vulcanize/vulcanizedb/pkg/transformers/shared"
	"github.com/vulcanize/vulcanizedb/test_config"
)

var _ = Describe("Repository utilities", func() {
	Describe("GetCheckedColumnNames", func() {
		It("gets the column names from checked_headers", func() {
			db := test_config.NewTestDB(test_config.NewTestNode())
			test_config.CleanTestDB(db)
			expectedColumnNames := getExpectedColumnNames()
			actualColumnNames, err := shared.Repository{}.GetCheckedColumnNames(db)
			Expect(err).NotTo(HaveOccurred())
			Expect(actualColumnNames).To(Equal(expectedColumnNames))
		})
	})

	Describe("CreateNotCheckedSQL", func() {
		It("generates a correct SQL string for one column", func() {
			columns := []string{"columnA"}
			expected := "NOT (columnA)"
			actual := shared.Repository{}.CreateNotCheckedSQL(columns)
			Expect(actual).To(Equal(expected))
		})

		It("generates a correct SQL string for several columns", func() {
			columns := []string{"columnA", "columnB"}
			expected := "NOT (columnA AND columnB)"
			actual := shared.Repository{}.CreateNotCheckedSQL(columns)
			Expect(actual).To(Equal(expected))
		})

		It("defaults to FALSE when there are no columns", func() {
			expected := "FALSE"
			actual := shared.Repository{}.CreateNotCheckedSQL([]string{})
			Expect(actual).To(Equal(expected))
		})
	})
})

func getExpectedColumnNames() []string {
	return []string{
		"price_feeds_checked",
		"flip_kick_checked",
		"frob_checked",
		"tend_checked",
		"bite_checked",
		"dent_checked",
		"pit_file_debt_ceiling_checked",
		"pit_file_ilk_checked",
		"vat_init_checked",
		"drip_file_ilk_checked",
		"drip_file_repo_checked",
		"drip_file_vow_checked",
		"deal_checked",
		"drip_drip_checked",
		"cat_file_chop_lump_checked",
		"cat_file_flip_checked",
		"cat_file_pit_vow_checked",
		"flop_kick_checked",
		"vat_move_checked",
		"vat_fold_checked",
		"vat_heal_checked",
		"vat_toll_checked",
		"vat_tune_checked",
		"vat_grab_checked",
		"vat_flux_checked",
		"vat_slip_checked",
		"vow_flog_checked",
		"flap_kick_checked",
	}
}