Files
tasks/common/persistence/tests/mysql_test.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

616 lines
17 KiB
Go

package tests
import (
"sync/atomic"
"testing"
"github.com/stretchr/testify/suite"
"go.temporal.io/server/common/log"
"go.temporal.io/server/common/metrics"
"go.temporal.io/server/common/persistence"
persistencetests "go.temporal.io/server/common/persistence/persistence-tests"
"go.temporal.io/server/common/persistence/serialization"
"go.temporal.io/server/common/persistence/sql"
"go.temporal.io/server/common/persistence/sql/sqlplugin"
_ "go.temporal.io/server/common/persistence/sql/sqlplugin/mysql"
sqltests "go.temporal.io/server/common/persistence/sql/sqlplugin/tests"
"go.temporal.io/server/common/resolver"
)
func TestMySQLShardStoreSuite(t *testing.T) {
testData, tearDown := setUpMySQLTest(t)
defer tearDown()
shardStore, err := testData.Factory.NewShardStore()
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
s := NewShardSuite(
t,
shardStore,
serialization.NewSerializer(),
testData.Logger,
)
suite.Run(t, s)
}
func TestMySQLExecutionMutableStateStoreSuite(t *testing.T) {
testData, tearDown := setUpMySQLTest(t)
defer tearDown()
shardStore, err := testData.Factory.NewShardStore()
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
executionStore, err := testData.Factory.NewExecutionStore()
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
s := NewExecutionMutableStateSuite(
t,
shardStore,
executionStore,
serialization.NewSerializer(),
&persistence.HistoryBranchUtilImpl{},
testData.Logger,
)
suite.Run(t, s)
}
func TestMySQLExecutionMutableStateTaskStoreSuite(t *testing.T) {
testData, tearDown := setUpMySQLTest(t)
defer tearDown()
shardStore, err := testData.Factory.NewShardStore()
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
executionStore, err := testData.Factory.NewExecutionStore()
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
s := NewExecutionMutableStateTaskSuite(
t,
shardStore,
executionStore,
serialization.NewSerializer(),
testData.Logger,
)
suite.Run(t, s)
}
func TestMySQLHistoryStoreSuite(t *testing.T) {
testData, tearDown := setUpMySQLTest(t)
defer tearDown()
store, err := testData.Factory.NewExecutionStore()
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
s := NewHistoryEventsSuite(t, store, testData.Logger)
suite.Run(t, s)
}
func TestMySQLTaskQueueSuite(t *testing.T) {
testData, tearDown := setUpMySQLTest(t)
defer tearDown()
taskQueueStore, err := testData.Factory.NewTaskStore()
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
testData.Factory.Close()
TearDownMySQLDatabase(t, testData.Cfg)
}()
s := NewTaskQueueSuite(t, taskQueueStore, testData.Logger)
suite.Run(t, s)
}
func TestMySQLTaskQueueTaskSuite(t *testing.T) {
testData, tearDown := setUpMySQLTest(t)
defer tearDown()
taskQueueStore, err := testData.Factory.NewTaskStore()
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
s := NewTaskQueueTaskSuite(t, taskQueueStore, testData.Logger)
suite.Run(t, s)
}
func TestMySQLTaskQueueUserDataSuite(t *testing.T) {
testData, tearDown := setUpMySQLTest(t)
defer tearDown()
taskQueueStore, err := testData.Factory.NewTaskStore()
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
s := NewTaskQueueUserDataSuite(t, taskQueueStore, testData.Logger)
suite.Run(t, s)
}
func TestMySQLVisibilityPersistenceSuite(t *testing.T) {
s := &VisibilityPersistenceSuite{
TestBase: persistencetests.NewTestBaseWithSQL(persistencetests.GetMySQLTestClusterOption()),
}
suite.Run(t, s)
}
// TODO: Merge persistence-tests into the tests directory.
func TestMySQLHistoryV2PersistenceSuite(t *testing.T) {
s := new(persistencetests.HistoryV2PersistenceSuite)
s.TestBase = persistencetests.NewTestBaseWithSQL(persistencetests.GetMySQLTestClusterOption())
s.TestBase.Setup(nil)
suite.Run(t, s)
}
func TestMySQLMetadataPersistenceSuiteV2(t *testing.T) {
s := new(persistencetests.MetadataPersistenceSuiteV2)
s.TestBase = persistencetests.NewTestBaseWithSQL(persistencetests.GetMySQLTestClusterOption())
s.TestBase.Setup(nil)
suite.Run(t, s)
}
func TestMySQLQueuePersistence(t *testing.T) {
s := new(persistencetests.QueuePersistenceSuite)
s.TestBase = persistencetests.NewTestBaseWithSQL(persistencetests.GetMySQLTestClusterOption())
s.TestBase.Setup(nil)
suite.Run(t, s)
}
func TestMySQLClusterMetadataPersistence(t *testing.T) {
s := new(persistencetests.ClusterMetadataManagerSuite)
s.TestBase = persistencetests.NewTestBaseWithSQL(persistencetests.GetMySQLTestClusterOption())
s.TestBase.Setup(nil)
suite.Run(t, s)
}
// SQL Store tests
func TestMySQLNamespaceSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewNamespaceSuite(t, store)
suite.Run(t, s)
}
func TestMySQLQueueMessageSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewQueueMessageSuite(t, store)
suite.Run(t, s)
}
func TestMySQLQueueMetadataSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewQueueMetadataSuite(t, store)
suite.Run(t, s)
}
func TestMySQLMatchingTaskSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewMatchingTaskSuite(t, store)
suite.Run(t, s)
}
func TestMySQLMatchingTaskQueueSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewMatchingTaskQueueSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryShardSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryShardSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryNodeSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryNodeSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryTreeSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryTreeSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryCurrentExecutionSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryCurrentExecutionSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryExecutionSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryExecutionSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryTransferTaskSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryTransferTaskSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryTimerTaskSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryTimerTaskSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryReplicationTaskSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryReplicationTaskSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryVisibilityTaskSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryVisibilityTaskSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryReplicationDLQTaskSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryReplicationDLQTaskSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryExecutionBufferSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryExecutionBufferSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryExecutionActivitySuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryExecutionActivitySuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryExecutionChildWorkflowSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryExecutionChildWorkflowSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryExecutionTimerSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryExecutionTimerSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryExecutionChasmSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryExecutionChasmSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryExecutionRequestCancelSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryExecutionRequestCancelSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryExecutionSignalSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryExecutionSignalSuite(t, store)
suite.Run(t, s)
}
func TestMySQLHistoryExecutionSignalRequestSuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindMain, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewHistoryExecutionSignalRequestSuite(t, store)
suite.Run(t, s)
}
func TestMySQLVisibilitySuite(t *testing.T) {
cfg := NewMySQLConfig()
SetupMySQLDatabase(t, cfg)
SetupMySQLSchema(t, cfg)
store, err := sql.NewSQLDB(sqlplugin.DbKindVisibility, cfg, resolver.NewNoopResolver(), log.NewTestLogger(), metrics.NoopMetricsHandler)
if err != nil {
t.Fatalf("unable to create MySQL DB: %v", err)
}
defer func() {
_ = store.Close()
TearDownMySQLDatabase(t, cfg)
}()
s := sqltests.NewVisibilitySuite(t, store)
suite.Run(t, s)
}
func TestMySQLClosedConnectionError(t *testing.T) {
testData, tearDown := setUpMySQLTest(t)
defer tearDown()
s := newConnectionSuite(t, testData.Factory)
suite.Run(t, s)
}
func TestMySQLQueueV2(t *testing.T) {
testData, tearDown := setUpMySQLTest(t)
t.Cleanup(tearDown)
RunQueueV2TestSuiteForSQL(t, testData.Factory)
}
func TestMySQLNexusEndpointPersistence(t *testing.T) {
testData, tearDown := setUpMySQLTest(t)
defer tearDown()
store, err := testData.Factory.NewNexusEndpointStore()
if err != nil {
t.Fatalf("unable to create MySQL NexusEndpointStore: %v", err)
}
tableVersion := atomic.Int64{}
t.Run("Generic", func(t *testing.T) {
RunNexusEndpointTestSuite(t, store, &tableVersion)
})
}