90 lines
2.6 KiB
Go
90 lines
2.6 KiB
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.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.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.CreateNotCheckedSQL(columns)
|
|
Expect(actual).To(Equal(expected))
|
|
})
|
|
|
|
It("defaults to FALSE when there are no columns", func() {
|
|
expected := "FALSE"
|
|
actual := shared.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",
|
|
}
|
|
} |