-
-
Notifications
You must be signed in to change notification settings - Fork 608
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Make code generator tests usable from other slick-extensions:
slick-testkit now depends on slick-codegen and provides the TestCodeGenerator and TestCodeRunner types which provide the tools needed to generate and run code generator tests.
- Loading branch information
Showing
4 changed files
with
143 additions
and
111 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
134 changes: 134 additions & 0 deletions
134
slick-testkit/src/main/scala/com/typesafe/slick/testkit/util/TestCodeGenerator.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,134 @@ | ||
| package com.typesafe.slick.testkit.util | ||
|
|
||
| import scala.concurrent.Await | ||
| import scala.concurrent.duration.Duration | ||
| import scala.concurrent.ExecutionContext.Implicits.global | ||
| import scala.io.{Codec, Source} | ||
|
|
||
| import java.util.concurrent.ExecutionException | ||
|
|
||
| import slick.codegen.{OutputHelpers, SourceCodeGenerator} | ||
| import slick.dbio._ | ||
| import slick.model.Model | ||
|
|
||
| import org.junit.Test | ||
|
|
||
| trait TestCodeGenerator { | ||
| def packageName: String | ||
| def defaultTestCode(c: Config): String | ||
| def configurations: Seq[Config] | ||
| def computeFullTdbName(tdbName: String) = StandardTestDBs.getClass.getName.replaceAll("\\$", "") + "." + tdbName | ||
|
|
||
| def main(args: Array[String]): Unit = try { | ||
| val clns = configurations.flatMap(_.generate(args(0)).toSeq) | ||
| new OutputHelpers { | ||
| def indent(code: String): String = code | ||
| def code: String = "" | ||
| }.writeStringToFile( | ||
| s""" | ||
| |package $packageName | ||
| |object AllTests extends com.typesafe.slick.testkit.util.TestCodeRunner.AllTests { | ||
| | val clns = Seq(${clns.map("\"" + _ + "\"").mkString(", ")}) | ||
| |} | ||
| """.stripMargin, args(0), packageName, "AllTests.scala" | ||
| ) | ||
| } catch { case ex: Throwable => | ||
| ex.printStackTrace(System.err) | ||
| System.exit(1) | ||
| } | ||
|
|
||
| class Config(val objectName: String, val tdb: JdbcTestDB, tdbName: String, initScripts: Seq[String]) { self => | ||
| def useSingleLineStatements = false | ||
|
|
||
| def slickDriver = tdb.driver.getClass.getName.replaceAll("\\$", "") | ||
|
|
||
| def fullTdbName = computeFullTdbName(tdbName) | ||
|
|
||
| def generate(dir: String): Option[String] = if(tdb.isEnabled || tdb.isInstanceOf[InternalJdbcTestDB]) { | ||
| tdb.cleanUpBefore() | ||
| try { | ||
| var init: DBIO[Any] = DBIO.successful(()) | ||
| var current: String = null | ||
| initScripts.foreach { initScript => | ||
| import tdb.driver.api._ | ||
| Source.fromURL(self.getClass.getResource(initScript))(Codec.UTF8).getLines().foreach { s => | ||
| if(current eq null) current = s else current = current + "\n" + s | ||
| if(s.trim.endsWith(";")) { | ||
| if(useSingleLineStatements) { | ||
| current = current.substring(0, current.length-1) | ||
| current = current.replace("\r", "").replace('\n', ' ') | ||
| } | ||
| init = init >> sqlu"#$current" | ||
| current = null | ||
| } | ||
| } | ||
| if(current ne null) { | ||
| if(useSingleLineStatements) current = current.replace("\r", "").replace('\n', ' ') | ||
| init = init >> sqlu"#$current" | ||
| } | ||
| } | ||
| val db = tdb.createDB() | ||
| try { | ||
| val m = Await.result(db.run((init >> generator).withPinnedSession), Duration.Inf) | ||
| m.writeToFile(profile=slickDriver, folder=dir, pkg=packageName, objectName, fileName=objectName+".scala" ) | ||
| } finally db.close | ||
| } | ||
| finally tdb.cleanUpAfter() | ||
| Some(s"$packageName.$objectName") | ||
| } else None | ||
|
|
||
| def generator: DBIO[SourceCodeGenerator] = | ||
| tdb.driver.createModel(ignoreInvalidDefaults=false).map(new MyGen(_)) | ||
|
|
||
| def testCode: String = defaultTestCode(this) | ||
|
|
||
| class MyGen(model:Model) extends SourceCodeGenerator(model) { | ||
| override def entityName = sqlName => { | ||
| val baseName = super.entityName(sqlName) | ||
| if(baseName.dropRight(3).last == 's') baseName.dropRight(4) | ||
| else baseName | ||
| } | ||
| override def parentType = Some("com.typesafe.slick.testkit.util.TestCodeRunner.TestCase") | ||
| override def code = { | ||
| s""" | ||
| |lazy val tdb = $fullTdbName | ||
| |def test = { | ||
| | import org.junit.Assert._ | ||
| | import scala.concurrent.ExecutionContext.Implicits.global | ||
| | $testCode | ||
| |} | ||
| """.stripMargin + super.code | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| class TestCodeRunner(tests: TestCodeRunner.AllTests) { | ||
| def run(cln: String): Unit = { | ||
| val t = Class.forName(cln+"$").getField("MODULE$").get(null).asInstanceOf[TestCodeRunner.TestCase] | ||
| val tdb = t.tdb | ||
| println(s"Running test $cln on ${tdb.confName}") | ||
| if(tdb.isEnabled) { | ||
| tdb.cleanUpBefore() | ||
| try { | ||
| val a = t.test | ||
| val db = tdb.createDB() | ||
| try Await.result(db.run(a.withPinnedSession), Duration.Inf) | ||
| catch { case e: ExecutionException => throw e.getCause } | ||
| finally db.close() | ||
| } finally tdb.cleanUpAfter() | ||
| } else println("- Test database is disabled") | ||
| } | ||
|
|
||
| @Test def allTests = tests.clns.foreach(run) | ||
| } | ||
|
|
||
| object TestCodeRunner { | ||
| trait AllTests { | ||
| def clns: Seq[String] | ||
| } | ||
| trait TestCase { | ||
| def test: slick.dbio.DBIO[Any] | ||
| def tdb: JdbcTestDB | ||
| } | ||
| } |
34 changes: 3 additions & 31 deletions
34
slick-testkit/src/test/scala/slick/test/codegen/GeneratedCodeTest.scala
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters