better handling of booleans for prefix/suffix

lib/attr_enumerator.rb

@@ -22,8 +22,10 @@ def attr_enumerator(field, enumerators, opts={})
       generate_scopes = options.delete(:generate_scopes)
 
       prefix = options.delete(:prefix)
-      prefix = { true => field.to_s + '_', false => '' }[prefix] || prefix.to_s
-      suffix = options.delete(:suffix).to_s
+      prefix = {true => field.to_s + '_', false => ''}[prefix] || prefix.to_s
+
+      suffix = options.delete(:suffix)
+      suffix = {true => '', false => ''}[suffix] || suffix.to_s
 
       if generate_constant
         const_name = generate_constant == true ? field.to_s.pluralize.upcase : generate_constant.to_s

spec/attr_enumerator_spec.rb

@@ -86,6 +86,16 @@ class TestModel
         instance.should_not be_colored_blue
       end
 
+      it "should allow for explicitly using the default prefix" do
+        subject.attr_enumerator :choice, ['red', 'blue'], :prefix => true
+        instance.should respond_to :choice_red?
+      end
+
+      it "should allow for no prefix" do
+        subject.attr_enumerator :choice, ['red', 'blue'], :prefix => false
+        instance.should respond_to :red?
+      end
+
       it "should allow for a custom suffix" do
         subject.attr_enumerator :choice, ['first', 'second'], :prefix => '', :suffix => '_choice'
         instance.choice = 'first'
@@ -94,6 +104,16 @@ class TestModel
         instance.should be_first_choice
       end
 
+      it "should allow for explicitly using the default suffix" do
+        subject.attr_enumerator :choice, ['red', 'blue'], :suffix => true
+        instance.should respond_to :choice_red?
+      end
+
+      it "should allow for no suffix" do
+        subject.attr_enumerator :choice, ['red', 'blue'], :suffix => false
+        instance.should respond_to :choice_red?
+      end
+
       it "should generate methods with friendly names" do
         enumerations = {
           :has_space? => 'has space',