---

yaml
---
r: 13807
b: refs/heads/try
c: 95a006c
h: refs/heads/master
i:
  13805: 08a75e8
  13803: 8087f6a
  13799: d5922f5
  13791: 3f6d19b
v: v3

Author: brson

[refs]

@@ -2,5 +2,5 @@
 refs/heads/master: 61b1875c16de39c166b0f4d54bba19f9c6777d1a
 refs/heads/snap-stage1: e33de59e47c5076a89eadeb38f4934f58a3618a6
 refs/heads/snap-stage3: 4a81779abd786ff22d71434c6d9a5917ea4cdfff
-refs/heads/try: a25bc195e25fcce97301b907e2a03501daafb9b9
+refs/heads/try: 95a006ce520c838fb1368471bcfba3042e176400
 refs/tags/release-0.1: 1f5c5126e96c79d22cb7862f75304136e204f105

branches/try/src/rustdoc/attr_pass.rs

@@ -23,7 +23,8 @@ fn run(
         fold_crate: fold_crate,
         fold_mod: fold_mod,
         fold_fn: fold_fn,
-        fold_const: fold_const
+        fold_const: fold_const,
+        fold_enum: fold_enum
         with *fold::default_seq_fold(srv)
     });
     fold.fold_crate(fold, doc)
@@ -249,3 +250,60 @@ fn fold_const_should_extract_docs() {
     assert doc.brief == some("foo");
     assert doc.desc == some("bar");
 }
+
+fn fold_enum(
+    fold: fold::fold<astsrv::srv>,
+    doc: doc::enumdoc
+) -> doc::enumdoc {
+    let srv = fold.ctxt;
+    let attrs = parse_item_attrs(srv, doc.id, attr_parser::parse_enum);
+
+    ~{
+        brief: attrs.brief,
+        desc: attrs.desc,
+        variants: vec::map(doc.variants) {|variant|
+            let attrs = astsrv::exec(srv) {|ctxt|
+                alt ctxt.map.get(doc.id) {
+                  ast_map::node_item(@{
+                    node: ast::item_enum(ast_variants, _), _
+                  }) {
+                    let ast_variant = option::get(
+                        vec::find(ast_variants) {|v|
+                            v.node.name == variant.name
+                        });
+
+                    attr_parser::parse_variant(ast_variant.node.attrs)
+                  }
+                }
+            };
+
+            ~{
+                desc: attrs.desc
+                with *variant
+            }
+        }
+        with *doc
+    }
+}
+
+#[test]
+fn fold_enum_should_extract_docs() {
+    let source = "#[doc(brief = \"a\", desc = \"b\")]\
+                  enum a { v }";
+    let srv = astsrv::mk_srv_from_str(source);
+    let doc = extract::from_srv(srv, "");
+    let fold = fold::default_seq_fold(srv);
+    let doc = fold_enum(fold, doc.topmod.enums[0]);
+    assert doc.brief == some("a");
+    assert doc.desc == some("b");
+}
+
+#[test]
+fn fold_enum_should_extract_variant_docs() {
+    let source = "enum a { #[doc = \"c\"] v }";
+    let srv = astsrv::mk_srv_from_str(source);
+    let doc = extract::from_srv(srv, "");
+    let fold = fold::default_seq_fold(srv);
+    let doc = fold_enum(fold, doc.topmod.enums[0]);
+    assert doc.variants[0].desc == some("c");
+}