// 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 tests

import (
	"testing"

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

func TestRows(t *testing.T) {
	assert.NoError(t, PrepareEngine())

	type UserRows struct {
		Id    int64
		IsMan bool
	}

	assert.NoError(t, testEngine.Sync(new(UserRows)))

	cnt, err := testEngine.Insert(&UserRows{
		IsMan: true,
	})
	assert.NoError(t, err)
	assert.EqualValues(t, 1, cnt)

	rows, err := testEngine.Rows(new(UserRows))
	assert.NoError(t, err)
	defer rows.Close()

	cnt = 0
	user := new(UserRows)
	for rows.Next() {
		err = rows.Scan(user)
		assert.NoError(t, err)
		cnt++
	}
	assert.EqualValues(t, 1, cnt)
	assert.False(t, rows.Next())
	assert.NoError(t, rows.Close())

	rows0, err := testEngine.Where("1>1").Rows(new(UserRows))
	assert.NoError(t, err)
	defer rows0.Close()

	cnt = 0
	user0 := new(UserRows)
	for rows0.Next() {
		err = rows0.Scan(user0)
		assert.NoError(t, err)
		cnt++
	}
	assert.EqualValues(t, 0, cnt)
	assert.NoError(t, rows0.Close())

	sess := testEngine.NewSession()
	defer sess.Close()

	rows1, err := sess.Prepare().Rows(new(UserRows))
	assert.NoError(t, err)
	defer rows1.Close()

	cnt = 0
	for rows1.Next() {
		err = rows1.Scan(user)
		assert.NoError(t, err)
		cnt++
	}
	assert.EqualValues(t, 1, cnt)

	tbName := testEngine.Quote(testEngine.TableName(user, true))
	rows2, err := testEngine.SQL("SELECT * FROM " + tbName).Rows(new(UserRows))
	assert.NoError(t, err)
	defer rows2.Close()

	cnt = 0
	for rows2.Next() {
		err = rows2.Scan(user)
		assert.NoError(t, err)
		cnt++
	}
	assert.EqualValues(t, 1, cnt)
}

func TestRowsMyTableName(t *testing.T) {
	assert.NoError(t, PrepareEngine())

	type UserRowsMyTable struct {
		Id    int64
		IsMan bool
	}

	tableName := "user_rows_my_table_name"

	assert.NoError(t, testEngine.Table(tableName).Sync(new(UserRowsMyTable)))

	cnt, err := testEngine.Table(tableName).Insert(&UserRowsMyTable{
		IsMan: true,
	})
	assert.NoError(t, err)
	assert.EqualValues(t, 1, cnt)

	rows, err := testEngine.Table(tableName).Rows(new(UserRowsMyTable))
	assert.NoError(t, err)

	cnt = 0
	user := new(UserRowsMyTable)
	for rows.Next() {
		err = rows.Scan(user)
		assert.NoError(t, err)
		cnt++
	}
	assert.EqualValues(t, 1, cnt)

	rows.Close()

	rows, err = testEngine.Table(tableName).Rows(&UserRowsMyTable{
		Id: 2,
	})
	assert.NoError(t, err)
	cnt = 0
	user = new(UserRowsMyTable)
	for rows.Next() {
		err = rows.Scan(user)
		assert.NoError(t, err)
		cnt++
	}
	assert.EqualValues(t, 0, cnt)
}

type UserRowsSpecTable struct {
	Id    int64
	IsMan bool
}

func (UserRowsSpecTable) TableName() string {
	return "user_rows_my_table_name"
}

func TestRowsSpecTableName(t *testing.T) {
	assert.NoError(t, PrepareEngine())
	assert.NoError(t, testEngine.Sync(new(UserRowsSpecTable)))

	cnt, err := testEngine.Insert(&UserRowsSpecTable{
		IsMan: true,
	})
	assert.NoError(t, err)
	assert.EqualValues(t, 1, cnt)

	rows, err := testEngine.Rows(new(UserRowsSpecTable))
	assert.NoError(t, err)
	defer rows.Close()

	cnt = 0
	user := new(UserRowsSpecTable)
	for rows.Next() {
		err = rows.Scan(user)
		assert.NoError(t, err)
		cnt++
	}
	assert.NoError(t, rows.Err())
	assert.EqualValues(t, 1, cnt)
}

func TestRowsScanVars(t *testing.T) {
	type RowsScanVars struct {
		Id   int64
		Name string
		Age  int
	}

	assert.NoError(t, PrepareEngine())
	assert.NoError(t, testEngine.Sync2(new(RowsScanVars)))

	cnt, err := testEngine.Insert(&RowsScanVars{
		Name: "xlw",
		Age:  42,
	}, &RowsScanVars{
		Name: "xlw2",
		Age:  24,
	})
	assert.NoError(t, err)
	assert.EqualValues(t, 2, cnt)

	rows, err := testEngine.Cols("name", "age").Rows(new(RowsScanVars))
	assert.NoError(t, err)
	defer rows.Close()

	cnt = 0
	for rows.Next() {
		var name string
		var age int
		err = rows.Scan(&name, &age)
		assert.NoError(t, err)
		if cnt == 0 {
			assert.EqualValues(t, "xlw", name)
			assert.EqualValues(t, 42, age)
		} else if cnt == 1 {
			assert.EqualValues(t, "xlw2", name)
			assert.EqualValues(t, 24, age)
		}
		cnt++
	}
	assert.NoError(t, rows.Err())
	assert.EqualValues(t, 2, cnt)
}

func TestRowsScanBytes(t *testing.T) {
	type RowsScanBytes struct {
		Id     int64
		Bytes1 []byte
		Bytes2 []byte
	}

	assert.NoError(t, PrepareEngine())
	assert.NoError(t, testEngine.Sync(new(RowsScanBytes)))

	cnt, err := testEngine.Insert(&RowsScanBytes{
		Bytes1: []byte("bytes1"),
		Bytes2: []byte("bytes2"),
	}, &RowsScanBytes{
		Bytes1: []byte("bytes1-1"),
		Bytes2: []byte("bytes2-2"),
	})
	assert.NoError(t, err)
	assert.EqualValues(t, 2, cnt)

	{
		rows, err := testEngine.Cols("bytes1, bytes2").Rows(new(RowsScanBytes))
		assert.NoError(t, err)
		defer rows.Close()

		cnt = 0
		var bytes1 []byte
		var bytes2 []byte
		for rows.Next() {
			err = rows.Scan(&bytes1, &bytes2)
			assert.NoError(t, err)
			if cnt == 0 {
				assert.EqualValues(t, []byte("bytes1"), bytes1)
				assert.EqualValues(t, []byte("bytes2"), bytes2)
			} else if cnt == 1 {
				// bytes1 now should be `bytes1` but will be override
				assert.EqualValues(t, []byte("bytes1-1"), bytes1)
				assert.EqualValues(t, []byte("bytes2-2"), bytes2)
			}
			cnt++
		}
		assert.NoError(t, rows.Err())
		assert.EqualValues(t, 2, cnt)
		rows.Close()
	}

	{
		rows, err := testEngine.Cols("bytes1, bytes2").Rows(new(RowsScanBytes))
		assert.NoError(t, err)
		defer rows.Close()

		cnt = 0
		var rsb RowsScanBytes
		for rows.Next() {
			err = rows.Scan(&rsb)
			assert.NoError(t, err)
			if cnt == 0 {
				assert.EqualValues(t, []byte("bytes1"), rsb.Bytes1)
				assert.EqualValues(t, []byte("bytes2"), rsb.Bytes2)
			} else if cnt == 1 {
				// bytes1 now should be `bytes1` but will be override
				assert.EqualValues(t, []byte("bytes1-1"), rsb.Bytes1)
				assert.EqualValues(t, []byte("bytes2-2"), rsb.Bytes2)
			}
			cnt++
		}
		assert.NoError(t, rows.Err())
		assert.EqualValues(t, 2, cnt)
		rows.Close()
	}
}