From ee8d2b9c69dd7dd4d47e5d88f47150770a15129a Mon Sep 17 00:00:00 2001
From: Kazu Yamamoto <kazu@iij.ad.jp>
Date: Mon, 24 Oct 2011 15:57:04 +0900
Subject: Supporting server side and brushing up.

---
 TestProtocol.hs | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

(limited to 'TestProtocol.hs')

diff --git a/TestProtocol.hs b/TestProtocol.hs
index 3c78cb6..5e2abe7 100644
--- a/TestProtocol.hs
+++ b/TestProtocol.hs
@@ -2,17 +2,15 @@
 
 module TestProtocol where
 
+import Data.IP
 import Network.DNS
 import Network.DNS.Internal
-import Network.DNS.Query
-import Network.DNS.Response
-import Data.IP
 import Test.Framework (defaultMain, testGroup, Test)
 import Test.Framework.Providers.HUnit
 import Test.HUnit hiding (Test)
 
 tests :: [Test]
-tests = 
+tests =
   [ testGroup "Test case"
       [ testCase "QueryA" (test_Format testQueryA)
       , testCase "QueryAAAA" (test_Format testQueryAAAA)
@@ -155,8 +153,8 @@ test_Format fmt = do
     let (Right fmt') = result
     assertEqual "fail" fmt fmt'
   where
-    bs = composeDNSFormat fmt 
-    result = runDNSFormat_ bs
+    bs = encode fmt
+    result = decode bs
 
 main :: IO ()
 main = defaultMain tests
-- 
cgit v1.2.3