github.com/cloudberrydb/gpbackup@v1.0.3-0.20240118031043-5410fd45eed6/restore/restore_suite_test.go (about) 1 package restore_test 2 3 /* 4 * This file contains integration tests for gprestore as a whole, rather than 5 * tests relating to functions in any particular file. 6 */ 7 8 import ( 9 "testing" 10 11 "github.com/DATA-DOG/go-sqlmock" 12 "github.com/cloudberrydb/gp-common-go-libs/dbconn" 13 "github.com/cloudberrydb/gpbackup/restore" 14 "github.com/cloudberrydb/gpbackup/testutils" 15 "github.com/spf13/pflag" 16 17 . "github.com/onsi/ginkgo/v2" 18 . "github.com/onsi/gomega" 19 . "github.com/onsi/gomega/gbytes" 20 ) 21 22 var ( 23 connectionPool *dbconn.DBConn 24 mock sqlmock.Sqlmock 25 stdout *Buffer 26 logfile *Buffer 27 buffer *Buffer 28 ) 29 30 func TestRestore(t *testing.T) { 31 RegisterFailHandler(Fail) 32 RunSpecs(t, "restore tests") 33 } 34 35 var cmdFlags *pflag.FlagSet 36 37 var _ = BeforeEach(func() { 38 connectionPool, mock, stdout, _, logfile = testutils.SetupTestEnvironment() 39 restore.SetConnection(connectionPool) 40 buffer = NewBuffer() 41 42 cmdFlags = pflag.NewFlagSet("gprestore", pflag.ExitOnError) 43 restore.SetCmdFlags(cmdFlags) 44 })