Merge pull request #2 from mauhiz/feature/sl-abs-cls

Also support abstract sealed classes

Author: lloydmeta

README.md

@@ -1,7 +1,7 @@
 # Enumeratum [![Build Status](https://travis-ci.org/lloydmeta/enumeratum.svg)](https://travis-ci.org/lloydmeta/enumeratum) [![Coverage Status](https://coveralls.io/repos/lloydmeta/enumeratum/badge.png)](https://coveralls.io/r/lloydmeta/enumeratum)
 
 Yet another enumeration implementation for Scala for the sake of exhaustive pattern match warnings, Enumeratum is
-an implementation based on a single Scala macro that searches for implementations of a sealed trait.
+an implementation based on a single Scala macro that searches for implementations of a sealed trait or class.
 
 Enumeratum aims to be similar enough to Scala's built in `Enumeration` to be easy-to-use and understand while offering
 more flexibility and power.
@@ -18,7 +18,7 @@ libraryDependencies ++= Seq(
 
 ## How-to + example
 
-Using Enumeratum is simple. Simply declare your own sealed trait `A`, and implement it as case objects inside
+Using Enumeratum is simple. Simply declare your own sealed trait or class `A`, and implement it as case objects inside
 an object that extends from `Enum[A]` as follows.
 
 *Note* `Enum` is BYOO (Bring Your Own Ordinality) - take care of ordinality in your own way when you implement 

macros/src/main/scala/enumeratum/EnumMacros.scala

@@ -14,10 +14,10 @@ object EnumMacros {
   }
 
   private def validateType(c: Context)(typeSymbol: c.universe.Symbol): Unit = {
-    if (!typeSymbol.asClass.isTrait || !typeSymbol.asClass.isSealed)
+    if (!typeSymbol.asClass.isSealed)
       c.abort(
         c.enclosingPosition,
-        "You can only use findValues on sealed traits"
+        "You can only use findValues on sealed traits or classes"
       )
   }
 

src/test/scala/enumeratum/EnumSpec.scala

@@ -34,7 +34,7 @@ class EnumSpec extends FunSpec with Matchers {
 
   }
 
-  describe("when wrapped in another object") {
+  describe("when a sealed trait is wrapped in another object") {
 
     import Wrapper._
     import Wrapper.SmartEnum._
@@ -65,6 +65,36 @@ class EnumSpec extends FunSpec with Matchers {
 
   }
 
+  describe("when a sealed abstract class is wrapped in another object") {
+
+    import InTheWoods.Mushroom._
+
+    describe("#values") {
+
+      it("should contain objects") {
+        values shouldBe Set(FlyAgaric, LSD, Shimeji)
+      }
+
+    }
+
+    describe("#withName") {
+
+      it("should return the proper object when passed the proper string") {
+        withName("FlyAgaric") should be(FlyAgaric)
+        withName("LSD") should be(LSD)
+        withName("Shimeji") should be(Shimeji)
+      }
+
+      it("should throw an error otherwise") {
+        intercept[IllegalArgumentException] {
+          withName("hello")
+        }
+      }
+
+    }
+
+  }
+
   describe("trying to use with improper types") {
 
     it("should fail to compile for unsealed traits") {

src/test/scala/enumeratum/Models.scala

@@ -26,4 +26,18 @@ object Wrapper {
 
   }
 
+}
+
+object InTheWoods {
+  sealed abstract class Mushroom(val toxic: Boolean)
+
+  object Mushroom extends Enum[Mushroom] {
+
+    val values = findValues
+
+    case object FlyAgaric extends Mushroom(true)
+    case object LSD extends Mushroom(false)
+    case object Shimeji extends Mushroom(false)
+
+  }
 }