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))
		}
	}
}