// Copyright 2017 The Xorm Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package xorm

import (
	"context"
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestPingContext(t *testing.T) {
	assert.NoError(t, prepareEngine())

	ctx, canceled := context.WithTimeout(context.Background(), time.Nanosecond)
	defer canceled()

	time.Sleep(time.Nanosecond)

	err := testEngine.(*Engine).PingContext(ctx)
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "context deadline exceeded")
}

func TestAutoTransaction(t *testing.T) {
	assert.NoError(t, prepareEngine())

	type TestTx struct {
		Id      int64     `xorm:"autoincr pk"`
		Msg     string    `xorm:"varchar(255)"`
		Created time.Time `xorm:"created"`
	}

	assert.NoError(t, testEngine.Sync2(new(TestTx)))

	engine := testEngine.(*Engine)

	// will success
	engine.Transaction(func(session *Session) (interface{}, error) {
		_, err := session.Insert(TestTx{Msg: "hi"})
		assert.NoError(t, err)

		return nil, nil
	})

	has, err := engine.Exist(&TestTx{Msg: "hi"})
	assert.NoError(t, err)
	assert.EqualValues(t, true, has)

	// will rollback
	_, err = engine.Transaction(func(session *Session) (interface{}, error) {
		_, err := session.Insert(TestTx{Msg: "hello"})
		assert.NoError(t, err)

		return nil, fmt.Errorf("rollback")
	})
	assert.Error(t, err)

	has, err = engine.Exist(&TestTx{Msg: "hello"})
	assert.NoError(t, err)
	assert.EqualValues(t, false, has)
}

func TestDump(t *testing.T) {
	assert.NoError(t, prepareEngine())

	type TestDumpStruct struct {
		Id   int64
		Name string
	}

	assertSync(t, new(TestDumpStruct))

	testEngine.Insert([]TestDumpStruct{
		{Name: "1"},
		{Name: "2\n"},
		{Name: "3;"},
		{Name: "4"},
		{Name: "5'"},
	})

	fp := testEngine.Dialect().URI().DBName + ".sql"
	os.Remove(fp)
	assert.NoError(t, testEngine.DumpAllToFile(fp))

	assert.NoError(t, prepareEngine())

	_, err := testEngine.ImportFile(fp)
	assert.NoError(t, err)
}

func TestImport(t *testing.T) {
	assert.NoError(t, prepareEngine())

	var sql = "/*Generated by xorm 2020-03-13 17:13:51, from sqlite3 to SQLITE3*/" +
		"CREATE TABLE IF NOT EXISTS `test_dump_struct` (`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, `name` TEXT NULL);" +
		"INSERT INTO `test_dump_struct` (`id`, `name`) VALUES (1, '1');" +
		"INSERT INTO `test_dump_struct` (`id`, `name`) VALUES (2, '2;');" +
		"INSERT INTO `test_dump_struct` (`id`, `name`) VALUES (3, '3''');" +
		"INSERT INTO `test_dump_struct` (`id`, `name`) VALUES (4, '4\n\n');" +
		"INSERT INTO `test_dump_struct` (`id`, `name`) VALUES (5, '5\n;\n');"

	f, err := ioutil.TempFile(os.TempDir(), "dump_file.sql")
	assert.NoError(t, err)
	_, err = f.WriteString(sql)
	assert.NoError(t, err)
	info, err := f.Stat()
	f.Close()
	assert.NoError(t, err)
	_, err = testEngine.ImportFile(filepath.Join(os.TempDir(), info.Name()))
	assert.NoError(t, err)
}