Add clause for stripping integers

#10
zotonic · May 10, 2014 · a1f8d44 · a1f8d44
1 parent 914b7f6
commit a1f8d44
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/z_html.erl b/src/z_html.erl
@@ -349,7 +349,9 @@ strip([]) ->
 strip(Html) when is_binary(Html) ->
     strip(Html, in_text, <<>>);
 strip(L) when is_list(L) ->
-    strip(list_to_binary(L)).
+    strip(list_to_binary(L));
+strip(N) when is_integer(N) ->
+    strip(integer_to_list(N)).
 
 strip(<<>>, _, Acc) -> Acc;
 strip(<<$<,T/binary>>, in_text, Acc) ->

diff --git a/test/z_html_test.erl b/test/z_html_test.erl
@@ -25,6 +25,10 @@ ensure_check_test() ->
 
 strip_test() ->
 	?assertEqual(<<"Hello">>, z_html:strip(<<"<p class='hello'>Hello</p>">>)),
+	?assertEqual(<<"Hello">>, z_html:strip(<<"Hello">>)),
+	?assertEqual(<<"">>, z_html:strip(<<"">>)),
+	?assertEqual([], z_html:strip([])),
+	?assertEqual(<<"1234">>, z_html:strip(1234)),
 	ok.
 
 abs_links_test() ->