ast: codeblocks with single words are classes not IDs

Signed-off-by: Daniel Silverstone <dsilvers@digital-scurf.org>

1 files changed, 4 insertions, 3 deletions

diff --git a/src/ast.rs b/src/ast.rs
index 2033553..9543efd 100644
--- a/src/ast.rs
+++ b/src/ast.rs
@@ -228,8 +228,8 @@ fn parse_code_block_attrs(attrs: &str) -> Attr {
                 keyvalues.push((k.to_string(), v.to_string()));
             }
         }
-    } else {
-        id = attrs.to_string();
+    } else if !attrs.is_empty() {
+        classes.push(attrs.to_string());
     }
     (id, classes, keyvalues)
 }
@@ -355,9 +355,10 @@ mod test {
 
     #[test]
     fn code_block_attrs() {
+        assert_eq!(parse_code_block_attrs(""), ("".to_string(), vec![], vec![]));
         assert_eq!(
             parse_code_block_attrs("foo"),
-            ("foo".to_string(), vec![], vec![])
+            ("".to_string(), vec!["foo".to_string()], vec![])
         );
         assert_eq!(
             parse_code_block_attrs("{#foo}"),