Use TestData in tracker spec

1 files changed, 10 insertions, 0 deletions

diff --git a/tests/Network/BitTorrent/Tracker/TestData.hs b/tests/Network/BitTorrent/Tracker/TestData.hs
index 5d9718e5..85544b91 100644
--- a/tests/Network/BitTorrent/Tracker/TestData.hs
+++ b/tests/Network/BitTorrent/Tracker/TestData.hs
@@ -1,6 +1,9 @@
+{-# LANGUAGE RecordWildCards #-}
 {-# OPTIONS_GHC -fno-warn-orphans #-}
 module Network.BitTorrent.Tracker.TestData
        ( TrackerEntry (..)
+       , isUdpTracker
+       , isHttpTracker
        , trackers
        ) where
 
@@ -29,6 +32,13 @@ data TrackerEntry = TrackerEntry
   , hashList    :: Maybe [InfoHash]
   }
 
+isUdpTracker :: TrackerEntry -> Bool
+isUdpTracker TrackerEntry {..} = uriScheme trackerURI == "udp:"
+
+isHttpTracker :: TrackerEntry -> Bool
+isHttpTracker TrackerEntry {..} = uriScheme trackerURI == "http:"
+                               || uriScheme trackerURI == "https:"
+
 instance IsString URI where
   fromString str = fromMaybe err $ parseURI str
     where