1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2018 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.api
21 
22 import com.github.t3hnar.bcrypt._
23 import javax.sql.DataSource
24 import org.apache.commons.dbcp2.BasicDataSource
25 import org.flywaydb.core.Flyway
26 import org.make.api.docker.DockerCockroachService
27 import org.make.api.extensions.MakeDBExecutionContextComponent
28 import scalikejdbc.{ConnectionPool, DataSourceConnectionPool, GlobalSettings, LoggingSQLAndTimeSettings}
29 
30 import scala.annotation.tailrec
31 import scala.jdk.CollectionConverters._
32 import scala.concurrent.ExecutionContext
33 import scala.util.{Failure, Success, Try}
34 import scala.concurrent.duration.DurationInt
35 
36 trait DatabaseTest extends MakeUnitTest with DockerCockroachService with MakeDBExecutionContextComponent {
37 
38   override val readExecutionContext: ExecutionContext = ExecutionContext.Implicits.global
39   override val writeExecutionContext: ExecutionContext = ExecutionContext.Implicits.global
40 
41   protected def databaseName: String = "makeapitest"
42 
43   protected def defaultClientId: String = "clientId"
44   protected def defaultClientSecret: String = "clientSecret"
45   protected def defaultBackofficeClientId: String = "backofficeClientId"
46   protected def defaultBackofficeClientSecret: String = "backofficeClientSecret"
47   protected def adminFirstName: String = "admin"
48   protected def adminEmail: String = "admin@example.com"
49   protected def adminPassword: String = "passpass".boundedBcrypt
50 
51   @SuppressWarnings(Array("org.wartremover.warts.ThreadSleep"))
52   override def beforeAll(): Unit = {
53     super.beforeAll()
54 
55     GlobalSettings.loggingSQLErrors = true
56     GlobalSettings.loggingSQLAndTime = LoggingSQLAndTimeSettings(
57       enabled = true,
58       warningEnabled = false,
59       printUnprocessedStackTrace = true,
60       logLevel = "info"
61     )
62 
63     val dataSource: DataSource = createDataSource
64 
65     logger.debug(s"Creating database with name: $databaseName")
66 
67     val connection = dataSource.getConnection
68     connection.prepareStatement(s"CREATE DATABASE $databaseName").execute()
69     connection.close()
70 
71     ConnectionPool.add("WRITE", new DataSourceConnectionPool(dataSource))
72     ConnectionPool.add("READ", new DataSourceConnectionPool(dataSource))
73 
74     Thread.sleep(1.second.toMillis)
75 
76     val flyway: Flyway = Flyway
77       .configure()
78       .dataSource(dataSource)
79       .initSql("SET autocommit_before_ddl = off")
80       .baselineOnMigrate(true)
81       .locations("classpath:db/migration", "classpath:org/make/api/migrations/db")
82       .placeholders(
83         Map(
84           "dbname" -> databaseName,
85           "clientId" -> defaultClientId,
86           "clientSecret" -> defaultClientSecret,
87           "backofficeClientId" -> defaultBackofficeClientId,
88           "backofficeClientSecret" -> defaultBackofficeClientSecret,
89           "adminEmail" -> adminEmail,
90           "adminFirstName" -> adminFirstName,
91           "adminEncryptedPassword" -> adminPassword.boundedBcrypt
92         ).asJava
93       )
94       .load()
95 
96     @SuppressWarnings(Array("org.wartremover.warts.Throw"))
97     @tailrec
98     def migrateFlyway(nRetries: Int = 3): Unit =
99       Try(flyway.migrate()) match {
100         case Success(_)                  =>
101         case Failure(e) if nRetries <= 0 => throw e
102         case _ =>
103           flyway.repair()
104           migrateFlyway(nRetries - 1)
105       }
106 
107     migrateFlyway()
108 
109     Try(flyway.validate()) match {
110       case Success(_) => logger.info("Database schema created")
111       case Failure(e) => logger.warn("Cannot migrate database:", e)
112     }
113 
114   }
115 
116   private def createDataSource: DataSource = {
117     val writeDataSource = new BasicDataSource()
118     writeDataSource.setDriverClassName("org.postgresql.Driver")
119     writeDataSource.setUrl(s"jdbc:postgresql://localhost:$cockroachExposedPort/$databaseName")
120     writeDataSource.setUsername("root")
121     writeDataSource
122   }
123 }