How to Enable Local and Automatic Testing of Atlas Search-Based Features

1
git clone git@github.com:mongodb-developer/atlas-search-local-testing.git

1
version: "2"
2
3
services:
4
    mongod:
5
        container_name: mongod
6
        image: mongodb/mongodb-enterprise-server:7.0-ubi8
7
        entrypoint: "/bin/sh -c \"echo \"$$KEYFILECONTENTS\" > \"$$KEYFILE\"\n\nchmod 400 \"$$KEYFILE\"\n\n\npython3 /usr/local/bin/docker-entrypoint.py mongod --transitionToAuth --keyFile \"$$KEYFILE\" --replSet \"$$REPLSETNAME\" --setParameter \"mongotHost=$$MONGOTHOST\" --setParameter \"searchIndexManagementHostAndPort=$$MONGOTHOST\"\""
8
        environment:
9
            MONGOTHOST: 10.6.0.6:27027
10
            KEYFILE: /data/db/keyfile
11
            KEYFILECONTENTS: sup3rs3cr3tk3y
12
            REPLSETNAME: local
13
        ports:
14
            - 27017:27017
15
        networks:
16
            network:
17
                ipv4_address: 10.6.0.5
18
    mongot:
19
        container_name: mongot
20
        image: mongodb/mongodb-atlas-search:preview
21
        entrypoint: "/bin/sh -c \"echo \"$$KEYFILECONTENTS\" > \"$$KEYFILE\"\n\n/etc/mongot-localdev/mongot  --mongodHostAndPort \"$$MONGOD_HOST_AND_PORT\" --keyFile \"$$KEYFILE\"\""
22
        environment:
23
            MONGOD_HOST_AND_PORT: 10.6.0.5:27017
24
            KEYFILE: /var/lib/mongot/keyfile
25
            KEYFILECONTENTS: sup3rs3cr3tk3y
26
        ports:
27
            - 27027:27027
28
        networks:
29
            network:
30
                ipv4_address: 10.6.0.6
31
networks:
32
  network:
33
    driver: bridge
34
    ipam:
35
      config:
36
        - subnet: 10.6.0.0/16
37
          gateway: 10.6.0.1

1
export default async () => {
2
    const environment = await new DockerComposeEnvironment(".", "docker-compose.yml").up()
3
    const port = environment.getContainer("mongod").getFirstMappedPort()
4
    const host = environment.getContainer("mongod").getHost()
5
    process.env.MONGO_URL = `mongodb://${host}:${port}/atlas-local-test?directConnection=true`
6
    const mongoClient = new MongoClient(process.env.MONGO_URL)
7
    try {
8
        await mongoClient
9
            .db()
10
            .admin()
11
            .command({
12
                replSetInitiate: {
13
                    _id: "local",
14
                    members: [{_id: 0, host: "10.6.0.5:27017"}]
15
                }
16
            })
17
        await new Promise((r) => setTimeout(r, 500))
18
        const indexDefinition = path.join(__dirname, "../index.json")
19
        const definition = JSON.parse(fs.readFileSync(indexDefinition).toString("utf-8"))
20
        const collection = await mongoClient.db("atlas-local-test").createCollection("items")
21
        await collection.createSearchIndex({name: "items-index", definition})
22
    } finally {
23
        await mongoClient.close()
24
    }
25
    global.__MONGO_ENV__ = environment
26
}

1
beforeAll(async () => {
2
    await mongoose.connect(process.env.MONGO_URL!)
3
    const itemModel1 = new MongoItem({
4
        name: "Cool Thing",
5
        price: 1337,
6
    })
7
    await MongoItemModel.create(itemModel1)
8
    const itemModel2 = new MongoItem({
9
        name: "Nice Thing",
10
        price: 10000,
11
    })
12
    await MongoItemModel.create(itemModel2)
13
    await new Promise((r) => setTimeout(r, 1000))
14
})
15
16
describe("MongoItemRepository", () => {
17
    describe("getItemsInPriceRange", () => {
18
        it("get all items in given price range", async () => {
19
            const items = await repository.getItemsInPriceRange(1000, 2000)
20
            expect(items).toHaveLength(1)
21
        })
22
    })
23
})
24
25
afterAll(async () => {
26
    await mongoose.connection.collection("items").deleteMany({})
27
    await mongoose.connection.close()
28
})