You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.6 KiB
45 lines
1.6 KiB
|
|
module Test.Packets.L3 |
|
( |
|
tests |
|
) where |
|
|
|
import Data.Binary (encode) |
|
import qualified Data.ByteString as B |
|
import Data.Word (Word8 (..)) |
|
import Hedgehog (MonadGen (..), forAll, property, (===)) |
|
import Hedgehog.Gen |
|
import qualified Hedgehog.Range as Range |
|
import Packets.L3 |
|
import Test.Tasty |
|
import Test.Tasty.Hedgehog |
|
import Test.Tasty.HUnit |
|
|
|
tests :: TestTree |
|
tests = testGroup "L3" [ unitTests] |
|
|
|
unitTests :: TestTree |
|
unitTests = testGroup "Unit tests" [ testIpChecksum |
|
, testSerializeEchoRequest ] |
|
|
|
testIpChecksum :: TestTree |
|
testIpChecksum = testCase "ipChecksum case 1" $ do |
|
let testIcmp = B.pack [ |
|
0x08, 0x00, 0x00, 0x00, 0xfc, 0x6b, 0x59, 0xfb, |
|
0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68] |
|
let checksum = ipChecksum testIcmp |
|
checksum @?= 0x1003 |
|
|
|
testSerializeEchoRequest :: TestTree |
|
testSerializeEchoRequest = testCase "serialize IcmpEchoRequest" $ do |
|
let payload = B.pack [ |
|
0x1c, 0x25, 0x31, 0x65, 0x00, 0x00, 0x00, 0x00, |
|
0x72, 0x51, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, |
|
0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f, 0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27, |
|
0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f, 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37 |
|
] |
|
let testIcmp = B.pack [ 0x08, 0x00, 0xd8, 0x7c, 0x9d, 0xd2, 0x00, 0x02] <> payload |
|
let icmp = IcmpEchoRequest (IcmpId 40402) (IcmpSqnum 2) payload |
|
(B.toStrict . encode) icmp @=? testIcmp |
|
|
|
|
|
|