Equals method overridden. Test added

slicebox · Oct 31, 2018 · f56c0c2 · f56c0c2
1 parent 692ab8f
commit f56c0c2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/main/scala/se/nimsa/dicom/data/CharacterSets.scala b/src/main/scala/se/nimsa/dicom/data/CharacterSets.scala
@@ -73,6 +73,11 @@ class CharacterSets(val charsetNames: Seq[String]) {
   }
 
   override def toString: String = s"${getClass.getSimpleName} [${specifiedCharsets.map(_.charset.toString).mkString(",")}]"
+
+  override def equals(that: Any): Boolean = that match {
+    case thatCharSets: CharacterSets => charsetNames == thatCharSets.charsetNames
+    case _ => this == that
+  }
 }
 
 object CharacterSets {

diff --git a/src/test/scala/se/nimsa/dicom/data/CharacterSetsTest.scala b/src/test/scala/se/nimsa/dicom/data/CharacterSetsTest.scala
@@ -103,6 +103,12 @@ class CharacterSetsTest extends TestKit(ActorSystem("CharacterSetsSpec")) with F
     checkPatientName(name, nameCodePoints)
   }
 
+  "CharacterSets" should "be comparable" in {
+    val cs1 = new CharacterSets(Seq("ISO 2022 IR 13", "ISO 2022 IR 87"))
+    val cs2 = new CharacterSets(Seq("ISO 2022 IR 13", "ISO 2022 IR 87"))
+    cs1 shouldBe cs2
+  }
+
   private def checkPatientName(name: String, expectedCodePoints: Array[Int]): Unit = {
     val codePoints = new Array[Int](name.codePointCount(0, name.length))
     val length = name.length