79 lines
2.2 KiB
Go
79 lines
2.2 KiB
Go
package app
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
_ "github.com/go-sql-driver/mysql"
|
|
_ "github.com/lib/pq"
|
|
"gitnet.fr/deblan/database-anonymizer/config"
|
|
"gitnet.fr/deblan/database-anonymizer/faker"
|
|
"testing"
|
|
)
|
|
|
|
func TestAppCreateSelectQuery(t *testing.T) {
|
|
c := config.SchemaConfigAction{Table: "foo"}
|
|
app := App{
|
|
FakeManager: faker.NewFakeManager(),
|
|
DbConfig: config.DatabaseConfig{Type: "mysql", Dsn: "mysql://foo:bar@tests"},
|
|
}
|
|
|
|
if app.CreateSelectQuery(c) != "SELECT * FROM `foo`" {
|
|
t.Fatalf("TestAppCreateSelectQuery: empty configured query check failed")
|
|
}
|
|
|
|
c = config.SchemaConfigAction{Table: "foo", Query: "query"}
|
|
|
|
if app.CreateSelectQuery(c) != "query" {
|
|
t.Fatalf("TestAppCreateSelectQuery: configured query check failed")
|
|
}
|
|
}
|
|
|
|
func TestAppDoAction(t *testing.T) {
|
|
c := config.SchemaConfigAction{Table: "foo"}
|
|
app := App{
|
|
FakeManager: faker.NewFakeManager(),
|
|
DbConfig: config.DatabaseConfig{Type: "mysql", Dsn: "mysql://foo:bar@tests"},
|
|
}
|
|
|
|
if app.CreateSelectQuery(c) != "SELECT * FROM `foo`" {
|
|
t.Fatalf("TestAppCreateSelectQuery: empty configured query check failed")
|
|
}
|
|
|
|
c = config.SchemaConfigAction{Table: "foo", Query: "query"}
|
|
|
|
if app.CreateSelectQuery(c) != "query" {
|
|
t.Fatalf("TestAppCreateSelectQuery: configured query check failed")
|
|
}
|
|
}
|
|
|
|
func TestAppRun(t *testing.T) {
|
|
schema, _ := config.LoadSchemaConfigFromFile("../tests/schema.yml")
|
|
|
|
dsns := make(map[string]string)
|
|
dsns["mysql"] = "mysql://root:root@tcp(service-mysql)/test"
|
|
dsns["postgres"] = "postgres://postgres:postgres@service-postgres:5432/test?sslmode=disable"
|
|
|
|
var count int
|
|
|
|
for dbtype, dsn := range dsns {
|
|
databaseConfig, _ := config.LoadDatabaseConfig(dsn)
|
|
db, _ := sql.Open(databaseConfig.Type, databaseConfig.Dsn)
|
|
app := App{}
|
|
app.Run(db, schema, faker.NewFakeManager(), databaseConfig)
|
|
|
|
row := db.QueryRow("SELECT COUNT(*) FROM table_truncate1")
|
|
row.Scan(&count)
|
|
|
|
if count != 0 {
|
|
t.Fatalf(fmt.Sprintf("TestAppRuny: table_truncate1 check failed (%s)", dbtype))
|
|
}
|
|
|
|
row = db.QueryRow("SELECT COUNT(*) FROM table_truncate2")
|
|
row.Scan(&count)
|
|
|
|
if count != 1 {
|
|
t.Fatalf(fmt.Sprintf("TestAppRuny: table_truncate2 check failed (%s)", dbtype))
|
|
}
|
|
}
|
|
}
|