mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-19 05:21:47 +03:00
0723925ffc
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/4688 GitOrigin-RevId: f87b2e23be69aed3cc7f95e814f9426db174dc9b
96 lines
4.1 KiB
Haskell
96 lines
4.1 KiB
Haskell
module Test.QuerySpec.RelationshipsSpec (spec) where
|
|
|
|
import Autodocodec.Extended (ValueWrapper (..))
|
|
import Control.Lens (ix, (^?))
|
|
import Data.Aeson (Object, Value (..))
|
|
import Data.Aeson qualified as J
|
|
import Data.Aeson.KeyMap qualified as KeyMap
|
|
import Data.Aeson.Lens (_Number)
|
|
import Data.HashMap.Strict qualified as HashMap
|
|
import Data.List.NonEmpty (NonEmpty (..))
|
|
import Data.Text (Text)
|
|
import Data.Vector qualified as Vector
|
|
import Hasura.Backends.DataConnector.API
|
|
import Servant.API (NamedRoutes)
|
|
import Servant.Client (Client, (//))
|
|
import Test.Data qualified as Data
|
|
import Test.Hspec (Spec, describe, it)
|
|
import Test.Hspec.Expectations.Pretty (shouldBe)
|
|
import Prelude
|
|
|
|
spec :: Client IO (NamedRoutes Routes) -> SourceName -> Config -> Spec
|
|
spec api sourceName config = describe "Relationship Queries" $ do
|
|
it "perform a many to one query by joining artist to albums" $ do
|
|
let query = albumsWithArtistQuery id
|
|
receivedAlbums <- (Data.sortBy "ArtistId" . getQueryResponse) <$> (api // _query) sourceName config query
|
|
|
|
let joinInArtist (album :: Object) =
|
|
let artist = (album ^? ix "ArtistId" . _Number) >>= \artistId -> Data.artistsAsJsonById ^? ix artistId
|
|
artistPropVal = maybe J.Null Object artist
|
|
in KeyMap.insert "Artist" artistPropVal album
|
|
let removeArtistId = KeyMap.delete "ArtistId"
|
|
|
|
let expectedAlbums = (removeArtistId . joinInArtist) <$> Data.albumsAsJson
|
|
receivedAlbums `shouldBe` expectedAlbums
|
|
|
|
it "perform a one to many query by joining albums to artists" $ do
|
|
let query = artistsWithAlbumsQuery id
|
|
receivedArtists <- (Data.sortBy "ArtistId" . getQueryResponse) <$> (api // _query) sourceName config query
|
|
|
|
let joinInAlbums (artist :: Object) =
|
|
let artistId = artist ^? ix "ArtistId" . _Number
|
|
albumFilter artistId' album = album ^? ix "ArtistId" . _Number == Just artistId'
|
|
albums = maybe [] (\artistId' -> filter (albumFilter artistId') Data.albumsAsJson) artistId
|
|
albums' = Object . KeyMap.delete "ArtistId" <$> albums
|
|
in KeyMap.insert "Albums" (Array . Vector.fromList $ albums') artist
|
|
|
|
let expectedAlbums = joinInAlbums <$> Data.artistsAsJson
|
|
receivedArtists `shouldBe` expectedAlbums
|
|
|
|
albumsWithArtistQuery :: (Query -> Query) -> Query
|
|
albumsWithArtistQuery modifySubquery =
|
|
let joinFieldMapping =
|
|
HashMap.fromList
|
|
[ (PrimaryKey $ ColumnName "ArtistId", ForeignKey $ ColumnName "ArtistId")
|
|
]
|
|
artistsSubquery = modifySubquery artistsQuery
|
|
fields =
|
|
KeyMap.fromList
|
|
[ ("AlbumId", columnField "AlbumId"),
|
|
("Title", columnField "Title"),
|
|
("Artist", RelationshipField $ RelField joinFieldMapping ObjectRelationship artistsSubquery)
|
|
]
|
|
in albumsQuery {fields}
|
|
|
|
artistsWithAlbumsQuery :: (Query -> Query) -> Query
|
|
artistsWithAlbumsQuery modifySubquery =
|
|
let joinFieldMapping =
|
|
HashMap.fromList
|
|
[ (PrimaryKey $ ColumnName "ArtistId", ForeignKey $ ColumnName "ArtistId")
|
|
]
|
|
albumFields = KeyMap.fromList [("AlbumId", columnField "AlbumId"), ("Title", columnField "Title")]
|
|
albumsSort = OrderBy (ColumnName "AlbumId") Ascending :| []
|
|
albumsSubquery = modifySubquery (albumsQuery {fields = albumFields, orderBy = Just albumsSort})
|
|
fields =
|
|
KeyMap.fromList
|
|
[ ("ArtistId", columnField "ArtistId"),
|
|
("Name", columnField "Name"),
|
|
("Albums", RelationshipField $ RelField joinFieldMapping ArrayRelationship albumsSubquery)
|
|
]
|
|
in artistsQuery {fields}
|
|
|
|
artistsQuery :: Query
|
|
artistsQuery =
|
|
let fields = KeyMap.fromList [("ArtistId", columnField "ArtistId"), ("Name", columnField "Name")]
|
|
tableName = TableName "Artist"
|
|
in Query fields tableName Nothing Nothing Nothing Nothing
|
|
|
|
albumsQuery :: Query
|
|
albumsQuery =
|
|
let fields = KeyMap.fromList [("AlbumId", columnField "AlbumId"), ("ArtistId", columnField "ArtistId"), ("Title", columnField "Title")]
|
|
tableName = TableName "Album"
|
|
in Query fields tableName Nothing Nothing Nothing Nothing
|
|
|
|
columnField :: Text -> Field
|
|
columnField = ColumnField . ValueWrapper . ColumnName
|