package test_config import ( "log" "os" . "github.com/onsi/gomega" "github.com/spf13/viper" "github.com/vulcanize/vulcanizedb/pkg/config" "github.com/vulcanize/vulcanizedb/pkg/core" "github.com/vulcanize/vulcanizedb/pkg/datastore/postgres" "github.com/vulcanize/vulcanizedb/pkg/datastore/postgres/repositories" ) var TestConfig *viper.Viper var DBConfig config.Database var TestClient config.Client var Infura *viper.Viper var InfuraClient config.Client var ABIFilePath string func init() { setTestConfig() setInfuraConfig() setABIPath() } func setTestConfig() { TestConfig = viper.New() TestConfig.SetConfigName("private") TestConfig.AddConfigPath("$GOPATH/src/github.com/vulcanize/vulcanizedb/environments/") err := TestConfig.ReadInConfig() ipc := TestConfig.GetString("client.ipcPath") if err != nil { log.Fatal(err) } hn := TestConfig.GetString("database.hostname") port := TestConfig.GetInt("database.port") name := TestConfig.GetString("database.name") DBConfig = config.Database{ Hostname: hn, Name: name, Port: port, } TestClient = config.Client{ IPCPath: ipc, } } func setInfuraConfig() { Infura = viper.New() Infura.SetConfigName("infura") Infura.AddConfigPath("$GOPATH/src/github.com/vulcanize/vulcanizedb/environments/") err := Infura.ReadInConfig() ipc := Infura.GetString("client.ipcpath") if err != nil { log.Fatal(err) } InfuraClient = config.Client{ IPCPath: ipc, } } func setABIPath() { gp := os.Getenv("GOPATH") ABIFilePath = gp + "/src/github.com/vulcanize/vulcanizedb/pkg/geth/testing/" } func NewTestDB(node core.Node) *postgres.DB { db, _ := postgres.NewDB(DBConfig, node) return db } func CleanTestDB(db *postgres.DB) { db.MustExec("DELETE FROM blocks") db.MustExec("DELETE FROM headers") db.MustExec("DELETE FROM checked_headers") db.MustExec("DELETE FROM log_filters") db.MustExec("DELETE FROM logs") db.MustExec("DELETE FROM maker.bite") db.MustExec("DELETE FROM maker.cat_file_chop_lump") db.MustExec("DELETE FROM maker.cat_file_flip") db.MustExec("DELETE FROM maker.cat_file_pit_vow") db.MustExec("DELETE FROM maker.deal") db.MustExec("DELETE FROM maker.dent") db.MustExec("DELETE FROM maker.drip_drip") db.MustExec("DELETE FROM maker.drip_file_ilk") db.MustExec("DELETE FROM maker.drip_file_repo") db.MustExec("DELETE FROM maker.drip_file_vow") db.MustExec("DELETE FROM maker.flip_kick") db.MustExec("DELETE FROM maker.vow_flog") db.MustExec("DELETE FROM maker.flop_kick") db.MustExec("DELETE FROM maker.frob") db.MustExec("DELETE FROM maker.pit_file_debt_ceiling") db.MustExec("DELETE FROM maker.pit_file_ilk") db.MustExec("DELETE FROM maker.pit_file_stability_fee") db.MustExec("DELETE FROM maker.price_feeds") db.MustExec("DELETE FROM maker.tend") db.MustExec("DELETE FROM maker.vat_grab") db.MustExec("DELETE FROM maker.vat_heal") db.MustExec("DELETE FROM maker.vat_init") db.MustExec("DELETE FROM maker.vat_move") db.MustExec("DELETE FROM maker.vat_fold") db.MustExec("DELETE FROM maker.vat_slip") db.MustExec("DELETE FROM maker.vat_toll") db.MustExec("DELETE FROM maker.vat_tune") db.MustExec("DELETE FROM maker.vat_flux") db.MustExec("DELETE FROM receipts") db.MustExec("DELETE FROM transactions") db.MustExec("DELETE FROM watched_contracts") } func NewTestNode() core.Node { return core.Node{ GenesisBlock: "GENESIS", NetworkID: 1, ID: "b6f90c0fdd8ec9607aed8ee45c69322e47b7063f0bfb7a29c8ecafab24d0a22d24dd2329b5ee6ed4125a03cb14e57fd584e67f9e53e6c631055cbbd82f080845", ClientName: "Geth/v1.7.2-stable-1db4ecdc/darwin-amd64/go1.9", } } func NewTestBlock(blockNumber int64, repository repositories.BlockRepository) (blockId int64) { blockId, err := repository.CreateOrUpdateBlock(core.Block{Number: blockNumber}) Expect(err).NotTo(HaveOccurred()) return blockId }