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
|
package xyz.driver.pdsuicommon
import java.io.{Closeable, PrintWriter}
import java.net.URL
import java.sql.Connection
import java.util.logging.Logger
import javax.sql.DataSource
import xyz.driver.pdsuicommon.db.SlickQueryBuilder.TableData
import xyz.driver.pdsuicommon.db._
import xyz.driver.pdsuicommon.http.HttpFetcher
import scala.concurrent.Future
class MockDataSource extends DataSource with Closeable {
override def getConnection: Connection = throw new NotImplementedError("MockDataSource.getConnection")
override def getConnection(username: String, password: String): Connection = {
throw new NotImplementedError(s"MockDataSource.getConnection($username, $password)")
}
override def close(): Unit = throw new NotImplementedError("MockDataSource.close")
override def setLogWriter(out: PrintWriter): Unit = throw new NotImplementedError("MockDataSource.setLogWriter")
override def getLoginTimeout: Int = throw new NotImplementedError("MockDataSource.getLoginTimeout")
override def setLoginTimeout(seconds: Int): Unit = throw new NotImplementedError("MockDataSource.setLoginTimeout")
override def getParentLogger: Logger = throw new NotImplementedError("MockDataSource.getParentLogger")
override def getLogWriter: PrintWriter = throw new NotImplementedError("MockDataSource.getLogWriter")
override def unwrap[T](iface: Class[T]): T = throw new NotImplementedError("MockDataSource.unwrap")
override def isWrapperFor(iface: Class[_]): Boolean = throw new NotImplementedError("MockDataSource.isWrapperFor")
}
class MockFactory()(implicit val sqlContext: PostgresContext) {
val MockHttpFetcher: HttpFetcher = { (url: URL) =>
Future.successful(Array.empty[Byte])
}
}
object MockQueryBuilder {
type MockRunnerIn = (SearchFilterExpr, Sorting, Option[Pagination])
type MockRunnerOut[T] = Future[Seq[T]]
type MockCountRunnerOut = SlickQueryBuilder.CountResult
def apply[T](matcher: PartialFunction[MockRunnerIn, MockRunnerOut[T]])(
countMatcher: PartialFunction[MockRunnerIn, MockCountRunnerOut])(
implicit context: PostgresContext): SlickQueryBuilder[T] = {
val runner: SlickQueryBuilder.Runner[T] = { parameters =>
matcher((parameters.filter, parameters.sorting, parameters.pagination))
}
val countRunner: SlickQueryBuilder.CountRunner = { parameters =>
countMatcher((parameters.filter, parameters.sorting, parameters.pagination))
}
val parameters = SlickPostgresQueryBuilderParameters(
databaseName = "test",
tableData = TableData("", None, Set.empty[String]),
links = Map.empty
)
new SlickPostgresQueryBuilder(parameters)(runner, countRunner)
}
}
|