package defs

import "github.com/featurebasedb/featurebase/v3/pql"

// join tests
var joinTestsUsers = TableTest{
	name: "jointestusers",
	Table: tbl(
		"users",
		srcHdrs(
			srcHdr("_id", fldTypeID),
			srcHdr("name", fldTypeString),
			srcHdr("age", fldTypeInt),
		),
		srcRows(
			srcRow(int64(0), string("a"), int64(21)),
			srcRow(int64(1), string("b"), int64(18)),
			srcRow(int64(2), string("c"), int64(28)),
			srcRow(int64(3), string("d"), int64(34)),
			srcRow(int64(4), string("e"), int64(36)),
		),
	),
	SQLTests: nil,
}

var joinTestsOrders = TableTest{
	name: "jointestorders",
	Table: tbl(
		"orders",
		srcHdrs(
			srcHdr("_id", fldTypeID),
			srcHdr("userid", fldTypeInt),
			srcHdr("price", fldTypeDecimal2),
		),
		srcRows(
			srcRow(int64(0), int64(1), float64(9.99)),
			srcRow(int64(1), int64(0), float64(3.99)),
			srcRow(int64(2), int64(2), float64(14.99)),
			srcRow(int64(3), int64(3), float64(5.99)),
			srcRow(int64(4), int64(1), float64(12.99)),
			srcRow(int64(5), int64(2), float64(1.99)),
		),
	),
	SQLTests: nil,
}
var joinTestsQuantity = TableTest{
	name: "jointestquantity",
	Table: tbl(
		"quantity",
		srcHdrs(
			srcHdr("_id", fldTypeID),
			srcHdr("userid", fldTypeInt),
			srcHdr("quantity", fldTypeInt),
		),
		srcRows(
			srcRow(int64(0), int64(1), int64(1)),
			srcRow(int64(1), int64(0), int64(2)),
			srcRow(int64(2), int64(2), int64(2)),
			srcRow(int64(3), int64(3), int64(1)),
			srcRow(int64(4), int64(1), int64(3)),
			srcRow(int64(5), int64(2), int64(5)),
		),
	),
	SQLTests: nil,
}

var joinTests = TableTest{
	name: "joinTests",
	SQLTests: []SQLTest{
		{
			name: "innerjoin-aggregate-groupby",
			SQLs: sqls(
				"select u._id, sum(orders.price) from orders o inner join users u on o.userid = u._id group by u._id;",
			),
			ExpHdrs: hdrs(
				hdr("_id", fldTypeID),
				hdr("", fldTypeDecimal2),
			),
			ExpRows: rows(
				row(int64(1), pql.NewDecimal(2298, 2)),
				row(int64(0), pql.NewDecimal(399, 2)),
				row(int64(2), pql.NewDecimal(1698, 2)),
				row(int64(3), pql.NewDecimal(599, 2)),
			),
			Compare: CompareExactOrdered,
		},
		{
			name: "innerjoin-aggregate-groupby-sum-filter",
			SQLs: sqls(
				"select sum(price) from orders o inner join users u on o.userid = u._id where u.age > 20;",
			),
			ExpHdrs: hdrs(
				hdr("", fldTypeDecimal2),
			),
			ExpRows: rows(
				row(pql.NewDecimal(2696, 2)),
			),
			Compare: CompareExactOrdered,
		},
		{
			name: "innerjoin-aggregate-groupby-sum-double-filter",
			SQLs: sqls(
				"select sum(price) from orders o inner join users u on o.userid = u._id where u.age > 20 and o.price < 10.00;",
			),
			ExpHdrs: hdrs(
				hdr("", fldTypeDecimal2),
			),
			ExpRows: rows(
				row(pql.NewDecimal(1197, 2)),
			),
			Compare: CompareExactOrdered,
		},
		{
			name: "innerjoin-aggregate-groupby-count-distinct-filter",
			SQLs: sqls(
				"SELECT COUNT(DISTINCT u.name) FROM orders o JOIN users u ON o.userid = u._id WHERE o.price > 9;",
			),
			ExpHdrs: hdrs(
				hdr("", fldTypeInt),
			),
			ExpRows: rows(
				row(int64(2)),
			),
			Compare: CompareExactOrdered,
		},
		{
			name: "innerjoin-aggregate-groupby-count-filter",
			SQLs: sqls(
				"SELECT COUNT(u.name) FROM orders o JOIN users u ON o.userid = u._id WHERE o.price > 9;",
			),
			ExpHdrs: hdrs(
				hdr("", fldTypeInt),
			),
			ExpRows: rows(
				row(int64(3)),
			),
			Compare: CompareExactOrdered,
		},
		{
			name: "nested-inner-join",
			SQLs: sqls(
				"select DISTINCT u._id from users u inner join orders o on o.userid = u._id join quantity q on o.userid = q.userid;",
			),
			ExpHdrs: hdrs(
				hdr("_id", fldTypeID),
			),
			ExpRows: rows(
				row(int64(0)),
				row(int64(1)),
				row(int64(2)),
				row(int64(3)),
			),
			Compare: CompareExactOrdered,
		},
		{
			name: "nested-inner-and-left-join",
			SQLs: sqls(
				"select DISTINCT u._id from users u inner join orders o on o.userid = u._id left join quantity q on o.userid = q.userid;",
			),
			ExpHdrs: hdrs(
				hdr("_id", fldTypeID),
			),
			ExpRows: rows(
				row(int64(0)),
				row(int64(1)),
				row(int64(2)),
				row(int64(3)),
			),
			Compare: CompareExactOrdered,
		},
		{
			name: "leftjoin",
			SQLs: sqls(
				"select u._id , o.userid from users u left join orders o on o.userid = u._id;",
			),
			ExpHdrs: hdrs(
				hdr("_id", fldTypeID),
				hdr("userid", fldTypeInt),
			),
			ExpRows: rows(
				row(int64(0), int64(0)),
				row(int64(1), int64(1)),
				row(int64(1), int64(1)),
				row(int64(2), int64(2)),
				row(int64(2), int64(2)),
				row(int64(3), int64(3)),
				row(int64(4), nil),
			),
			Compare: CompareExactOrdered,
		},
		{
			name: "nested-left-join",
			SQLs: sqls(
				"select distinct u._id, o.userid from users u left join orders o on o.userid = u._id left join quantity q on o.userid = q.userid;",
			),
			ExpHdrs: hdrs(
				hdr("_id", fldTypeID),
				hdr("userid", fldTypeInt),
			),
			ExpRows: rows(
				row(int64(0), int64(0)),
				row(int64(1), int64(1)),
				row(int64(2), int64(2)),
				row(int64(3), int64(3)),
				row(int64(4), nil),
			),
			Compare: CompareExactOrdered,
		},
		// test u.* and expect select list is expanded to all collumns in table alias u
		{
			name: "join-select-start",
			SQLs: sqls(
				"select distinct u.* from users u join orders o on o.userid = u._id;",
			),
			ExpHdrs: hdrs(
				hdr("_id", fldTypeID),
				hdr("name", fldTypeString),
				hdr("age", fldTypeInt),
			),
			ExpRows: rows(
				row(int64(0), string("a"), int64(21)),
				row(int64(1), string("b"), int64(18)),
				row(int64(2), string("c"), int64(28)),
				row(int64(3), string("d"), int64(34)),
			),
			Compare: CompareExactOrdered,
		},
		{
			name: "Unmatched-columns-in-join",
			SQLs: sqls(
				"select u._id, o.userid from users u join orders o on u.name = o.userid;",
			),
			ExpErr: "types 'string' and 'int' are not equatable",
		},
		{
			name: "fulljoin",
			SQLs: sqls(
				"select u._id , o.userid from users u full join orders o on o.userid = u._id;",
			),
			ExpErr: "FULL join types are not supported",
		},
		{
			name: "outerjoin",
			SQLs: sqls(
				"select u._id , o.userid from users u right join orders o on o.userid = u._id;",
			),
			ExpErr: "RIGHT join types are not supported",
		},
		{
			name: "nested-inner-join-with-right-join",
			SQLs: sqls(
				"select * from users u inner join orders o on o.userid = u._id right join quantity q on o.userid = q.userid;",
			),
			ExpErr: "RIGHT join types are not supported",
		},
		{
			name: "commajoin",
			SQLs: sqls(
				"select u._id, u.name, u.age, u2._id as u2_id, u2.name as u2name, u2.age as u2age from users u, (select * from users where _id=2) u2 where u._id=u2._id;",
			),
			ExpHdrs: hdrs(
				hdr("_id", fldTypeID),
				hdr("name", fldTypeString),
				hdr("age", fldTypeInt),
				hdr("u2_id", fldTypeID),
				hdr("u2name", fldTypeString),
				hdr("u2age", fldTypeInt),
			),
			ExpRows: rows(
				row(int64(2), string("c"), int64(28), int64(2), string("c"), int64(28)),
			),
			Compare: CompareExactUnordered,
		},
	},
	PQLTests: []PQLTest{
		{
			name:  "distinctjoin",
			Table: "users",
			PQLs:  []string{"Intersect(Distinct(Row(price > 10), index=orders, field=userid))"},
			ExpHdrs: hdrs(
				hdr("_id", fldTypeID),
			),
			ExpRows: rows(
				row(int64(1)),
				row(int64(2)),
			),
		},
	},
}