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  })