Files
tasks/tools/sql/clitest/conn_tests.go
Alex ShtinandGitHub 8e8f1b0f7c Remove license header from every file (#7689)
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.

## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
2025-05-01 18:50:21 -07:00

97 lines
2.1 KiB
Go

package clitest
import (
"net"
"github.com/stretchr/testify/require"
"go.temporal.io/server/common/config"
"go.temporal.io/server/common/log"
"go.temporal.io/server/common/log/tag"
"go.temporal.io/server/tools/common/schema/test"
"go.temporal.io/server/tools/sql"
)
type (
// SQLConnTestSuite defines a test suite
SQLConnTestSuite struct {
test.DBTestBase
host string
port string
pluginName string
sqlQuery string
}
)
var _ test.DB = (*sql.Connection)(nil)
const (
testUser = "temporal"
testPassword = "temporal"
)
// NewSQLConnTestSuite returns the test suite
func NewSQLConnTestSuite(
host string,
port string,
pluginName string,
sqlQuery string,
) *SQLConnTestSuite {
return &SQLConnTestSuite{
host: host,
port: port,
pluginName: pluginName,
sqlQuery: sqlQuery,
}
}
// SetupTest setups test
func (s *SQLConnTestSuite) SetupTest() {
s.Assertions = require.New(s.T()) // Have to define our overridden assertions in the test setup. If we did it earlier, s.T() will return nil
}
// SetupSuite setups test suite
func (s *SQLConnTestSuite) SetupSuite() {
conn, err := newTestConn("", s.host, s.port, s.pluginName)
if err != nil {
logger := log.NewTestLogger()
logger.Fatal("error creating sql conn", tag.Error(err))
}
s.SetupSuiteBase(conn)
}
// TearDownSuite tear down test suite
func (s *SQLConnTestSuite) TearDownSuite() {
s.TearDownSuiteBase()
}
// TestParseCQLFile test
func (s *SQLConnTestSuite) TestParseCQLFile() {
s.RunParseFileTest(s.sqlQuery)
}
// TestSQLConn test
func (s *SQLConnTestSuite) TestSQLConn() {
conn, err := newTestConn(s.DBName, s.host, s.port, s.pluginName)
s.Nil(err)
s.RunCreateTest(conn)
s.RunUpdateTest(conn)
s.RunDropTest(conn)
conn.Close()
}
func newTestConn(
database string,
host string,
port string,
pluginName string,
) (*sql.Connection, error) {
connectAddr := net.JoinHostPort(host, port)
return sql.NewConnection(&config.SQL{
ConnectAddr: connectAddr,
User: testUser,
Password: testPassword,
PluginName: pluginName,
DatabaseName: database,
}, log.NewTestLogger())
}