1 files changed, 51 insertions, 0 deletions
diff --git a/src/test/scala/xyz/driver/common/BaseSuite.scala b/src/test/scala/xyz/driver/common/BaseSuite.scala
new file mode 100644
index 0000000..b4c3d03
--- /dev/null
+++ b/src/test/scala/xyz/driver/common/BaseSuite.scala
@@ -0,0 +1,51 @@
+package xyz.driver.common
+
+import java.time.{LocalDateTime, ZoneId}
+
+import org.scalatest.FreeSpecLike
+import org.scalatest.concurrent.ScalaFutures
+import org.scalatest.time.{Millis, Span}
+import xyz.driver.common.db.{MysqlQueryBuilder, SearchFilterExpr, SqlContext, Transactions}
+import xyz.driver.common.domain.{Email, LongId, PasswordHash, User}
+import xyz.driver.common.error.UnexpectedFilterException
+import xyz.driver.common.utils.DiffUtils
+
+import scala.concurrent.ExecutionContext.Implicits._
+import scala.concurrent.Future
+
+trait BaseSuite extends FreeSpecLike with DiffUtils with ScalaFutures {
+
+  implicit val defaultPatience = PatienceConfig(timeout = Span(1000, Millis), interval = Span(20, Millis))
+  implicit val sqlContext = new MockSqlContext(global)
+
+  def sampleUser(role: User.Role,
+                 email: String = "test@example.com",
+                 password: String = "123") = User(
+    id = LongId(2001),
+    email = Email(email),
+    name = "Test",
+    role = role,
+    passwordHash = PasswordHash(password),
+    latestActivity = Some(LocalDateTime.now(ZoneId.of("Z"))),
+    deleted = None
+  )
+
+  def createMockQueryBuilder[T](isExpectedFilter: SearchFilterExpr => Boolean): MysqlQueryBuilder[T] = {
+    MockQueryBuilder[T] {
+      case (filter, _, _) if isExpectedFilter(filter) =>
+        Future.successful(Seq.empty)
+      case (filter, _, _) =>
+        Future.failed(new UnexpectedFilterException(s"Filter is unexpected: $filter"))
+    } {
+      case _ =>
+        Future.successful((0, Option.empty[LocalDateTime]))
+    }
+  }
+
+  def transactions = new Transactions {
+    override def run[T](f: (SqlContext) => T): Future[T] = {
+      Future(f(sqlContext))
+    }
+  }
+
+}