mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-22 06:51:32 +03:00
cd6fe41b99
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/5731 GitOrigin-RevId: 0af22a85bd6b1cb8888562e05cde640047e56b41
587 lines
16 KiB
Haskell
587 lines
16 KiB
Haskell
{-# LANGUAGE QuasiQuotes #-}
|
|
|
|
-- | Query Tests for Data Connector Backend
|
|
module Test.DataConnector.QuerySpec
|
|
( spec,
|
|
)
|
|
where
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
import Harness.Backend.DataConnector qualified as DataConnector
|
|
import Harness.GraphqlEngine qualified as GraphqlEngine
|
|
import Harness.Quoter.Graphql (graphql)
|
|
import Harness.Quoter.Yaml (yaml)
|
|
import Harness.Test.Fixture qualified as Fixture
|
|
import Harness.TestEnvironment (TestEnvironment)
|
|
import Harness.TestEnvironment qualified as TE
|
|
import Harness.Yaml (shouldReturnYaml)
|
|
import Hasura.Prelude
|
|
import Test.Hspec (SpecWith, describe, it, pendingWith)
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- Reference Agent Query Tests
|
|
|
|
spec :: SpecWith TestEnvironment
|
|
spec =
|
|
Fixture.runWithLocalTestEnvironment
|
|
( ( \(DataConnector.TestSourceConfig backendType backendConfig _sourceConfig md) ->
|
|
(Fixture.fixture $ Fixture.Backend backendType)
|
|
{ Fixture.setupTeardown = \(testEnv, _) ->
|
|
[ DataConnector.setupFixtureAction
|
|
md
|
|
backendConfig
|
|
testEnv
|
|
]
|
|
}
|
|
)
|
|
<$> DataConnector.backendConfigs
|
|
)
|
|
tests
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
tests :: Fixture.Options -> SpecWith (TestEnvironment, a)
|
|
tests opts = describe "Queries" $ do
|
|
describe "Basic Tests" $ do
|
|
it "works with simple object query" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums(limit: 1) {
|
|
id
|
|
title
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums:
|
|
- id: 1
|
|
title: For Those About To Rock We Salute You
|
|
|]
|
|
|
|
it "works with a primary key" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums_by_pk(id: 1) {
|
|
id
|
|
title
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums_by_pk:
|
|
id: 1
|
|
title: "For Those About To Rock We Salute You"
|
|
|]
|
|
|
|
it "works with non existent primary key" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums_by_pk(id: 999999) {
|
|
id
|
|
title
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums_by_pk: null
|
|
|]
|
|
|
|
it "works with a composite primary key" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
PlaylistTrack_by_pk(PlaylistId: 1, TrackId: 2) {
|
|
Playlist {
|
|
Name
|
|
}
|
|
Track {
|
|
Name
|
|
}
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
PlaylistTrack_by_pk:
|
|
Playlist:
|
|
Name: "Music"
|
|
Track:
|
|
Name: "Balls to the Wall"
|
|
|]
|
|
|
|
it "works with pagination" $ \(testEnvironment, _) -> do
|
|
-- NOTE: We order by in this pagination test to ensure that the rows are ordered correctly (which they are not in db.chinook.sqlite)
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums (limit: 3, offset: 2, order_by: {id: asc}) {
|
|
id
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums:
|
|
- id: 3
|
|
- id: 4
|
|
- id: 5
|
|
|]
|
|
|
|
describe "Array Relationships" $ do
|
|
describe "Manual" $ do
|
|
it "joins on album id" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getArtist {
|
|
artists_by_pk(id: 1) {
|
|
id
|
|
name
|
|
albums {
|
|
title
|
|
}
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
artists_by_pk:
|
|
name: AC/DC
|
|
id: 1
|
|
albums:
|
|
- title: For Those About To Rock We Salute You
|
|
- title: Let There Be Rock
|
|
|]
|
|
|
|
describe "Foreign Key Constraint On" do
|
|
it "joins on PlaylistId" $ \(testEnvironment, _) -> do
|
|
-- NOTE: Ordering is used for the query due to inconsistencies in data-set ordering.
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getPlaylist {
|
|
Playlist_by_pk(PlaylistId: 1) {
|
|
Tracks (order_by: {TrackId: desc}, limit: 3) {
|
|
TrackId
|
|
}
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
Playlist_by_pk:
|
|
Tracks:
|
|
- TrackId: 3503
|
|
- TrackId: 3502
|
|
- TrackId: 3501
|
|
|]
|
|
|
|
describe "Object Relationships" do
|
|
describe "Manual" do
|
|
it "joins on artist id" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums_by_pk(id: 1) {
|
|
id
|
|
title
|
|
artist {
|
|
name
|
|
}
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums_by_pk:
|
|
id: 1
|
|
title: "For Those About To Rock We Salute You"
|
|
artist:
|
|
name: "AC/DC"
|
|
|]
|
|
|
|
describe "Foreign Key Constraint On" $ do
|
|
it "joins on PlaylistId" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getPlaylist {
|
|
PlaylistTrack_by_pk(PlaylistId: 1, TrackId: 2) {
|
|
Playlist {
|
|
Name
|
|
}
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
PlaylistTrack_by_pk:
|
|
Playlist:
|
|
Name: "Music"
|
|
|]
|
|
|
|
describe "Where Clause Tests" $ do
|
|
it "works with '_in' predicate" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums(where: {id: {_in: [1, 3, 5]}}) {
|
|
id
|
|
title
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums:
|
|
- id: 1
|
|
title: For Those About To Rock We Salute You
|
|
- id: 3
|
|
title: Restless and Wild
|
|
- id: 5
|
|
title: Big Ones
|
|
|]
|
|
|
|
it "works with '_nin' predicate" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums(where: {id: {_in: [1, 3, 5]}, title: {_nin: ["Big Ones"]}}) {
|
|
id
|
|
title
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums:
|
|
- id: 1
|
|
title: For Those About To Rock We Salute You
|
|
- id: 3
|
|
title: Restless and Wild
|
|
|]
|
|
|
|
it "works with '_eq' predicate" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums(where: {id: {_eq: 1}}) {
|
|
id
|
|
title
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums:
|
|
- id: 1
|
|
title: For Those About To Rock We Salute You
|
|
|]
|
|
|
|
it "works with '_neq' predicate" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums(where: {id: {_neq: 2, _in: [1, 2, 3]}}) {
|
|
id
|
|
title
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums:
|
|
- id: 1
|
|
title: For Those About To Rock We Salute You
|
|
- id: 3
|
|
title: Restless and Wild
|
|
|]
|
|
|
|
it "works with '_lt' predicate" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums(where: {id: {_lt: 2}}) {
|
|
id
|
|
title
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums:
|
|
- id: 1
|
|
title: For Those About To Rock We Salute You
|
|
|]
|
|
|
|
it "works with '_lte' predicate" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getArtists {
|
|
artists(where: {id: {_lte: 2}}) {
|
|
id
|
|
name
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
artists:
|
|
- id: 1
|
|
name: AC/DC
|
|
- id: 2
|
|
name: Accept
|
|
|]
|
|
|
|
it "works with '_gt' predicate" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getArtists {
|
|
artists(where: {id: {_gt: 274}}) {
|
|
id
|
|
name
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
artists:
|
|
- id: 275
|
|
name: Philip Glass Ensemble
|
|
|]
|
|
|
|
it "works with '_gte' predicate" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getArtists {
|
|
artists(where: {id: {_gte: 274}}) {
|
|
id
|
|
name
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
artists:
|
|
- id: 274
|
|
name: Nash Ensemble
|
|
- id: 275
|
|
name: Philip Glass Ensemble
|
|
|]
|
|
|
|
describe "Order By Tests" $ do
|
|
it "works with order_by id asc" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums(limit: 3, order_by: {id: asc}) {
|
|
id
|
|
title
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums:
|
|
- id: 1
|
|
title: For Those About To Rock We Salute You
|
|
- id: 2
|
|
title: Balls to the Wall
|
|
- id: 3
|
|
title: Restless and Wild
|
|
|]
|
|
|
|
it "works with order_by id desc" $ \(testEnvironment, _) ->
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbum {
|
|
albums(limit: 3, order_by: {id: desc}) {
|
|
id
|
|
title
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums:
|
|
- id: 347
|
|
title: Koyaanisqatsi (Soundtrack from the Motion Picture)
|
|
- id: 346
|
|
title: 'Mozart: Chamber Music'
|
|
- id: 345
|
|
title: 'Monteverdi: L''Orfeo'
|
|
|]
|
|
|
|
it "can order by an aggregate" $ \(testEnvironment, _) -> do
|
|
when (TE.backendType testEnvironment == Just Fixture.DataConnectorSqlite) (pendingWith "TODO: Test currently broken for SQLite DataConnector")
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getArtists {
|
|
artists(limit: 3, order_by: {albums_aggregate: {count: desc}}) {
|
|
name
|
|
albums_aggregate {
|
|
aggregate {
|
|
count
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
artists:
|
|
- name: Iron Maiden
|
|
albums_aggregate:
|
|
aggregate:
|
|
count: 21
|
|
- name: Led Zeppelin
|
|
albums_aggregate:
|
|
aggregate:
|
|
count: 14
|
|
- name: Deep Purple
|
|
albums_aggregate:
|
|
aggregate:
|
|
count: 11
|
|
|]
|
|
|
|
it "can order by a related field" $ \(testEnvironment, _) -> do
|
|
when (TE.backendType testEnvironment == Just Fixture.DataConnectorSqlite) (pendingWith "TODO: Test currently broken for SQLite DataConnector")
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query getAlbums {
|
|
albums(limit: 4, order_by: [{artist: {name: asc}}, {title: desc}]) {
|
|
artist {
|
|
name
|
|
}
|
|
title
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
albums:
|
|
- artist:
|
|
name: AC/DC
|
|
title: Let There Be Rock
|
|
- artist:
|
|
name: AC/DC
|
|
title: For Those About To Rock We Salute You
|
|
- artist:
|
|
name: Aaron Copland & London Symphony Orchestra
|
|
title: A Copland Celebration, Vol. I
|
|
- artist:
|
|
name: Aaron Goldberg
|
|
title: Worlds
|
|
|]
|
|
describe "Custom scalar types and operators" $ do
|
|
it "works with custom scalar types and comparison operators" $ \(testEnvironment, _) -> do
|
|
when (TE.backendType testEnvironment == Just Fixture.DataConnectorSqlite) do
|
|
pendingWith "TODO: Test currently broken for SQLite DataConnector"
|
|
shouldReturnYaml
|
|
opts
|
|
( GraphqlEngine.postGraphql
|
|
testEnvironment
|
|
[graphql|
|
|
query MyQuery {
|
|
employees(where: {birth_date: {in_year: 1965}}) {
|
|
birth_date
|
|
last_name
|
|
}
|
|
}
|
|
|]
|
|
)
|
|
[yaml|
|
|
data:
|
|
employees:
|
|
- birth_date: '1965-03-03T00:00:00-08:00'
|
|
last_name: Johnson
|
|
|]
|