fixed tests for BBCode/Lexer.hs

1 files changed, 2 insertions, 1 deletions

diff --git a/bbcode/test/Text/BBCode/LexerSpec.hs b/bbcode/test/Text/BBCode/LexerSpec.hs
index df96d52..e225fab 100644
--- a/bbcode/test/Text/BBCode/LexerSpec.hs
+++ b/bbcode/test/Text/BBCode/LexerSpec.hs
@@ -11,7 +11,7 @@ import Test.QuickCheck.Instances
 import Text.BBCode.Lexer
 
 import Data.Text (Text)
-import qualified Data.Text as T (singleton, replace, last, null)
+import qualified Data.Text as T (singleton, replace, last, null, init)
 
 import Data.Monoid ((<>), mconcat, Endo(..))
 import Data.List (intersperse)
@@ -29,6 +29,7 @@ coToken (BBOpen t xs) = "[" <> escape [']'] t <> " " <> attrs <> " ]"
     attrs = mconcat . intersperse " " $ map attr xs
     attr (key, "") = escape ['=', ']', ' '] key
     attr (key, val) = escape ['=', ']', ' '] key <> "=\"" <> escape ['\"'] val <> "\""
+coToken (BBContained t xs) = (<> "/]") . T.init . coToken $ BBOpen t xs
 coToken (BBClose t) = "[/" <> escape [']'] t <> "]"
 coToken (BBStr t) = escape ['['] t