test/junit/scala/PartialFunctionSerializationTest.scala


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

package scala

import org.junit.Test
import org.junit.Assert._
import org.junit.runner.RunWith
import org.junit.runners.JUnit4

@RunWith(classOf[JUnit4])
class PartialFunctionSerializationTest {
  val pf1: PartialFunction[Int, Int] = { case n if n > 0 => 1 }
  val pf2: PartialFunction[Int, Int] = { case n if n <= 0 => 2 }


  private def assertSerializable[A,B](fn: A => B): Unit = {
    import java.io._
    new ObjectOutputStream(new ByteArrayOutputStream()).writeObject(fn)
  }

  @Test def canSerializeLiteral = assertSerializable(pf1)

  @Test def canSerializeLifted = assertSerializable(pf1.lift)

  @Test def canSerializeOrElse = assertSerializable(pf1 orElse pf2)

  @Test def canSerializeUnlifted = assertSerializable(Function.unlift((x: Int) => Some(x)))

  @Test def canSerializeAndThen = assertSerializable(pf1.andThen((x: Int) => x))

  @Test def canSerializeEmpty = assertSerializable(PartialFunction.empty)
}