aboutsummaryrefslogtreecommitdiff
path: root/shared/src/test/scala/FieldNameTests.scala
blob: f1b76a971789fe2d04043fa633fac4558a4be230 (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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package spray.json

import org.scalatest._

trait SnakeCaseFormats { self: DerivedFormats =>
  override def extractFieldName(paramName: String) =
    FieldNaming.substituteCamel(paramName, '_')
}
trait KebabCaseFormats { self: DerivedFormats =>
  override def extractFieldName(paramName: String) =
    FieldNaming.substituteCamel(paramName, '-')
}

object FieldNaming {

  @inline final private def isLower(ch: Char): Boolean =
    ((ch & 0x20) != 0) || (ch == '_')

  @inline def substituteCamel(paramName: String, substitute: Char) = {
    val length = paramName.length
    val builder = new StringBuilder(length)
    var i = 0
    while (i < length) {
      val cur = paramName(i)
      if (isLower(cur) && i + 1 < length) {
        builder.append(cur)
        val next = paramName(i + 1)
        if (!isLower(next)) {
          builder.append(substitute)
          builder.append((next ^ 0x20).toChar)
        } else {
          builder.append(next)
        }
        i += 1
      } else {
        builder.append((cur ^ 0x20).toChar)
      }
      i += 1
    }
    builder.result()
  }

}

class FieldNameTests extends FlatSpec with FormatTests {

  case class A(camelCASE: String, `__a_aB__`: Int, `a-a_B`: Int)
  case class B(camelCaseA: A)

  trait All extends DefaultJsonProtocol with DerivedFormats {
    implicit val bFormat = jsonFormat[B]
  }

  {
    object Protocol extends All with SnakeCaseFormats
    import Protocol._
    "snake_case" should behave like checkRoundtrip(
      B(A("helloWorld", 0, 0)),
      """{"camel_case_a":{"camel_case":"helloWorld","__a_a_b__":0,"a-a_b":0}}"""
    )
  }

  {
    object Protocol extends All with KebabCaseFormats
    import Protocol._
    "kebab-case" should behave like checkRoundtrip(
      B(A("helloWorld", 0, 0)),
      """{"camel-case-a":{"camel-case":"helloWorld","__a_a-b__":0,"a-a_b":0}}"""
    )
  }

}