291 lines
10 KiB
Plaintext
291 lines
10 KiB
Plaintext
|
|
Tests for parsing scalar expressions
|
|
|
|
> {-# LANGUAGE OverloadedStrings #-}
|
|
> module Language.SQL.SimpleSQL.ScalarExprs (scalarExprTests) where
|
|
|
|
> import Language.SQL.SimpleSQL.TestTypes
|
|
> import Language.SQL.SimpleSQL.Syntax
|
|
|
|
> scalarExprTests :: TestItem
|
|
> scalarExprTests = Group "scalarExprTests"
|
|
> [literals
|
|
> ,identifiers
|
|
> ,star
|
|
> ,dots
|
|
> ,app
|
|
> ,caseexp
|
|
> ,operators
|
|
> ,parens
|
|
> ,subqueries
|
|
> ,aggregates
|
|
> ,windowFunctions
|
|
> ]
|
|
|
|
> literals :: TestItem
|
|
> literals = Group "literals" $ map (uncurry TestScalarExpr)
|
|
> [("3", NumLit "3")
|
|
> ,("3.", NumLit "3.")
|
|
> ,("3.3", NumLit "3.3")
|
|
> ,(".3", NumLit ".3")
|
|
> ,("3.e3", NumLit "3.e3")
|
|
> ,("3.3e3", NumLit "3.3e3")
|
|
> ,(".3e3", NumLit ".3e3")
|
|
> ,("3e3", NumLit "3e3")
|
|
> ,("3e+3", NumLit "3e+3")
|
|
> ,("3e-3", NumLit "3e-3")
|
|
> ,("'string'", StringLit "string")
|
|
> ,("'string with a '' quote'", StringLit "string with a ' quote")
|
|
> ,("'1'", StringLit "1")
|
|
> ,("interval '3' day", IntervalLit "3" "day" Nothing)
|
|
> ,("interval '3' day (3)", IntervalLit "3" "day" $ Just 3)
|
|
> ,("interval '3 weeks'", TypedLit (TypeName "interval") "3 weeks")
|
|
> ]
|
|
|
|
> identifiers :: TestItem
|
|
> identifiers = Group "identifiers" $ map (uncurry TestScalarExpr)
|
|
> [("iden1", Iden "iden1")
|
|
> --,("t.a", Iden2 "t" "a")
|
|
> ,("\"quoted identifier\"", Iden $ QName "quoted identifier")
|
|
> ]
|
|
|
|
> star :: TestItem
|
|
> star = Group "star" $ map (uncurry TestScalarExpr)
|
|
> [("*", Star)
|
|
> --,("t.*", Star2 "t")
|
|
> --,("ROW(t.*,42)", App "ROW" [Star2 "t", NumLit "42"])
|
|
> ]
|
|
|
|
> dots :: TestItem
|
|
> dots = Group "dot" $ map (uncurry TestScalarExpr)
|
|
> [("t.a", BinOp (Iden "t") "." (Iden "a"))
|
|
> ,("t.*", BinOp (Iden "t") "." Star)
|
|
> ,("a.b.c", BinOp (BinOp (Iden "a") "." (Iden "b")) "." (Iden "c"))
|
|
> ,("ROW(t.*,42)", App "ROW" [BinOp (Iden "t") "." Star, NumLit "42"])
|
|
> ]
|
|
|
|
> app :: TestItem
|
|
> app = Group "app" $ map (uncurry TestScalarExpr)
|
|
> [("f()", App "f" [])
|
|
> ,("f(a)", App "f" [Iden "a"])
|
|
> ,("f(a,b)", App "f" [Iden "a", Iden "b"])
|
|
> ]
|
|
|
|
> caseexp :: TestItem
|
|
> caseexp = Group "caseexp" $ map (uncurry TestScalarExpr)
|
|
> [("case a when 1 then 2 end"
|
|
> ,Case (Just $ Iden "a") [([NumLit "1"]
|
|
> ,NumLit "2")] Nothing)
|
|
|
|
> ,("case a when 1 then 2 when 3 then 4 end"
|
|
> ,Case (Just $ Iden "a") [([NumLit "1"], NumLit "2")
|
|
> ,([NumLit "3"], NumLit "4")] Nothing)
|
|
|
|
> ,("case a when 1 then 2 when 3 then 4 else 5 end"
|
|
> ,Case (Just $ Iden "a") [([NumLit "1"], NumLit "2")
|
|
> ,([NumLit "3"], NumLit "4")] (Just $ NumLit "5"))
|
|
|
|
> ,("case when a=1 then 2 when a=3 then 4 else 5 end"
|
|
> ,Case Nothing [([BinOp (Iden "a") "=" (NumLit "1")], NumLit "2")
|
|
> ,([BinOp (Iden "a") "=" (NumLit "3")], NumLit "4")]
|
|
> (Just $ NumLit "5"))
|
|
|
|
> ,("case a when 1,2 then 10 when 3,4 then 20 end"
|
|
> ,Case (Just $ Iden "a") [([NumLit "1",NumLit "2"]
|
|
> ,NumLit "10")
|
|
> ,([NumLit "3",NumLit "4"]
|
|
> ,NumLit "20")]
|
|
> Nothing)
|
|
|
|
> ]
|
|
|
|
> operators :: TestItem
|
|
> operators = Group "operators"
|
|
> [binaryOperators
|
|
> ,unaryOperators
|
|
> ,casts
|
|
> ,miscOps]
|
|
|
|
> binaryOperators :: TestItem
|
|
> binaryOperators = Group "binaryOperators" $ map (uncurry TestScalarExpr)
|
|
> [("a + b", BinOp (Iden "a") "+" (Iden "b"))
|
|
> -- sanity check fixities
|
|
> -- todo: add more fixity checking
|
|
|
|
> ,("a + b * c"
|
|
> ,BinOp (Iden "a") "+"
|
|
> (BinOp (Iden "b") "*" (Iden "c")))
|
|
|
|
> ,("a * b + c"
|
|
> ,BinOp (BinOp (Iden "a") "*" (Iden "b"))
|
|
> "+" (Iden "c"))
|
|
> ]
|
|
|
|
> unaryOperators :: TestItem
|
|
> unaryOperators = Group "unaryOperators" $ map (uncurry TestScalarExpr)
|
|
> [("not a", PrefixOp "not" $ Iden "a")
|
|
> ,("not not a", PrefixOp "not" $ PrefixOp "not" $ Iden "a")
|
|
> ,("+a", PrefixOp "+" $ Iden "a")
|
|
> ,("-a", PrefixOp "-" $ Iden "a")
|
|
> ]
|
|
|
|
|
|
> casts :: TestItem
|
|
> casts = Group "operators" $ map (uncurry TestScalarExpr)
|
|
> [("cast('1' as int)"
|
|
> ,Cast (StringLit "1") $ TypeName "int")
|
|
|
|
> ,("int '3'"
|
|
> ,TypedLit (TypeName "int") "3")
|
|
|
|
> ,("cast('1' as double precision)"
|
|
> ,Cast (StringLit "1") $ TypeName "double precision")
|
|
|
|
> ,("cast('1' as float(8))"
|
|
> ,Cast (StringLit "1") $ PrecTypeName "float" 8)
|
|
|
|
> ,("cast('1' as decimal(15,2))"
|
|
> ,Cast (StringLit "1") $ Prec2TypeName "decimal" 15 2)
|
|
|
|
|
|
> ,("double precision '3'"
|
|
> ,TypedLit (TypeName "double precision") "3")
|
|
> ]
|
|
|
|
> subqueries :: TestItem
|
|
> subqueries = Group "unaryOperators" $ map (uncurry TestScalarExpr)
|
|
> [("exists (select a from t)", SubQueryExpr SqExists ms)
|
|
> ,("(select a from t)", SubQueryExpr SqSq ms)
|
|
|
|
> ,("a in (select a from t)"
|
|
> ,In True (Iden "a") (InQueryExpr ms))
|
|
|
|
> ,("a not in (select a from t)"
|
|
> ,In False (Iden "a") (InQueryExpr ms))
|
|
|
|
> ,("a > all (select a from t)"
|
|
> ,BinOp (Iden "a") ">" (SubQueryExpr SqAll ms))
|
|
|
|
> ,("a = some (select a from t)"
|
|
> ,BinOp (Iden "a") "=" (SubQueryExpr SqSome ms))
|
|
|
|
> ,("a <= any (select a from t)"
|
|
> ,BinOp (Iden "a") "<=" (SubQueryExpr SqAny ms))
|
|
> ]
|
|
> where
|
|
> ms = makeSelect
|
|
> {qeSelectList = [(Nothing,Iden "a")]
|
|
> ,qeFrom = [TRSimple "t"]
|
|
> }
|
|
|
|
> miscOps :: TestItem
|
|
> miscOps = Group "unaryOperators" $ map (uncurry TestScalarExpr)
|
|
> [("a in (1,2,3)"
|
|
> ,In True (Iden "a") $ InList $ map NumLit ["1","2","3"])
|
|
|
|
> ,("a between b and c", SpecialOp "between" [Iden "a"
|
|
> ,Iden "b"
|
|
> ,Iden "c"])
|
|
|
|
> ,("a not between b and c", SpecialOp "not between" [Iden "a"
|
|
> ,Iden "b"
|
|
> ,Iden "c"])
|
|
|
|
> ,("a is null", PostfixOp "is null" (Iden "a"))
|
|
> ,("a is not null", PostfixOp "is not null" (Iden "a"))
|
|
> ,("a is true", PostfixOp "is true" (Iden "a"))
|
|
> ,("a is not true", PostfixOp "is not true" (Iden "a"))
|
|
> ,("a is false", PostfixOp "is false" (Iden "a"))
|
|
> ,("a is not false", PostfixOp "is not false" (Iden "a"))
|
|
> ,("a is unknown", PostfixOp "is unknown" (Iden "a"))
|
|
> ,("a is not unknown", PostfixOp "is not unknown" (Iden "a"))
|
|
> ,("a is distinct from b", BinOp (Iden "a") "is distinct from"(Iden "b"))
|
|
|
|
> ,("a is not distinct from b"
|
|
> ,BinOp (Iden "a") "is not distinct from" (Iden "b"))
|
|
|
|
> ,("a like b", BinOp (Iden "a") "like" (Iden "b"))
|
|
> ,("a not like b", BinOp (Iden "a") "not like" (Iden "b"))
|
|
> ,("a is similar to b", BinOp (Iden "a") "is similar to" (Iden "b"))
|
|
|
|
> ,("a is not similar to b"
|
|
> ,BinOp (Iden "a") "is not similar to" (Iden "b"))
|
|
|
|
> ,("a overlaps b", BinOp (Iden "a") "overlaps" (Iden "b"))
|
|
> ,("extract(day from t)", SpecialOp "extract" [Iden "day", Iden "t"])
|
|
|
|
> ,("substring(x from 1 for 2)"
|
|
> ,SpecialOp "substring" [Iden "x", NumLit "1", NumLit "2"])
|
|
|
|
> ,("(1,2)"
|
|
> ,SpecialOp "rowctor" [NumLit "1", NumLit "2"])
|
|
|
|
> ]
|
|
|
|
> aggregates :: TestItem
|
|
> aggregates = Group "aggregates" $ map (uncurry TestScalarExpr)
|
|
> [("count(*)",App "count" [Star])
|
|
|
|
> ,("sum(a order by a)"
|
|
> ,AggregateApp "sum" Nothing [Iden "a"] [(OrderField (Iden "a") Asc NullsOrderDefault)])
|
|
|
|
> ,("sum(all a)"
|
|
> ,AggregateApp "sum" (Just All) [Iden "a"] [])
|
|
|
|
> ,("count(distinct a)"
|
|
> ,AggregateApp "count" (Just Distinct) [Iden "a"] [])
|
|
> ]
|
|
|
|
> windowFunctions :: TestItem
|
|
> windowFunctions = Group "windowFunctions" $ map (uncurry TestScalarExpr)
|
|
> [("max(a) over ()", WindowApp "max" [Iden "a"] [] [] Nothing)
|
|
> ,("count(*) over ()", WindowApp "count" [Star] [] [] Nothing)
|
|
|
|
> ,("max(a) over (partition by b)"
|
|
> ,WindowApp "max" [Iden "a"] [Iden "b"] [] Nothing)
|
|
|
|
> ,("max(a) over (partition by b,c)"
|
|
> ,WindowApp "max" [Iden "a"] [Iden "b",Iden "c"] [] Nothing)
|
|
|
|
> ,("sum(a) over (order by b)"
|
|
> ,WindowApp "sum" [Iden "a"] [] [(OrderField (Iden "b") Asc NullsOrderDefault)] Nothing)
|
|
|
|
> ,("sum(a) over (order by b desc,c)"
|
|
> ,WindowApp "sum" [Iden "a"] [] [(OrderField (Iden "b") Desc NullsOrderDefault)
|
|
> ,(OrderField (Iden "c") Asc NullsOrderDefault)] Nothing)
|
|
|
|
> ,("sum(a) over (partition by b order by c)"
|
|
> ,WindowApp "sum" [Iden "a"] [Iden "b"] [OrderField (Iden "c") Asc NullsOrderDefault] Nothing)
|
|
|
|
> ,("sum(a) over (partition by b order by c range unbounded preceding)"
|
|
> ,WindowApp "sum" [Iden "a"] [Iden "b"] [OrderField (Iden "c") Asc NullsOrderDefault]
|
|
> $ Just $ FrameFrom FrameRange UnboundedPreceding)
|
|
|
|
> ,("sum(a) over (partition by b order by c range 5 preceding)"
|
|
> ,WindowApp "sum" [Iden "a"] [Iden "b"] [OrderField (Iden "c") Asc NullsOrderDefault]
|
|
> $ Just $ FrameFrom FrameRange $ Preceding (NumLit "5"))
|
|
|
|
> ,("sum(a) over (partition by b order by c range current row)"
|
|
> ,WindowApp "sum" [Iden "a"] [Iden "b"] [OrderField (Iden "c") Asc NullsOrderDefault]
|
|
> $ Just $ FrameFrom FrameRange Current)
|
|
|
|
> ,("sum(a) over (partition by b order by c rows 5 following)"
|
|
> ,WindowApp "sum" [Iden "a"] [Iden "b"] [OrderField (Iden "c") Asc NullsOrderDefault]
|
|
> $ Just $ FrameFrom FrameRows $ Following (NumLit "5"))
|
|
|
|
> ,("sum(a) over (partition by b order by c range unbounded following)"
|
|
> ,WindowApp "sum" [Iden "a"] [Iden "b"] [OrderField (Iden "c") Asc NullsOrderDefault]
|
|
> $ Just $ FrameFrom FrameRange UnboundedFollowing)
|
|
|
|
> ,("sum(a) over (partition by b order by c range between 5 preceding and 5 following)"
|
|
> ,WindowApp "sum" [Iden "a"] [Iden "b"] [OrderField (Iden "c") Asc NullsOrderDefault]
|
|
> $ Just $ FrameBetween FrameRange (Preceding (NumLit "5")) (Following (NumLit "5")))
|
|
|
|
> ]
|
|
|
|
> parens :: TestItem
|
|
> parens = Group "parens" $ map (uncurry TestScalarExpr)
|
|
> [("(a)", Parens (Iden "a"))
|
|
> ,("(a + b)", Parens (BinOp (Iden "a") "+" (Iden "b")))
|
|
> ]
|