112 lines
3.3 KiB
Go
112 lines
3.3 KiB
Go
// VulcanizeDB
|
|
// Copyright © 2022 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 <http://www.gnu.org/licenses/>.
|
|
package postgres_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/big"
|
|
"strings"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
"github.com/vulcanize/ipld-ethcl-indexer/pkg/database/sql"
|
|
"github.com/vulcanize/ipld-ethcl-indexer/pkg/database/sql/postgres"
|
|
"github.com/vulcanize/ipld-ethcl-indexer/pkg/testhelpers"
|
|
)
|
|
|
|
var _ = Describe("Pgx", func() {
|
|
|
|
var (
|
|
ctx context.Context
|
|
)
|
|
|
|
BeforeEach(func() {
|
|
ctx = context.Background()
|
|
})
|
|
|
|
Describe("Connecting to the DB", Label("integration"), func() {
|
|
Context("But connection is unsucessful", func() {
|
|
It("throws error when can't connect to the database", func() {
|
|
_, err := postgres.NewPostgresDB(postgres.Config{
|
|
Driver: "PGX",
|
|
})
|
|
Expect(err).To(HaveOccurred())
|
|
|
|
present, err := doesContainsSubstring(err.Error(), sql.DbConnectionFailedMsg)
|
|
Expect(present).To(BeTrue())
|
|
Expect(err).NotTo(HaveOccurred())
|
|
})
|
|
})
|
|
Context("The connection is successful", func() {
|
|
It("Should create a DB object", func() {
|
|
db, err := postgres.NewPostgresDB(postgres.DefaultConfig)
|
|
Expect(err).To(BeNil())
|
|
defer db.Close()
|
|
})
|
|
})
|
|
})
|
|
Describe("Write to the DB", Label("integration"), func() {
|
|
Context("Serialize big.Int to DB", func() {
|
|
It("Should serialize successfully", func() {
|
|
dbPool, err := postgres.NewPostgresDB(postgres.DefaultConfig)
|
|
Expect(err).To(BeNil())
|
|
defer dbPool.Close()
|
|
|
|
bi := new(big.Int)
|
|
bi.SetString("34940183920000000000", 10)
|
|
isEqual, err := testhelpers.IsEqual(bi.String(), "34940183920000000000")
|
|
Expect(err).To(BeNil())
|
|
Expect(isEqual).To(BeTrue())
|
|
|
|
defer func() {
|
|
_, err := dbPool.Exec(ctx, `DROP TABLE IF EXISTS example`)
|
|
Expect(err).To(BeNil())
|
|
}()
|
|
_, err = dbPool.Exec(ctx, "CREATE TABLE example ( id INTEGER, data NUMERIC )")
|
|
Expect(err).To(BeNil())
|
|
|
|
sqlStatement := `
|
|
INSERT INTO example (id, data)
|
|
VALUES (1, cast($1 AS NUMERIC))`
|
|
_, err = dbPool.Exec(ctx, sqlStatement, bi.String())
|
|
Expect(err).To(BeNil())
|
|
|
|
var data string
|
|
err = dbPool.QueryRow(ctx, `SELECT cast(data AS TEXT) FROM example WHERE id = 1`).Scan(&data)
|
|
Expect(err).To(BeNil())
|
|
|
|
isEqual, err = testhelpers.IsEqual(data, bi.String())
|
|
Expect(isEqual).To(BeTrue())
|
|
Expect(err).To(BeNil())
|
|
actual := new(big.Int)
|
|
actual.SetString(data, 10)
|
|
|
|
isEqual, err = testhelpers.IsEqual(actual, bi)
|
|
Expect(isEqual).To(BeTrue())
|
|
Expect(err).To(BeNil())
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
func doesContainsSubstring(full string, sub string) (bool, error) {
|
|
if !strings.Contains(full, sub) {
|
|
return false, fmt.Errorf("Expected \"%v\" to contain substring \"%v\"\n", full, sub)
|
|
}
|
|
return true, nil
|
|
}
|