Fix parser to return comment line as is

Author: schoettl

src/Parse.hs

@@ -390,8 +390,14 @@ line = (anyChar `manyTill` newline) <?> "rest of line"
 whitespacechar = oneOf " \t"
 whitespace = many whitespacechar
 whitespaceline = try (newline >> return "") <|> try (whitespacechar >> whitespacechar `manyTill` newlineoreof)
+
 -- a line beginning with optional whitespace and #, or beginning with one or more * (an org node)
-commentline = try ((many1 (char '*') <|> (whitespace >> many1 (char '#'))) >> lineoreof) <?> "comments"
+commentline = try (do
+  prefix <- many1 (char '*') <|> (whitespace >> many1 (char '#'))
+  rest <- lineoreof
+  return $ prefix ++ rest
+  ) <?> "comments"
+
 whitespaceorcommentline = commentline <|> whitespaceline
 whitespaceorcommentlineoreof = choice [eofasstr, commentline, whitespaceline]
 eofasstr = eof >> return ""

src/Print.hs

@@ -58,7 +58,7 @@ printShellTest format actualMode ShellTest{command=c,stdin=i,comments=comments,
      | otherwise = fail "Unsupported argument for --actual option. Allowed: all, update, or a prefix thereof."
 
 printComments :: [String] -> IO ()
-printComments = mapM_ (putStrLn . ("#"++)) -- TODO commentline discards leading space and comment character :(
+printComments = mapM_ putStrLn
 
 printStdin :: String -> Maybe String -> IO ()
 printStdin _ (Just "") = return ()