github.com/snowflakedb/gosnowflake@v1.9.0/prepared_statement_test.go (about)

     1  // Copyright (c) 2021-2022 Snowflake Computing Inc. All rights reserved.
     2  
     3  package gosnowflake
     4  
     5  import (
     6  	"testing"
     7  )
     8  
     9  // TestPreparedStatement creates a basic prepared statement, inserting values
    10  // after the statement has been prepared
    11  func TestPreparedStatement(t *testing.T) {
    12  	runDBTest(t, func(dbt *DBTest) {
    13  		dbt.mustExec("create or replace table test_prep_statement(c1 INTEGER, c2 FLOAT, c3 BOOLEAN, c4 STRING)")
    14  		defer dbt.mustExec(deleteTableSQL)
    15  
    16  		intArray := []int{1, 2, 3}
    17  		fltArray := []float64{0.1, 2.34, 5.678}
    18  		boolArray := []bool{true, false, true}
    19  		strArray := []string{"test1", "test2", "test3"}
    20  		stmt := dbt.mustPrepare("insert into TEST_PREP_STATEMENT values(?, ?, ?, ?)")
    21  		if _, err := stmt.Exec(Array(&intArray), Array(&fltArray), Array(&boolArray), Array(&strArray)); err != nil {
    22  			t.Fatal(err)
    23  		}
    24  		rows := dbt.mustQuery(selectAllSQL)
    25  		defer rows.Close()
    26  
    27  		var v1 int
    28  		var v2 float64
    29  		var v3 bool
    30  		var v4 string
    31  		if rows.Next() {
    32  			err := rows.Scan(&v1, &v2, &v3, &v4)
    33  			if err != nil {
    34  				t.Fatal(err)
    35  			}
    36  			if v1 != 1 && v2 != 0.1 && v3 != true && v4 != "test1" {
    37  				t.Fatalf("failed to fetch. expected: 1, 0.1, true, test1. got: %v, %v, %v, %v", v1, v2, v3, v4)
    38  			}
    39  		} else {
    40  			t.Error("failed to query")
    41  		}
    42  
    43  		if rows.Next() {
    44  			err := rows.Scan(&v1, &v2, &v3, &v4)
    45  			if err != nil {
    46  				t.Fatal(err)
    47  			}
    48  			if v1 != 2 && v2 != 2.34 && v3 != false && v4 != "test2" {
    49  				t.Fatalf("failed to fetch. expected: 2, 2.34, false, test2. got: %v, %v, %v, %v", v1, v2, v3, v4)
    50  			}
    51  		} else {
    52  			t.Error("failed to query")
    53  		}
    54  
    55  		if rows.Next() {
    56  			err := rows.Scan(&v1, &v2, &v3, &v4)
    57  			if err != nil {
    58  				t.Fatal(err)
    59  			}
    60  			if v1 != 3 && v2 != 5.678 && v3 != true && v4 != "test3" {
    61  				t.Fatalf("failed to fetch. expected: 3, test3. got: %v, %v, %v, %v", v1, v2, v3, v4)
    62  			}
    63  		} else {
    64  			t.Error("failed to query")
    65  		}
    66  	})
    67  }