~ make info hash test meaningful

1 files changed, 16 insertions, 29 deletions

diff --git a/tests/info-hash.hs b/tests/info-hash.hs
index 1e511e43..07c01c9c 100644
--- a/tests/info-hash.hs
+++ b/tests/info-hash.hs
@@ -1,49 +1,36 @@
 {-# LANGUAGE OverloadedStrings #-}
 module Main (main) where
 
-import Control.Applicative
 import Data.Foldable
-import Data.Maybe
-import qualified Data.Map as M
 import           Data.ByteString (ByteString)
 import qualified Data.ByteString as B
+import qualified Data.ByteString.Char8 as BC
 import qualified Data.ByteString.Lazy as L
 import qualified Data.ByteString.Builder as B
 import qualified Data.ByteString.Builder.Prim as B
-
-import System.Environment
-
 import Data.Torrent
-import Data.BEncode
-
+import System.Environment
+import System.Exit
 
-checkInfo :: B.ByteString
+checkInfo :: ByteString
 checkInfo = "0221caf96aa3cb94f0f58d458e78b0fc344ad8bf"
 
--- | _should_ be 'id' if content is correct
-reencode :: B.ByteString -> Result L.ByteString
-reencode content = (encode . toBEncode . (`asTypeOf` (undefined :: Torrent)))
-            `fmap` (fromBEncode =<< decode content)
-
 ppHex :: B.ByteString -> B.ByteString
 ppHex = L.toStrict . B.toLazyByteString . foldMap (B.primFixed B.word8HexFixed) . B.unpack
 
-chunk :: Int -> B.ByteString -> [B.ByteString]
-chunk size b | B.length b == 0 = [b]
-             | otherwise       =
-               let (x, xs) = B.splitAt size b
-               in x : chunk size xs
-
-showInfos :: ByteString -> (TorrentInfo, BEncode)
-showInfos bs =
-  let Right (be@(BDict dict)) = decode bs
-      Right t = tInfo <$> (fromBEncode be)
-      orig = BDict $ let BDict info = fromJust (M.lookup "info" dict)
-                     in M.insert "pieces" (BString "") info
-  in (t { tPieces = "" }, orig)
+torrentFileName :: String
+torrentFileName = "tests/dapper-dvd-amd64.iso.torrent"
 
 main :: IO ()
 main = do
-  [path] <- getArgs
+  args <- getArgs
+  let path = if length args == 0 then torrentFileName else head args
+
   Right t <- fromFile path
-  B.putStrLn (ppHex (tInfoHash t))
+
+  BC.putStr "info hash: "
+  BC.putStrLn (ppHex (tInfoHash t))
+
+  let passed = checkInfo == ppHex (tInfoHash t)
+  print passed
+  if passed then exitSuccess else exitFailure