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.docker
21 
22 import com.whisk.docker.{DockerContainer, DockerReadyChecker}
23 import org.scalatest.Suite
24 
25 import scala.concurrent.duration.DurationInt
26 
27 trait DockerElasticsearchService extends DockerBaseTest {
28   self: Suite =>
29 
30   val defaultElasticsearchHttpPort = 9200
31   val defaultElasticsearchClientPort = 9300
32 
33   // Port range: [30000-30999]
34   def elasticsearchExposedPort: Int
35 
36   val defaultElasticsearchProposalIndex = "proposals"
37   val defaultElasticsearchProposalDocType = "proposal"
38 
39   private def elasticSearchContainer =
40     DockerContainer(
41       "docker.elastic.co/elasticsearch/elasticsearch:8.18.0",
42       name = Some(s"${getClass.getSimpleName}-elasticsearch")
43     ).withEntrypoint(
44         "/bin/sh",
45         "-c",
46         "./bin/elasticsearch-plugin list | grep -q analysis-icu || ./bin/elasticsearch-plugin install analysis-icu; ./bin/elasticsearch-plugin list | grep -q analysis-stempel || ./bin/elasticsearch-plugin install analysis-stempel; /usr/local/bin/docker-entrypoint.sh"
47       )
48       .withPorts(defaultElasticsearchHttpPort -> Some(elasticsearchExposedPort))
49       .withEnv(
50         "xpack.security.enabled=false",
51         "transport.host=localhost",
52         "ES_JVM_OPTIONS=-Xmx256M -Xms256M",
53         "ES_JAVA_OPTS=-Xmx256M -Xms256M"
54       )
55       .withReadyChecker(
56         DockerReadyChecker
57           .HttpResponseCode(defaultElasticsearchHttpPort, "/")
58           .within(100.millis)
59           .looped(300, 1.second)
60       )
61 
62   abstract override def dockerContainers: List[DockerContainer] =
63     elasticSearchContainer :: super.dockerContainers
64 }