blob: 498ef6796f8c2bfb6ed831d2b8bfab3f499a0752 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
{-# LANGUAGE OverloadedStrings #-}
{-# OPTIONS_GHC -fno-warn-orphans #-}
module Network.KRPC.MessageSpec (spec) where
import Control.Applicative
import Data.ByteString.Lazy as BL
import Test.Hspec
import Test.QuickCheck
import Test.QuickCheck.Instances ()
import Data.BEncode as BE
import Network.KRPC.Message
instance Arbitrary ErrorCode where
arbitrary = arbitraryBoundedEnum
instance Arbitrary KError where
arbitrary = KError <$> arbitrary <*> arbitrary <*> arbitrary
instance Arbitrary KQuery where
arbitrary = KQuery <$> pure (BInteger 0) <*> arbitrary <*> arbitrary
instance Arbitrary KResponse where
-- TODO: Abitrary instance for ReflectedIP
arbitrary = KResponse <$> pure (BList []) <*> arbitrary <*> pure Nothing
instance Arbitrary KMessage where
arbitrary = frequency
[ (1, Q <$> arbitrary)
, (1, R <$> arbitrary)
, (1, E <$> arbitrary)
]
spec :: Spec
spec = do
describe "error message" $ do
it "properly bencoded (iso)" $ property $ \ ke ->
BE.decode (BL.toStrict (BE.encode ke)) `shouldBe` Right (ke :: KError)
it "properly bencoded" $ do
BE.decode "d1:eli201e23:A Generic Error Ocurrede1:t2:aa1:y1:ee"
`shouldBe` Right (KError GenericError "A Generic Error Ocurred" "aa")
BE.decode "d1:eli202e22:A Server Error Ocurrede1:t2:bb1:y1:ee"
`shouldBe` Right (KError ServerError "A Server Error Ocurred" "bb")
BE.decode "d1:eli203e24:A Protocol Error Ocurrede1:t2:cc1:y1:ee"
`shouldBe` Right (KError ProtocolError "A Protocol Error Ocurred" "cc")
BE.decode "d1:eli204e30:Attempt to call unknown methode1:t2:dd1:y1:ee"
`shouldBe` Right
(KError MethodUnknown "Attempt to call unknown method" "dd")
describe "query message" $ do
it "properly bencoded (iso)" $ property $ \ kq ->
BE.decode (BL.toStrict (BE.encode kq)) `shouldBe` Right (kq :: KQuery)
it "properly bencoded" $ do
BE.decode "d1:ale1:q4:ping1:t2:aa1:y1:qe" `shouldBe`
Right (KQuery (BList []) "ping" "aa")
describe "response message" $ do
it "properly bencoded (iso)" $ property $ \ kr ->
BE.decode (BL.toStrict (BE.encode kr)) `shouldBe` Right (kr :: KResponse)
it "properly bencoded" $ do
BE.decode "d1:rle1:t2:aa1:y1:re" `shouldBe`
Right (KResponse (BList []) "aa" Nothing)
describe "generic message" $ do
it "properly bencoded (iso)" $ property $ \ km ->
BE.decode (BL.toStrict (BE.encode km)) `shouldBe` Right (km :: KMessage)
|
