+{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE RecordWildCards #-}
-{-# LANGUAGE StandaloneDeriving #-}
{-# LANGUAGE TemplateHaskell #-}
{-# LANGUAGE TypeFamilies #-}
-- real meat.
--
module TSN.XML.InjuriesDetail (
+ dtd,
pickle_message,
-- * Tests
injuries_detail_tests,
import Control.Monad ( forM_ )
import Data.Time ( UTCTime )
import Data.Tuple.Curry ( uncurryN )
+import qualified Data.Vector.HFixed as H ( HVector, cons, convert, tail )
+import qualified Data.Vector.HFixed.Cont as H (ContVec)
import Database.Groundhog (
DefaultKey,
- migrate )
+ countAll,
+ deleteAll,
+ migrate,
+ runMigration,
+ silentMigrationLogger )
+import Database.Groundhog.Generic ( runDbConn )
+import Database.Groundhog.Sqlite ( withSqliteConn )
import Database.Groundhog.TH (
groundhog,
mkPersist )
+import qualified GHC.Generics as GHC ( Generic )
import Test.Tasty ( TestTree, testGroup )
import Test.Tasty.HUnit ( (@?=), testCase )
import Text.XML.HXT.Core (
import TSN.Picklers( xp_date, xp_time_stamp )
import TSN.XmlImport ( XmlImport(..), XmlImportFk(..) )
import Xml (
+ Child(..),
FromXml(..),
FromXmlFk(..),
ToDb(..),
pickle_unpickle,
- unpickleable )
+ unpickleable,
+ unsafe_unpickle )
+
+
+
+-- | The DTD to which this module corresponds. Used to invoke dbimport.
+--
+dtd :: String
+dtd = "Injuries_Detail_XML.dtd"
--
xml_sport :: String,
xml_listings :: [InjuriesDetailListingXml],
xml_time_stamp :: UTCTime }
- deriving (Eq, Show)
+ deriving (Eq, GHC.Generic, Show)
+
+
+-- | For 'H.convert'.
+--
+instance H.HVector Message
+
-- | Database representation of a 'Message'.
--
xml_full_name :: String, -- ^ Team full name
xml_player_listings :: [InjuriesDetailListingPlayerListingXml] }
- deriving (Eq, Show)
+ deriving (Eq, GHC.Generic, Show)
+
+
+-- | For 'H.convert'.
+--
+instance H.HVector InjuriesDetailListingXml
+
instance ToDb InjuriesDetailListingXml where
-- | The database analogue of an 'InjuriesDetailListingXml' is a
-- 'InjuriesDetailListing'.
type Db InjuriesDetailListingXml = InjuriesDetailListing
-instance FromXmlFk InjuriesDetailListingXml where
+
+instance Child InjuriesDetailListingXml where
-- | Each 'InjuriesDetailListingXml' is contained in an
-- 'InjuriesDetail'.
type Parent InjuriesDetailListingXml = InjuriesDetail
+
+instance FromXmlFk InjuriesDetailListingXml where
-- | Construct a 'InjuriesDetailListing' from a
-- 'InjuriesDetailListingXml' and a foreign key to a
-- 'InjuriesDetail'.
-- * InjuriesDetailListingPlayerListing
-- | XML representation of a \<PlayerListing\>, the main type of
--- element contains in Injuries_Detail_XML messages.
+-- element contains in Injuries_Detail_XML messages. The leading
+-- underscores prevent unused field warnings.
--
data InjuriesDetailListingPlayerListingXml =
InjuriesDetailListingPlayerListingXml {
- xml_player_team_id :: String, -- ^ TeamIDs are (apparently) three
+ _xml_player_team_id :: String, -- ^ TeamIDs are (apparently) three
-- characters long and not
-- necessarily numeric. Postgres
-- imposes no performance penalty
-- bound of three characters.
-- We add the \"player\" to avoid conflict
-- with 'InjuriesDetailListingXml'.
- xml_player_id :: Int,
- xml_date :: UTCTime,
- xml_pos :: String,
- xml_name :: String,
- xml_injury :: String,
- xml_status :: String,
- xml_fantasy :: Maybe String, -- ^ Nobody knows what this is.
- xml_injured :: Bool,
- xml_type :: String }
- deriving (Eq, Show)
-
+ _xml_player_id :: Int,
+ _xml_date :: UTCTime,
+ _xml_pos :: String,
+ _xml_name :: String,
+ _xml_injury :: String,
+ _xml_status :: String,
+ _xml_fantasy :: Maybe String, -- ^ Nobody knows what this is.
+ _xml_injured :: Bool,
+ _xml_type :: String }
+ deriving (Eq, GHC.Generic, Show)
+
+
+-- | For 'H.convert'.
+--
+instance H.HVector InjuriesDetailListingPlayerListingXml
-- | Database representation of a
-- 'InjuriesDetailListingPlayerListingXml'. We drop the team_id
--- because it's redundant.
+-- because it's redundant. The leading underscores prevent unused
+-- field warnings.
--
data InjuriesDetailListingPlayerListing =
InjuriesDetailListingPlayerListing {
- db_injuries_detail_listings_id :: DefaultKey InjuriesDetailListing,
- db_player_id :: Int,
- db_date :: UTCTime,
- db_pos :: String,
- db_name :: String,
- db_injury :: String,
- db_status :: String,
- db_fantasy :: Maybe String, -- ^ Nobody knows what this is.
- db_injured :: Bool,
- db_type :: String }
+ _db_injuries_detail_listings_id :: DefaultKey InjuriesDetailListing,
+ _db_player_id :: Int,
+ _db_date :: UTCTime,
+ _db_pos :: String,
+ _db_name :: String,
+ _db_injury :: String,
+ _db_status :: String,
+ _db_fantasy :: Maybe String, -- ^ Nobody knows what this is.
+ _db_injured :: Bool,
+ _db_type :: String }
+ deriving ( GHC.Generic )
+
+-- | For 'H.cons', 'H.tail', etc.
+--
+instance H.HVector InjuriesDetailListingPlayerListing
instance ToDb InjuriesDetailListingPlayerListingXml where
type Db InjuriesDetailListingPlayerListingXml =
InjuriesDetailListingPlayerListing
-instance FromXmlFk InjuriesDetailListingPlayerListingXml where
+
+instance Child InjuriesDetailListingPlayerListingXml where
-- | Each 'InjuriesDetailListingPlayerListingXml' is contained in an
-- 'InjuriesDetailListing'.
--
type Parent InjuriesDetailListingPlayerListingXml = InjuriesDetailListing
+
+instance FromXmlFk InjuriesDetailListingPlayerListingXml where
-- | To construct a 'InjuriesDetailListingPlayerListing' from a
-- 'InjuriesDetailListingPlayerListingXml' we need to supply a
-- foreign key to an 'InjuriesDetailListing'.
--
- from_xml_fk fk InjuriesDetailListingPlayerListingXml{..} =
- InjuriesDetailListingPlayerListing {
- db_injuries_detail_listings_id = fk,
- db_player_id = xml_player_id,
- db_date = xml_date,
- db_pos = xml_pos,
- db_name = xml_name,
- db_injury = xml_injury,
- db_status = xml_status,
- db_fantasy = xml_fantasy,
- db_injured = xml_injured,
- db_type = xml_type }
+ from_xml_fk fk = (H.cons fk) . asCont . H.tail
+ where
+ -- Should be in the library soon.
+ asCont :: H.ContVec a -> H.ContVec a
+ asCont = id
+
-- | This lets us insert the XML representation
-- 'InjuriesDetailListingPlayerListingXml' directly.
constructors:
- name: InjuriesDetailListingPlayerListing
fields:
- - name: db_injuries_detail_listings_id
+ - name: _db_injuries_detail_listings_id
reference:
onDelete: cascade
|]
pickle_player_listing :: PU InjuriesDetailListingPlayerListingXml
pickle_player_listing =
xpElem "PlayerListing" $
- xpWrap (from_tuple, to_tuple) $
+ xpWrap (from_tuple, H.convert) $
xp10Tuple (xpElem "TeamID" xpText)
(xpElem "PlayerID" xpInt)
(xpElem "Date" xp_date)
(xpElem "Type" xpText)
where
from_tuple = uncurryN InjuriesDetailListingPlayerListingXml
- to_tuple pl = (xml_player_team_id pl,
- xml_player_id pl,
- xml_date pl,
- xml_pos pl,
- xml_name pl,
- xml_injury pl,
- xml_status pl,
- xml_fantasy pl,
- xml_injured pl,
- xml_type pl)
-- | Convert 'Listing's to/from XML.
pickle_listing :: PU InjuriesDetailListingXml
pickle_listing =
xpElem "Listing" $
- xpWrap (from_tuple, to_tuple) $
+ xpWrap (from_tuple, H.convert) $
xpTriple (xpElem "TeamID" xpText)
(xpElem "FullName" xpText)
(xpList pickle_player_listing)
where
from_tuple = uncurryN InjuriesDetailListingXml
- to_tuple l = (xml_team_id l,
- xml_full_name l,
- xml_player_listings l)
-- | Convert 'Message's to/from XML.
pickle_message :: PU Message
pickle_message =
xpElem "message" $
- xpWrap (from_tuple, to_tuple) $
+ xpWrap (from_tuple, H.convert) $
xp6Tuple (xpElem "XML_File_ID" xpInt)
(xpElem "heading" xpText)
(xpElem "category" xpText)
(xpElem "time_stamp" xp_time_stamp)
where
from_tuple = uncurryN Message
- to_tuple m = (xml_xml_file_id m,
- xml_heading m,
- xml_category m,
- xml_sport m,
- xml_listings m,
- xml_time_stamp m)
--
injuries_detail_tests =
testGroup
"InjuriesDetail tests"
- [ test_pickle_of_unpickle_is_identity,
+ [ test_on_delete_cascade,
+ test_pickle_of_unpickle_is_identity,
test_unpickle_succeeds ]
actual <- unpickleable path pickle_message
let expected = True
actual @?= expected
+
+
+-- | Make sure everything gets deleted when we delete the top-level
+-- record.
+--
+test_on_delete_cascade :: TestTree
+test_on_delete_cascade = testGroup "cascading delete tests"
+ [ check "delete of injuries_detail deletes its children"
+ "test/xml/Injuries_Detail_XML.xml",
+
+ check "delete of injuries_detail deletes its children (non-int team_id)"
+ "test/xml/Injuries_Detail_XML-noninteger-team-id.xml" ]
+ where
+ check desc path = testCase desc $ do
+ inj <- unsafe_unpickle path pickle_message
+ let a = undefined :: InjuriesDetail
+ let b = undefined :: InjuriesDetailListing
+ let c = undefined :: InjuriesDetailListingPlayerListing
+ actual <- withSqliteConn ":memory:" $ runDbConn $ do
+ runMigration silentMigrationLogger $ do
+ migrate a
+ migrate b
+ migrate c
+ _ <- dbimport inj
+ deleteAll a
+ count_a <- countAll a
+ count_b <- countAll b
+ count_c <- countAll c
+ return $ count_a + count_b + count_c
+ let expected = 0
+ actual @?= expected
projects / dead / htsn-import.git / blobdiff