Also implement must-use-type for enums.

1 files changed, 7 insertions, 1 deletions

diff --git a/src/codegen/mod.rs b/src/codegen/mod.rs
index 2522922a..b823fb34 100644
--- a/src/codegen/mod.rs
+++ b/src/codegen/mod.rs
@@ -2023,7 +2023,8 @@ impl CodeGenerator for CompInfo {
             attributes.push(attributes::derives(&derives))
         }
 
-        if item.annotations().must_use_type() || ctx.must_use_type_by_name(item) {
+        if item.annotations().must_use_type() || ctx.must_use_type_by_name(item)
+        {
             attributes.push(attributes::must_use());
         }
 
@@ -3002,6 +3003,11 @@ impl CodeGenerator for Enum {
             attrs.push(attributes::doc(comment));
         }
 
+        if item.annotations().must_use_type() || ctx.must_use_type_by_name(item)
+        {
+            attrs.push(attributes::must_use());
+        }
+
         if !variation.is_const() {
             let mut derives = derives_of_item(item, ctx);
             // For backwards compat, enums always derive Clone/Eq/PartialEq/Hash, even