package csv

import (
	"bytes"
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/mayswind/ezbookkeeping/pkg/core"
)

func TestCsvFileBasicDataTableDataRowCount(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{
		{"A1", "B1", "C1"},
		{"A2", "B2", "C2"},
		{"A3", "B3", "C3"},
	}, false)

	assert.Equal(t, 3, datatable.DataRowCount())
}

func TestCsvFileBasicDataTableDataRowCount_HasTitleLine(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{
		{"A1", "B1", "C1"},
		{"A2", "B2", "C2"},
		{"A3", "B3", "C3"},
	}, true)

	assert.Equal(t, 2, datatable.DataRowCount())
}

func TestCsvFileBasicDataTableDataRowCount_OnlyHeaderLine(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{
		{"A1", "B1", "C1"},
	}, true)

	assert.Equal(t, 0, datatable.DataRowCount())
}

func TestCsvFileBasicDataTableDataRowCount_EmptyContent(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{}, false)
	assert.Equal(t, 0, datatable.DataRowCount())

	datatable = CreateNewCustomCsvBasicDataTable([][]string{}, true)
	assert.Equal(t, 0, datatable.DataRowCount())
}

func TestCsvFileBasicDataTableHeaderColumnNames(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{
		{"A1", "B1", "C1"},
		{"A2", "B2", "C2"},
		{"A3", "B3", "C3"},
	}, true)

	assert.EqualValues(t, []string{"A1", "B1", "C1"}, datatable.HeaderColumnNames())
}

func TestCsvFileBasicDataTableHeaderColumnNames_EmptyContent(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{}, false)
	assert.Nil(t, datatable.HeaderColumnNames())

	datatable = CreateNewCustomCsvBasicDataTable([][]string{}, true)
	assert.Nil(t, datatable.HeaderColumnNames())
}

func TestCsvFileBasicDataTableRowIterator(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{
		{"A1", "B1", "C1"},
		{"A2", "B2", "C2"},
		{"A3", "B3", "C3"},
	}, false)

	iterator := datatable.DataRowIterator()
	assert.True(t, iterator.HasNext())

	// data row 1
	assert.NotNil(t, iterator.Next())
	assert.True(t, iterator.HasNext())

	// data row 2
	assert.NotNil(t, iterator.Next())
	assert.True(t, iterator.HasNext())

	// data row 3
	assert.NotNil(t, iterator.Next())
	assert.False(t, iterator.HasNext())

	// not existed data row 4
	assert.Nil(t, iterator.Next())
	assert.False(t, iterator.HasNext())
}

func TestCsvFileBasicDataTableRowIterator_HasTitleLine(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{
		{"A1", "B1", "C1"},
		{"A2", "B2", "C2"},
		{"A3", "B3", "C3"},
	}, true)

	iterator := datatable.DataRowIterator()
	assert.True(t, iterator.HasNext())

	// data row 1
	assert.NotNil(t, iterator.Next())
	assert.True(t, iterator.HasNext())

	// data row 2
	assert.NotNil(t, iterator.Next())
	assert.False(t, iterator.HasNext())

	// not existed data row 3
	assert.Nil(t, iterator.Next())
	assert.False(t, iterator.HasNext())

	// not existed data row 4
	assert.Nil(t, iterator.Next())
	assert.False(t, iterator.HasNext())
}

func TestCsvFileBasicDataTableRowColumnCount(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{
		{"A1", "B1", "C1"},
		{"A2", "B2", "C2"},
		{"A3", "B3", "C3"},
	}, true)

	iterator := datatable.DataRowIterator()

	row1 := iterator.Next()
	assert.EqualValues(t, 3, row1.ColumnCount())

	row2 := iterator.Next()
	assert.EqualValues(t, 3, row2.ColumnCount())
}

func TestCsvFileBasicDataTableRowGetData(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{
		{"A1", "B1", "C1"},
		{"A2", "B2", "C2"},
		{"A3", "B3", "C3"},
	}, false)

	iterator := datatable.DataRowIterator()

	row1 := iterator.Next()
	assert.Equal(t, "A1", row1.GetData(0))
	assert.Equal(t, "B1", row1.GetData(1))
	assert.Equal(t, "C1", row1.GetData(2))

	row2 := iterator.Next()
	assert.Equal(t, "A2", row2.GetData(0))
	assert.Equal(t, "B2", row2.GetData(1))
	assert.Equal(t, "C2", row2.GetData(2))

	row3 := iterator.Next()
	assert.Equal(t, "A3", row3.GetData(0))
	assert.Equal(t, "B3", row3.GetData(1))
	assert.Equal(t, "C3", row3.GetData(2))
}

func TestCsvFileBasicDataTableRowGetData_HasTitleLine(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{
		{"A1", "B1", "C1"},
		{"A2", "B2", "C2"},
		{"A3", "B3", "C3"},
	}, true)

	iterator := datatable.DataRowIterator()

	row1 := iterator.Next()
	assert.Equal(t, "A2", row1.GetData(0))
	assert.Equal(t, "B2", row1.GetData(1))
	assert.Equal(t, "C2", row1.GetData(2))

	row2 := iterator.Next()
	assert.Equal(t, "A3", row2.GetData(0))
	assert.Equal(t, "B3", row2.GetData(1))
	assert.Equal(t, "C3", row2.GetData(2))
}

func TestCsvFileBasicDataTableRowGetData_GetNotExistedColumnData(t *testing.T) {
	datatable := CreateNewCustomCsvBasicDataTable([][]string{
		{"A1", "B1", "C1"},
		{"A2", "B2", "C2"},
		{"A3", "B3", "C3"},
	}, true)

	iterator := datatable.DataRowIterator()

	row1 := iterator.Next()
	assert.Equal(t, "", row1.GetData(3))
}

func TestCreateNewCsvBasicDataTable(t *testing.T) {
	context := core.NewNullContext()
	reader := bytes.NewReader([]byte("A1,B1,C1\n" +
		"A2,B2,C2\n" +
		"A3,B3,C3\n"))
	datatable, err := CreateNewCsvBasicDataTable(context, reader, true)
	assert.Nil(t, err)

	assert.Equal(t, 2, datatable.DataRowCount())

	iterator := datatable.DataRowIterator()
	assert.True(t, iterator.HasNext())

	row1 := iterator.Next()
	assert.EqualValues(t, 3, row1.ColumnCount())
	assert.Equal(t, "A2", row1.GetData(0))
	assert.Equal(t, "B2", row1.GetData(1))
	assert.Equal(t, "C2", row1.GetData(2))
	assert.True(t, iterator.HasNext())

	row2 := iterator.Next()
	assert.EqualValues(t, 3, row2.ColumnCount())
	assert.Equal(t, "A3", row2.GetData(0))
	assert.Equal(t, "B3", row2.GetData(1))
	assert.Equal(t, "C3", row2.GetData(2))
	assert.False(t, iterator.HasNext())
}

func TestCreateNewCsvBasicDataTable_SkipBlankLine(t *testing.T) {
	context := core.NewNullContext()
	reader := bytes.NewReader([]byte("\n" +
		"A1,B1,C1\n" +
		"A2,B2,C2\n" +
		"\n" +
		"A3,B3,C3\n"))
	datatable, err := CreateNewCsvBasicDataTable(context, reader, true)
	assert.Nil(t, err)

	assert.Equal(t, 2, datatable.DataRowCount())

	iterator := datatable.DataRowIterator()
	assert.True(t, iterator.HasNext())

	row1 := iterator.Next()
	assert.EqualValues(t, 3, row1.ColumnCount())
	assert.Equal(t, "A2", row1.GetData(0))
	assert.Equal(t, "B2", row1.GetData(1))
	assert.Equal(t, "C2", row1.GetData(2))
	assert.True(t, iterator.HasNext())

	row2 := iterator.Next()
	assert.EqualValues(t, 3, row2.ColumnCount())
	assert.Equal(t, "A3", row2.GetData(0))
	assert.Equal(t, "B3", row2.GetData(1))
	assert.Equal(t, "C3", row2.GetData(2))
	assert.False(t, iterator.HasNext())
}