implemented high-level logic for strFromConn

test/Network/Gemini/Capsule/InternalSpec.hs

@@ -38,6 +38,8 @@ spec = describe "Internal" $ do
   readURLSpec
   sendResponseSpec
   strFromConnSpec
+  readMaxSpec
+  stripCRLFSpec
 
 readURLSpec :: Spec
 readURLSpec = describe "readURL" $ mapM_
@@ -56,10 +58,10 @@ readURLSpec = describe "readURL" $ mapM_
   ]
 
   where
-    validConn   = mkConn "gemini://example.com/\r\n"
-    longConn    = mkConn longBS
-    tooLongConn = mkConn tooLongBS
-    gibConn     = mkConn "aosidjfwoeinboijwefr"
+    validConn   = mkInConn ["gemini://example.com/\r\n"]
+    longConn    = mkInConn [longBS]
+    tooLongConn = mkInConn [tooLongBS]
+    gibConn     = mkInConn ["aosidjfwoeinboijwefr"]
     longBS      = BS.pack (take 1024 bytes) <> "\r\n"
     tooLongBS   = BS.pack (take 1025 bytes) <> "\r\n"
     bytes       = BS.unpack prefix ++ repeat (fromIntegral $ ord 'A')
@@ -67,14 +69,32 @@ readURLSpec = describe "readURL" $ mapM_
     longExp     = validExp { gemPath = [longDir] }
     longDir     = replicate (1024 - BS.length prefix) 'A'
     prefix      = "gemini://example.com/"
-    mkConn bs   = do
-      s <- nullInput
-      unRead bs s
-      return sampleConnection { source = s }
 
 sendResponseSpec :: Spec
 sendResponseSpec = describe "sendResponse" $ return ()
 
+strFromConnSpec :: Spec
+strFromConnSpec = describe "strFromConn" $ mapM_
+  ( \(desc, maxLen, ioConn, expect) -> context desc $
+    xit ("should return " ++ show expect) $ do
+      conn <- ioConn
+      strFromConn maxLen conn `shouldReturn` expect
+  )
+
+  --  description,    max size, connection,                expected
+  [ ( "valid string", 100,      mkInConn ["foo\r\n"],      Just "foo"  )
+  , ( "long string",  5,        mkInConn ["too long\r\n"], Nothing     )
+  , ( "no CR/LF",     100,      mkInConn ["foo"],          Nothing     )
+  , ( "bad UTF-8",    100,      mkInConn ["foo\xff\r\n"],  Nothing     )
+  , ( "non-ASCII",    100,      mkInConn ["\xc3\xa9\r\n"], Just "\xe9" )
+  ]
+
+mkInConn :: [BS.ByteString] -> IO (Connection a)
+mkInConn bss = do
+  s <- nullInput
+  mapM_ (`unRead` s) (reverse bss)
+  return sampleConnection { source = s }
+
 sampleConnection :: Connection a
 sampleConnection = Connection
   { source        = undefined
@@ -83,7 +103,10 @@ sampleConnection = Connection
   , connExtraInfo = undefined
   }
 
-strFromConnSpec :: Spec
-strFromConnSpec = describe "strFromConn" $ return ()
+readMaxSpec :: Spec
+readMaxSpec = describe "readMax" $ return ()
+
+stripCRLFSpec :: Spec
+stripCRLFSpec = describe "stripCRLF" $ return ()
 
 --jl