steve-chavez
82b38341bf
* test text/html and drop HtmlRawOutputSpec.hs * all tests passing, removed all pendingWith * make functions compatible with pg <= 12 * move custom media types tests to own spec * anyelement aggregate * apply aggregates without a final function * overriding works * overriding anyelement with particular agg * cannot override vendored media types * plan spec works with custom aggregate * renamed media types to make clear which ones are overridable * correct content negotiation with same weight * text/tab-separated-values media type * text/csv with BOM plus content-disposition header
225 lines
10 KiB
Haskell
225 lines
10 KiB
Haskell
module Feature.Query.CustomMediaSpec where
|
|
|
|
import Network.Wai (Application)
|
|
|
|
import Network.HTTP.Types
|
|
import Network.Wai.Test (SResponse (simpleBody, simpleHeaders, simpleStatus))
|
|
import Test.Hspec
|
|
import Test.Hspec.Wai
|
|
import Test.Hspec.Wai.JSON
|
|
import Text.Heredoc (str)
|
|
|
|
import Protolude hiding (get)
|
|
import SpecHelper
|
|
|
|
spec :: SpecWith ((), Application)
|
|
spec = describe "custom media types" $ do
|
|
context "for tables with aggregate" $ do
|
|
it "can query if there's an aggregate defined for the table" $ do
|
|
r <- request methodGet "/lines" (acceptHdrs "application/vnd.twkb") ""
|
|
liftIO $ do
|
|
simpleBody r `shouldBe` readFixtureFile "lines.twkb"
|
|
simpleHeaders r `shouldContain` [("Content-Type", "application/vnd.twkb")]
|
|
|
|
it "can query by id if there's an aggregate defined for the table" $ do
|
|
r <- request methodGet "/lines?id=eq.1" (acceptHdrs "application/vnd.twkb") ""
|
|
liftIO $ do
|
|
simpleBody r `shouldBe` readFixtureFile "1.twkb"
|
|
simpleHeaders r `shouldContain` [("Content-Type", "application/vnd.twkb")]
|
|
|
|
it "will fail if there's no aggregate defined for the table" $ do
|
|
request methodGet "/lines" (acceptHdrs "text/plain") ""
|
|
`shouldRespondWith`
|
|
[json| {"code":"PGRST107","details":null,"hint":null,"message":"None of these media types are available: text/plain"} |]
|
|
{ matchStatus = 415
|
|
, matchHeaders = [matchContentTypeJson]
|
|
}
|
|
|
|
it "can get raw xml output with Accept: text/xml if there's an aggregate defined" $ do
|
|
request methodGet "/xmltest" (acceptHdrs "text/xml") ""
|
|
`shouldRespondWith`
|
|
"<myxml>foo</myxml>bar<foobar><baz/></foobar>"
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "text/xml; charset=utf-8"]
|
|
}
|
|
|
|
-- TODO SOH (start of heading) is being added to results
|
|
context "for tables with anyelement aggregate" $ do
|
|
it "will use the application/vnd.geo2+json media type for any table" $
|
|
request methodGet "/lines" (acceptHdrs "application/vnd.geo2+json") ""
|
|
`shouldRespondWith`
|
|
"\SOH{\"type\": \"FeatureCollection\", \"hello\": \"world\"}"
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "application/vnd.geo2+json"]
|
|
}
|
|
|
|
it "will use the more specific application/vnd.geo2 handler for this table" $ do
|
|
request methodGet "/shop_bles" (acceptHdrs "application/vnd.geo2+json") ""
|
|
`shouldRespondWith`
|
|
"\SOH\"anyelement overridden\""
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "application/vnd.geo2+json"]
|
|
}
|
|
|
|
request methodGet "/rpc/get_shop_bles" (acceptHdrs "application/vnd.geo2+json") ""
|
|
`shouldRespondWith`
|
|
"\SOH\"anyelement overridden\""
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "application/vnd.geo2+json"]
|
|
}
|
|
|
|
context "Proc that returns scalar" $ do
|
|
it "can get raw output with Accept: text/html" $ do
|
|
request methodGet "/rpc/welcome.html" (acceptHdrs "text/html") ""
|
|
`shouldRespondWith`
|
|
[str|
|
|
|<html>
|
|
| <head>
|
|
| <title>PostgREST</title>
|
|
| </head>
|
|
| <body>
|
|
| <h1>Welcome to PostgREST</h1>
|
|
| </body>
|
|
|</html>
|
|
|]
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "text/html"]
|
|
}
|
|
|
|
it "can get raw output with Accept: text/plain" $ do
|
|
request methodGet "/rpc/welcome" (acceptHdrs "text/plain") ""
|
|
`shouldRespondWith` "Welcome to PostgREST"
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "text/plain; charset=utf-8"]
|
|
}
|
|
|
|
it "can get raw xml output with Accept: text/xml" $ do
|
|
request methodGet "/rpc/return_scalar_xml" (acceptHdrs "text/xml") ""
|
|
`shouldRespondWith`
|
|
"<my-xml-tag/>"
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "text/xml; charset=utf-8"]
|
|
}
|
|
|
|
it "can get raw xml output with Accept: text/xml" $ do
|
|
request methodGet "/rpc/welcome.xml" (acceptHdrs "text/xml") ""
|
|
`shouldRespondWith`
|
|
"<html>\n <head>\n <title>PostgREST</title>\n </head>\n <body>\n <h1>Welcome to PostgREST</h1>\n </body>\n</html>"
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "text/xml; charset=utf-8"]
|
|
}
|
|
|
|
it "should fail with function returning text and Accept: text/xml" $ do
|
|
request methodGet "/rpc/welcome" (acceptHdrs "text/xml") ""
|
|
`shouldRespondWith`
|
|
[json|
|
|
{"code":"PGRST107","details":null,"hint":null,"message":"None of these media types are available: text/xml"}
|
|
|]
|
|
{ matchStatus = 415
|
|
, matchHeaders = ["Content-Type" <:> "application/json; charset=utf-8"]
|
|
}
|
|
|
|
context "Proc that returns scalar based on a table" $ do
|
|
it "can get an image with Accept: image/png" $ do
|
|
r <- request methodGet "/rpc/ret_image" (acceptHdrs "image/png") ""
|
|
liftIO $ do
|
|
simpleBody r `shouldBe` readFixtureFile "A.png"
|
|
simpleHeaders r `shouldContain` [("Content-Type", "image/png")]
|
|
|
|
context "Proc that returns set of scalars and Accept: text/plain" $
|
|
it "will err because only scalars work with media type domains" $ do
|
|
request methodGet "/rpc/welcome_twice"
|
|
(acceptHdrs "text/plain")
|
|
""
|
|
`shouldRespondWith`
|
|
[json|{"code":"PGRST107","details":null,"hint":null,"message":"None of these media types are available: text/plain"}|]
|
|
{ matchStatus = 415
|
|
, matchHeaders = ["Content-Type" <:> "application/json; charset=utf-8"]
|
|
}
|
|
|
|
context "Proc that returns rows and accepts custom media type" $ do
|
|
it "works if it has an aggregate defined" $ do
|
|
r <- request methodGet "/rpc/get_lines" [("Accept", "application/vnd.twkb")] ""
|
|
liftIO $ do
|
|
simpleBody r `shouldBe` readFixtureFile "lines.twkb"
|
|
simpleHeaders r `shouldContain` [("Content-Type", "application/vnd.twkb")]
|
|
|
|
it "fails if doesn't have an aggregate defined" $ do
|
|
request methodGet "/rpc/get_lines"
|
|
(acceptHdrs "application/octet-stream") ""
|
|
`shouldRespondWith`
|
|
[json| {"code":"PGRST107","details":null,"hint":null,"message":"None of these media types are available: application/octet-stream"} |]
|
|
{ matchStatus = 415 }
|
|
|
|
-- TODO SOH (start of heading) is being added to results
|
|
it "works if there's an anyelement aggregate defined" $ do
|
|
request methodGet "/rpc/get_lines" (acceptHdrs "application/vnd.geo2+json") ""
|
|
`shouldRespondWith`
|
|
"\SOH{\"type\": \"FeatureCollection\", \"hello\": \"world\"}"
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "application/vnd.geo2+json"]
|
|
}
|
|
|
|
context "overriding" $ do
|
|
it "will override the application/json handler for a single table" $
|
|
request methodGet "/ov_json" (acceptHdrs "application/json") ""
|
|
`shouldRespondWith`
|
|
[json| {"overridden": "true"} |]
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "application/json; charset=utf-8"]
|
|
}
|
|
|
|
-- TODO SOH (start of heading) is being added to results
|
|
it "will override the application/geo+json handler for a single table" $
|
|
request methodGet "/lines?id=eq.1" (acceptHdrs "application/geo+json") ""
|
|
`shouldRespondWith`
|
|
"\SOH{\"crs\": {\"type\": \"name\", \"properties\": {\"name\": \"EPSG:4326\"}}, \"type\": \"FeatureCollection\", \"features\": [{\"type\": \"Feature\", \"geometry\": {\"type\": \"LineString\", \"coordinates\": [[1, 1], [5, 5]]}, \"properties\": {\"id\": 1, \"name\": \"line-1\"}}]}"
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "application/geo+json; charset=utf-8"]
|
|
}
|
|
|
|
it "will not override vendored media types like application/vnd.pgrst.object" $
|
|
request methodGet "/projects?id=eq.1" (acceptHdrs "application/vnd.pgrst.object") ""
|
|
`shouldRespondWith`
|
|
[json|{"id":1,"name":"Windows 7","client_id":1}|]
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "application/vnd.pgrst.object+json; charset=utf-8"]
|
|
}
|
|
|
|
context "matches requested media type correctly" $ do
|
|
-- https://github.com/PostgREST/postgrest/issues/1462
|
|
it "will match image/png according to q values" $ do
|
|
r1 <- request methodGet "/rpc/ret_image" (acceptHdrs "image/png, */*") ""
|
|
liftIO $ do
|
|
simpleBody r1 `shouldBe` readFixtureFile "A.png"
|
|
simpleHeaders r1 `shouldContain` [("Content-Type", "image/png")]
|
|
|
|
r2 <- request methodGet "/rpc/ret_image" (acceptHdrs "text/html,application/xhtml+xml,application/xml;q=0.9,image/png,*/*;q=0.8") ""
|
|
liftIO $ do
|
|
simpleBody r2 `shouldBe` readFixtureFile "A.png"
|
|
simpleHeaders r2 `shouldContain` [("Content-Type", "image/png")]
|
|
|
|
-- https://github.com/PostgREST/postgrest/issues/2170
|
|
it "will match json in presence of text/plain" $ do
|
|
r <- request methodGet "/projects?id=eq.1" (acceptHdrs "text/plain, application/json") ""
|
|
liftIO $ do
|
|
simpleStatus r `shouldBe` status200
|
|
simpleHeaders r `shouldContain` [("Content-Type", "application/json; charset=utf-8")]
|
|
|
|
-- https://github.com/PostgREST/postgrest/issues/1102
|
|
it "will match a custom text/tab-separated-values" $ do
|
|
request methodGet "/projects?id=in.(1,2)" (acceptHdrs "text/tab-separated-values") ""
|
|
`shouldRespondWith`
|
|
"id\tname\tclient_id\n1\tWindows 7\t1\n2\tWindows 10\t1\n"
|
|
{ matchStatus = 200
|
|
, matchHeaders = ["Content-Type" <:> "text/tab-separated-values"]
|
|
}
|
|
|
|
-- https://github.com/PostgREST/postgrest/issues/1371#issuecomment-519248984
|
|
it "will match a custom text/csv with BOM" $ do
|
|
r <- request methodGet "/lines" (acceptHdrs "text/csv") ""
|
|
liftIO $ do
|
|
simpleBody r `shouldBe` readFixtureFile "lines.csv"
|
|
simpleHeaders r `shouldContain` [("Content-Type", "text/csv; charset=utf-8")]
|
|
simpleHeaders r `shouldContain` [("Content-Disposition", "attachment; filename=\"lines.csv\"")]
|