com.flextrade.jfixture.JFixtureSugar.scala Maven / Gradle / Ivy

Go to download
package com.flextrade.jfixture

import java.lang.reflect.Type

import com.flextrade.jfixture.internal.ScalaRelays
import com.flextrade.jfixture.utility.Interceptor

import scala.reflect.{ClassTag, classTag}
import scala.reflect.runtime.universe._

trait JFixtureSugar {
  val defaultFixture: JFixture = new JFixture()
    .addBuilderToStartOfPipeline(ScalaRelays.collections)
    .addBuilderToStartOfPipeline(ScalaRelays.primitives)
    .addBuilderToStartOfPipeline(ScalaRelays.reflection)

  def fixture[T : ClassTag:TypeTag]: T = {
    defaultFixture.create(typeOf[T]).asInstanceOf[T]
  }

  /**
    * The following methods can be accessed when customising the fixture - e.g.,
    *     *   
    *     defaultFixture.customise().scalaLazyInstance[SomeCaseClass](SomeCaseClass(1,2,3))
    *   
    * 
    */
  implicit class FluentCustomisationScalaConversion(customisation: FluentCustomisation) {

    def scalaIntercept[T: ClassTag](interceptor: T => T): FluentCustomisation = {
      customisation.intercept(classTag[T].runtimeClass.asInstanceOf[Class[T]], new Interceptor[T] {
        override def intercept(instance: T): Unit = interceptor(instance)
      })
    }

    def scalaLazyInstance[T: ClassTag](instanceSupplier: => T): FluentCustomisation = {
      customisation.lazyInstance(classTag[T].runtimeClass.asInstanceOf[Type], new SpecimenSupplier[T] {
        override def create(): T = instanceSupplier
      })
    }

  }

}