MainSpec.hs

{-# LANGUAGE OverloadedStrings #-}
module Tshark.MainSpec (spec)
where
import Data.Maybe (fromJust)
import Distribution.Simple.Utils (TempFileOptions(..), withTempFileEx)
import MptcpAnalyzer.ArtificialFields
import MptcpAnalyzer.Pcap
import MptcpAnalyzer.Stream
import Net.IP
import Net.IPv4 (localhost)
import Net.Tcp.Connection
import System.Exit (ExitCode(ExitSuccess))
import System.IO
import Test.Hspec
import Test.QuickCheck hiding (Success)
import Tshark.Main



exampleTcpConnectionLocalhost :: TcpConnection
exampleTcpConnectionLocalhost = TcpConnection (fromIPv4 localhost) (fromIPv4 localhost) 24 42 (StreamId 0)

exampleTcpConnection0 :: TcpConnection
exampleTcpConnection0 = TcpConnection (fromJust $ decode "10.0.0.1") (fromJust $ decode "192.10.0.2") 24 42 (StreamId 1)

opts :: TempFileOptions
opts = TempFileOptions True

-- main :: IO ()
-- main = hspec $ do
spec :: Spec
spec =
  describe "tshark tests" $ do
    it "Generate bidirectional tshark filter" $
      genReadFilterFromTcpConnection exampleTcpConnectionLocalhost Nothing
        `shouldBe` "tcp and ip.addr==127.0.0.1 and ip.addr==127.0.0.1 and tcp.port==42 and tcp.port==24"
    it "Check the unidirectional filter" $
      genReadFilterFromTcpConnection exampleTcpConnection0 (Just RoleClient)
        `shouldBe` "tcp and ip.src==10.0.0.1 and ip.dst==192.10.0.2 and tcp.srcport==24 and tcp.dstport==42"
      -- exportToCsv "mptcpanalyzer/examples/client_2_filtered.pcapng"
    it "Tshark generates a proper CSV file" $ do
      -- cant find the profile so there is some herr being written
      withFile "/tmp/mptcp.csv" ReadWriteMode (exportToCsv defaultTsharkPrefs "examples/client_2_filtered.pcapng") >>= \x -> fst x `shouldBe` ( ExitSuccess)

      -- `shouldThrow`
    it "Test frame loading" $
      pendingWith "test"