cloudsql_test.go

// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package cloudsql

import (
"bytes"
"net/http/httptest"
"os"
"strings"
"testing"
)

// dbConfig holds database connection information derived from the environment.
typedbConfigstruct {
// user is the user name
userstring
// pass is the user password
passstring
// name is the database name
namestring
// port is the port used with the TCP host
portstring
// host is a TCP address
hoststring
// unixPath is the unix socket file path
unixPathstring
// instConnName is the Cloud SQL instance connection name
instConnNamestring
}

typeconnTypeint

const (
useTCPconnType=iota
useUnix
useConnector
)

// dbConfigFromEnv reads database configuration from the environment, selecting
// the provided connType in the configuration.
funcdbConfigFromEnv(t*testing.T, ctconnType) dbConfig {
testEnv:=func(namestring) string {
n:=os.Getenv(name)
ifn=="" {
t.Fatalf("failed to get env var %q", name)
 }
returnn
 }
d:=dbConfig{
user: testEnv("MYSQL_USER"),
pass: testEnv("MYSQL_PASSWORD"),
name: testEnv("MYSQL_DATABASE"),
port: testEnv("MYSQL_PORT"),
host: testEnv("MYSQL_HOST"),
unixPath: testEnv("MYSQL_UNIX_SOCKET"),
instConnName: testEnv("MYSQL_INSTANCE"),
 }
// Zero out all but requested conn type
switchct {
caseuseTCP:
// use host
d.unixPath=""
d.instConnName=""
caseuseUnix:
d.host=""
// use unix path
d.instConnName=""
default: // connector
d.host=""
d.unixPath=""
// use instance connection name
 }
returnd
}

funcsetupTestEnv(cdbConfig) func() {
oldDBUser:=os.Getenv("DB_USER")
oldDBPass:=os.Getenv("DB_PASS")
oldDBName:=os.Getenv("DB_NAME")
oldDBPort:=os.Getenv("DB_PORT")
oldHost:=os.Getenv("INSTANCE_HOST")
oldUnix:=os.Getenv("INSTANCE_UNIX_SOCKET")
oldInstance:=os.Getenv("INSTANCE_CONNECTION_NAME")

os.Setenv("DB_USER", c.user)
os.Setenv("DB_PASS", c.pass)
os.Setenv("DB_NAME", c.name)
os.Setenv("DB_PORT", c.port)
os.Setenv("INSTANCE_HOST", c.host)
os.Setenv("INSTANCE_UNIX_SOCKET", c.unixPath)
os.Setenv("INSTANCE_CONNECTION_NAME", c.instConnName)

returnfunc() {
os.Setenv("DB_USER", oldDBUser)
os.Setenv("DB_PASS", oldDBPass)
os.Setenv("DB_NAME", oldDBName)
os.Setenv("DB_PORT", oldDBPort)
os.Setenv("INSTANCE_HOST", oldHost)
os.Setenv("INSTANCE_UNIX", oldUnix)
os.Setenv("INSTANCE_CONNECTION_NAME", oldInstance)
 }
}

functestGetVotes(t*testing.T) {
rr:=httptest.NewRecorder()
req:=httptest.NewRequest("GET", "/", nil)
Votes(rr, req)
resp:=rr.Result()
body:=rr.Body.String()

wantStatus:=200
ifgotStatus:=resp.StatusCode; wantStatus!=gotStatus {
t.Errorf("want = %v, got = %v", wantStatus, gotStatus)
 }

want:="Tabs VS Spaces"
if!strings.Contains(body, want) {
t.Errorf("failed to find %q in resp = %v", want, body)
 }
}

functestCastVote(t*testing.T) {
rr:=httptest.NewRecorder()
req:=httptest.NewRequest("POST", "/", bytes.NewBuffer([]byte("team=SPACES")))
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
Votes(rr, req)
resp:=rr.Result()
body:=rr.Body.String()

wantStatus:=200
ifgotStatus:=resp.StatusCode; wantStatus!=gotStatus {
t.Errorf("want = %v, got = %v", wantStatus, gotStatus)
 }

want:="Vote successfully cast for SPACES"
if!strings.Contains(body, want) {
t.Errorf("failed to find %q in resp = %v", want, body)
 }
}

funcTestGetVotes(t*testing.T) {
ifos.Getenv("GOLANG_SAMPLES_E2E_TEST") =="" {
t.Skip()
 }

testCases:= []struct {
descstring
ctconnType
 }{
 {desc: "connecting with a TCP Socket", ct: useTCP},
 {desc: "connecting with a Unix Socket", ct: useUnix},
 {desc: "connecting with a connector", ct: useConnector},
 }

for_, tc:=rangetestCases {
t.Run(tc.desc, func(t*testing.T) {
conf:=dbConfigFromEnv(t, tc.ct)
cleanup:=setupTestEnv(conf)
defercleanup()

// initialize database connection based on environment
db=mustConnect()

testGetVotes(t)
 })
 }
}

funcTestCastVote(t*testing.T) {
ifos.Getenv("GOLANG_SAMPLES_E2E_TEST") =="" {
t.Skip()
 }

testCases:= []struct {
descstring
ctconnType
 }{
 {desc: "connecting with a TCP Socket", ct: useTCP},
 {desc: "connecting with a Unix Socket", ct: useUnix},
 {desc: "connecting with a connector", ct: useConnector},
 }

for_, tc:=rangetestCases {
t.Run(tc.desc, func(t*testing.T) {
conf:=dbConfigFromEnv(t, tc.ct)
cleanup:=setupTestEnv(conf)
defercleanup()

// initialize database connection based on environment
db=mustConnect()

testCastVote(t)
 })
 }
}