connector_test.go

Documentation: github.com/lib/pq

     1  //go:build go1.10
     2  // +build go1.10
     3  
     4  package pq
     5  
     6  import (
     7  	"context"
     8  	"database/sql"
     9  	"database/sql/driver"
    10  	"testing"
    11  )
    12  
    13  func TestNewConnector_WorksWithOpenDB(t *testing.T) {
    14  	name := ""
    15  	c, err := NewConnector(name)
    16  	if err != nil {
    17  		t.Fatal(err)
    18  	}
    19  	db := sql.OpenDB(c)
    20  	defer db.Close()
    21  	// database/sql might not call our Open at all unless we do something with
    22  	// the connection
    23  	txn, err := db.Begin()
    24  	if err != nil {
    25  		t.Fatal(err)
    26  	}
    27  	txn.Rollback()
    28  }
    29  
    30  func TestNewConnector_Connect(t *testing.T) {
    31  	name := ""
    32  	c, err := NewConnector(name)
    33  	if err != nil {
    34  		t.Fatal(err)
    35  	}
    36  	db, err := c.Connect(context.Background())
    37  	if err != nil {
    38  		t.Fatal(err)
    39  	}
    40  	defer db.Close()
    41  	// database/sql might not call our Open at all unless we do something with
    42  	// the connection
    43  	txn, err := db.(driver.ConnBeginTx).BeginTx(context.Background(), driver.TxOptions{})
    44  	if err != nil {
    45  		t.Fatal(err)
    46  	}
    47  	txn.Rollback()
    48  }
    49  
    50  func TestNewConnector_Driver(t *testing.T) {
    51  	name := ""
    52  	c, err := NewConnector(name)
    53  	if err != nil {
    54  		t.Fatal(err)
    55  	}
    56  	db, err := c.Driver().Open(name)
    57  	if err != nil {
    58  		t.Fatal(err)
    59  	}
    60  	defer db.Close()
    61  	// database/sql might not call our Open at all unless we do something with
    62  	// the connection
    63  	txn, err := db.(driver.ConnBeginTx).BeginTx(context.Background(), driver.TxOptions{})
    64  	if err != nil {
    65  		t.Fatal(err)
    66  	}
    67  	txn.Rollback()
    68  }
    69
View as plain text