aboutsummaryrefslogtreecommitdiff
path: root/tests/run/t2526.scala
blob: 4d567bbbf44a56d3540ce7d3f630c09ed4e1f8ae (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
/**
 * Checks that various foreach methods overridden in mutable.HashMap as part of ticket #2526
 * still work correctly.
 */
object Test {
  import collection._

  def main(args: Array[String]): Unit = {
    val m = new mutable.HashMap[String, String]

    /* Use non hash-based structure for verification */
    val keys = List("a", "b", "c", "d", "e")
    val valueSuffix = "value"
    val values = keys.map(_ + valueSuffix)
    val entries = keys.zip(values)

    for (k <- keys) m(k) = k + valueSuffix

    assertForeach(keys, m.keySet.iterator)
    assertForeach(keys, m.keysIterator)
    assertForeach(keys, m.keySet)

    assertForeach(values, m.values.iterator)
    assertForeach(values, m.valuesIterator)

    assertForeach(entries, m)
  }

  /* Checks foreach of `actual` goes over all the elements in `expected` */
  private def assertForeach[E](expected: Traversable[E], actual: Iterator[E]): Unit = {
    val notYetFound = new mutable.ArrayBuffer[E]() ++= expected
    actual.foreach { e =>
      assert(notYetFound.contains(e))
      notYetFound -= e
    }
    assert(notYetFound.size == 0, "mutable.HashMap.foreach should have iterated over: " + notYetFound)
  }

  /*
   * Checks foreach of `actual` goes over all the elements in `expected`
   * We duplicate the method above because there is no common inteface between Traversable and
   * Iterator and we want to avoid converting between collections to ensure that we test what
   * we mean to test.
   */
  private def assertForeach[E](expected: Traversable[E], actual: Traversable[E]): Unit = {
    val notYetFound = new mutable.ArrayBuffer[E]() ++= expected
    actual.foreach { e =>
      assert(notYetFound.contains(e))
      notYetFound -= e
    }
    assert(notYetFound.size == 0, "mutable.HashMap.foreach should have iterated over: " + notYetFound)
  }
}