mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-18 13:41:49 +03:00
106 lines
4.0 KiB
Go
106 lines
4.0 KiB
Go
package libsq_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/neilotoole/sq/drivers/mysql"
|
|
|
|
"github.com/neilotoole/sq/libsq/source"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
//nolint:exhaustive,lll
|
|
func TestQuery_where(t *testing.T) {
|
|
testCases := []queryTestCase{
|
|
{
|
|
name: "operator/eq",
|
|
in: `@sakila | .actor | where(.actor_id == 100)`,
|
|
wantSQL: `SELECT * FROM "actor" WHERE "actor_id" = 100`,
|
|
override: map[source.DriverType]string{mysql.Type: "SELECT * FROM `actor` WHERE `actor_id` = 100"},
|
|
wantRecCount: 1,
|
|
},
|
|
{
|
|
name: "operator/ne",
|
|
in: `@sakila | .actor | where(.actor_id != 100)`,
|
|
wantSQL: `SELECT * FROM "actor" WHERE "actor_id" != 100`,
|
|
override: map[source.DriverType]string{mysql.Type: "SELECT * FROM `actor` WHERE `actor_id` != 100"},
|
|
wantRecCount: 199,
|
|
},
|
|
{
|
|
name: "operator/lt",
|
|
in: `@sakila | .actor | where(.actor_id < 100)`,
|
|
wantSQL: `SELECT * FROM "actor" WHERE "actor_id" < 100`,
|
|
override: map[source.DriverType]string{mysql.Type: "SELECT * FROM `actor` WHERE `actor_id` < 100"},
|
|
wantRecCount: 99,
|
|
},
|
|
{
|
|
name: "operator/lte",
|
|
in: `@sakila | .actor | where(.actor_id <= 100)`,
|
|
wantSQL: `SELECT * FROM "actor" WHERE "actor_id" <= 100`,
|
|
override: map[source.DriverType]string{mysql.Type: "SELECT * FROM `actor` WHERE `actor_id` <= 100"},
|
|
wantRecCount: 100,
|
|
},
|
|
{
|
|
name: "operator/gt",
|
|
in: `@sakila | .actor | where(.actor_id > 100)`,
|
|
wantSQL: `SELECT * FROM "actor" WHERE "actor_id" > 100`,
|
|
override: map[source.DriverType]string{mysql.Type: "SELECT * FROM `actor` WHERE `actor_id` > 100"},
|
|
wantRecCount: 100,
|
|
},
|
|
{
|
|
name: "operator/gte",
|
|
in: `@sakila | .actor | where(.actor_id >= 100)`,
|
|
wantSQL: `SELECT * FROM "actor" WHERE "actor_id" >= 100`,
|
|
override: map[source.DriverType]string{mysql.Type: "SELECT * FROM `actor` WHERE `actor_id` >= 100"},
|
|
wantRecCount: 101,
|
|
},
|
|
{
|
|
name: "error/no-args",
|
|
in: `@sakila | .actor | where()`,
|
|
wantSQL: `SELECT * FROM "actor" WHERE "actor_id" = 1`,
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "select_alias",
|
|
in: `@sakila | .actor | select(.actor_id == 1)`,
|
|
wantSQL: `SELECT * FROM "actor" WHERE "actor_id" = 1`,
|
|
override: map[source.DriverType]string{mysql.Type: "SELECT * FROM `actor` WHERE `actor_id` = 1"},
|
|
wantRecCount: 1,
|
|
},
|
|
{
|
|
name: "where_compound_1",
|
|
in: `@sakila | .actor | where(.actor_id >= 100 && .actor_id < 150)`,
|
|
wantSQL: `SELECT * FROM "actor" WHERE "actor_id" >= 100 AND "actor_id" < 150`,
|
|
override: map[source.DriverType]string{mysql.Type: "SELECT * FROM `actor` WHERE `actor_id` >= 100 AND `actor_id` < 150"},
|
|
wantRecCount: 50,
|
|
},
|
|
{
|
|
name: "where_compound_2",
|
|
in: `@sakila | .actor | where(.actor_id >= 100 || (.actor_id < 150 && .first_name == "TOM"))`,
|
|
wantSQL: `SELECT * FROM "actor" WHERE "actor_id" >= 100 OR ("actor_id" < 150 AND "first_name" = 'TOM')`,
|
|
override: map[source.DriverType]string{mysql.Type: "SELECT * FROM `actor` WHERE `actor_id` >= 100 OR (`actor_id` < 150 AND `first_name` = 'TOM')"},
|
|
wantRecCount: 103,
|
|
},
|
|
{
|
|
name: "where_using_col_alias",
|
|
in: `@sakila | .actor | .first_name:given_name | where(.given_name == "TOM")`,
|
|
wantSQL: `SELECT "first_name" AS "given_name" FROM "actor" WHERE "given_name" = 'TOM'`,
|
|
override: map[source.DriverType]string{mysql.Type: "SELECT `first_name` AS `given_name` FROM `actor` WHERE `given_name` = 'TOM'"},
|
|
wantRecCount: 2,
|
|
// Skip because this only works on SQLite, not the other SQL databases.
|
|
// I'm not sure if this will ever be implemented. Perhaps sq could look at
|
|
// the ".given_name" element in the where clause, and substitute in the
|
|
// unaliased value? Could be messy. Either which way, not a high priority.
|
|
skip: true,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
tc := tc
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
execQueryTestCase(t, tc)
|
|
})
|
|
}
|
|
}
|