mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
143 lines
4.6 KiB
Go
143 lines
4.6 KiB
Go
package persistencetests
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/hanzoai/tasks/common/config"
|
|
"github.com/hanzoai/tasks/common/persistence/sql/sqlplugin/mysql"
|
|
"github.com/hanzoai/tasks/common/persistence/sql/sqlplugin/postgresql"
|
|
"github.com/hanzoai/tasks/common/persistence/sql/sqlplugin/sqlite"
|
|
"github.com/hanzoai/tasks/temporal/environment"
|
|
)
|
|
|
|
const (
|
|
testCassandraSchemaDir = "schema/cassandra/"
|
|
|
|
testMySQLUser = "temporal"
|
|
testMySQLPassword = "temporal"
|
|
testMySQLSchemaDir = "schema/mysql/v8"
|
|
|
|
testPostgreSQLUser = "temporal"
|
|
testPostgreSQLPassword = "temporal"
|
|
testPostgreSQLSchemaDir = "schema/postgresql/v12"
|
|
|
|
testSQLiteUser = ""
|
|
testSQLitePassword = ""
|
|
testSQLiteMode = "memory"
|
|
testSQLiteCache = "private"
|
|
testSQLiteSchemaDir = "schema/sqlite/v3" // specify if mode is not "memory"
|
|
)
|
|
|
|
// GetTestClusterOption returns test options for the given store type and driver.
|
|
func GetTestClusterOption(storeType, driver string) *TestBaseOptions {
|
|
switch storeType {
|
|
case config.StoreTypeSQL:
|
|
switch driver {
|
|
case mysql.PluginName:
|
|
return GetMySQLTestClusterOption()
|
|
case postgresql.PluginName:
|
|
return GetPostgreSQLTestClusterOption()
|
|
case postgresql.PluginNamePGX:
|
|
return GetPostgreSQLPGXTestClusterOption()
|
|
case sqlite.PluginName:
|
|
return GetSQLiteMemoryTestClusterOption()
|
|
default:
|
|
panic(fmt.Sprintf("unknown sql driver: %v", driver))
|
|
}
|
|
case config.StoreTypeNoSQL:
|
|
return GetCassandraTestClusterOption()
|
|
default:
|
|
panic(fmt.Sprintf("unknown store type: %v", storeType))
|
|
}
|
|
}
|
|
|
|
// GetCassandraTestClusterOption returns test options
|
|
func GetCassandraTestClusterOption() *TestBaseOptions {
|
|
return &TestBaseOptions{
|
|
DBName: "test_" + GenerateRandomDBName(3),
|
|
DBHost: environment.GetCassandraAddress(),
|
|
DBPort: environment.GetCassandraPort(),
|
|
SchemaDir: testCassandraSchemaDir,
|
|
StoreType: config.StoreTypeNoSQL,
|
|
}
|
|
}
|
|
|
|
// GetMySQLTestClusterOption return test options
|
|
func GetMySQLTestClusterOption() *TestBaseOptions {
|
|
return &TestBaseOptions{
|
|
SQLDBPluginName: mysql.PluginName,
|
|
DBName: "test_" + GenerateRandomDBName(3),
|
|
DBUsername: testMySQLUser,
|
|
DBPassword: testMySQLPassword,
|
|
DBHost: environment.GetMySQLAddress(),
|
|
DBPort: environment.GetMySQLPort(),
|
|
SchemaDir: testMySQLSchemaDir,
|
|
StoreType: config.StoreTypeSQL,
|
|
}
|
|
}
|
|
|
|
// GetPostgreSQLTestClusterOption return test options
|
|
func GetPostgreSQLTestClusterOption() *TestBaseOptions {
|
|
return &TestBaseOptions{
|
|
SQLDBPluginName: postgresql.PluginName,
|
|
DBName: "test_" + GenerateRandomDBName(3),
|
|
DBUsername: testPostgreSQLUser,
|
|
DBPassword: testPostgreSQLPassword,
|
|
DBHost: environment.GetPostgreSQLAddress(),
|
|
DBPort: environment.GetPostgreSQLPort(),
|
|
SchemaDir: testPostgreSQLSchemaDir,
|
|
StoreType: config.StoreTypeSQL,
|
|
}
|
|
}
|
|
|
|
// GetPostgreSQLPGXTestClusterOption return test options
|
|
func GetPostgreSQLPGXTestClusterOption() *TestBaseOptions {
|
|
return &TestBaseOptions{
|
|
SQLDBPluginName: postgresql.PluginNamePGX,
|
|
DBName: "test_" + GenerateRandomDBName(3),
|
|
DBUsername: testPostgreSQLUser,
|
|
DBPassword: testPostgreSQLPassword,
|
|
DBHost: environment.GetPostgreSQLAddress(),
|
|
DBPort: environment.GetPostgreSQLPort(),
|
|
SchemaDir: testPostgreSQLSchemaDir,
|
|
StoreType: config.StoreTypeSQL,
|
|
}
|
|
}
|
|
|
|
// GetSQLiteFileTestClusterOption return test options
|
|
func GetSQLiteFileTestClusterOption() *TestBaseOptions {
|
|
return &TestBaseOptions{
|
|
SQLDBPluginName: sqlite.PluginName,
|
|
DBName: filepath.Join(os.TempDir(), "test_"+GenerateRandomDBName(3)), // put files in temp to avoid cluttering the project
|
|
DBUsername: testSQLiteUser,
|
|
DBPassword: testSQLitePassword,
|
|
DBHost: environment.GetLocalhostIP(),
|
|
DBPort: 0,
|
|
SchemaDir: testSQLiteSchemaDir,
|
|
StoreType: config.StoreTypeSQL,
|
|
ConnectAttributes: map[string]string{
|
|
"cache": "shared",
|
|
"busy_timeout": "30000",
|
|
"journal_mode": "wal",
|
|
"synchronous": "normal",
|
|
},
|
|
}
|
|
}
|
|
|
|
// GetSQLiteMemoryTestClusterOption return test options
|
|
func GetSQLiteMemoryTestClusterOption() *TestBaseOptions {
|
|
return &TestBaseOptions{
|
|
SQLDBPluginName: sqlite.PluginName,
|
|
DBName: "test_" + GenerateRandomDBName(3),
|
|
DBUsername: testSQLiteUser,
|
|
DBPassword: testSQLitePassword,
|
|
DBHost: environment.GetLocalhostIP(),
|
|
DBPort: 0,
|
|
SchemaDir: "",
|
|
StoreType: config.StoreTypeSQL,
|
|
ConnectAttributes: map[string]string{"mode": testSQLiteMode, "cache": testSQLiteCache},
|
|
}
|
|
}
|