Skip to content

Commit 0dfe4e8

Browse files
leourbinaleourbina
committed
feat: Adds Cockroachdb container
1 parent 3b48cc0 commit 0dfe4e8

File tree

10 files changed

+319
-0
lines changed

10 files changed

+319
-0
lines changed

‎docs/modules/cockroachdb.md‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
# CockroachDB Module
2+
3+
[CockroachDB](https://github.com/cockroachdb/cockroach) is a cloud-native, postgresql compatible, distributed SQL database designed to build, scale, and manage modern, data-intensive applications.
4+
5+
6+
## Install
7+
8+
```bash
9+
npm install @testcontainers/cockroachdb --save-dev
10+
```
11+
12+
## Examples
13+
14+
<!--codeinclude-->
15+
[Connect and execute query:](../../packages/modules/cockroachdb/src/cockroachdb-container.test.ts) inside_block:connect
16+
<!--/codeinclude-->
17+
18+
<!--codeinclude-->
19+
[Connect and execute query using URI:](../../packages/modules/cockroachdb/src/cockroachdb-container.test.ts) inside_block:uriConnect
20+
<!--/codeinclude-->
21+
22+
<!--codeinclude-->
23+
[Set database:](../../packages/modules/cockroachdb/src/cockroachdb-container.test.ts) inside_block:setDatabase
24+
<!--/codeinclude-->
25+
26+
<!--codeinclude-->
27+
[Set username:](../../packages/modules/cockroachdb/src/cockroachdb-container.test.ts) inside_block:setUsername
28+
<!--/codeinclude-->

‎mkdocs.yml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ nav:
4747
- Cassandra: modules/cassandra.md
4848
- ChromaDB: modules/chromadb.md
4949
- Couchbase: modules/couchbase.md
50+
- CockroachDB: modules/cockroachdb.md
5051
- Elasticsearch: modules/elasticsearch.md
5152
- EventStoreDB: modules/eventstoredb.md
5253
- GCloud: modules/gcloud.md

‎package-lock.json‎

Lines changed: 16 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎packages/modules/cockroachdb/jest.config.ts‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
import type { Config } from "jest";
2+
import * as path from "path";
3+
4+
const config: Config = {
5+
preset: "ts-jest",
6+
moduleNameMapper: {
7+
"^testcontainers$": path.resolve(__dirname, "../../testcontainers/src"),
8+
},
9+
};
10+
11+
export default config;

‎packages/modules/cockroachdb/package.json‎

Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,39 @@
1+
{
2+
"name": "@testcontainers/cockroachdb",
3+
"version": "10.18.0",
4+
"license": "MIT",
5+
"keywords": [
6+
"cockroachdb",
7+
"crdb",
8+
"testing",
9+
"docker",
10+
"testcontainers"
11+
],
12+
"description": "CockroachDB module for Testcontainers",
13+
"homepage": "https://github.com/testcontainers/testcontainers-node#readme",
14+
"repository": {
15+
"type": "git",
16+
"url": "https://github.com/testcontainers/testcontainers-node"
17+
},
18+
"bugs": {
19+
"url": "https://github.com/testcontainers/testcontainers-node/issues"
20+
},
21+
"main": "build/index.js",
22+
"files": [
23+
"build"
24+
],
25+
"publishConfig": {
26+
"access": "public"
27+
},
28+
"scripts": {
29+
"prepack": "shx cp ../../../README.md . && shx cp ../../../LICENSE .",
30+
"build": "tsc --project tsconfig.build.json"
31+
},
32+
"devDependencies": {
33+
"@types/pg": "^8.11.6",
34+
"pg": "^8.12.0"
35+
},
36+
"dependencies": {
37+
"testcontainers": "^10.18.0"
38+
}
39+
}

‎packages/modules/cockroachdb/src/cockroachdb-container.test.ts‎

Lines changed: 110 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,110 @@
1+
import { Client } from "pg";
2+
import { CockroachDbContainer } from "./cockroachdb-container";
3+
4+
describe("CockroachDbContainer", () => {
5+
jest.setTimeout(180_000);
6+
7+
// connect {
8+
it("should connect and return a query result", async () => {
9+
const container = await new CockroachDbContainer().start();
10+
11+
console.log(container.getDatabase(), container.getHost(), container.getPort());
12+
13+
const client = new Client({
14+
host: container.getHost(),
15+
port: container.getPort(),
16+
database: container.getDatabase(),
17+
user: container.getUsername(),
18+
ssl: false,
19+
});
20+
21+
await client.connect();
22+
23+
const result = await client.query("SELECT 1");
24+
expect(result.rows[0]).toEqual({ "?column?": "1" });
25+
26+
await client.end();
27+
await container.stop();
28+
});
29+
// }
30+
31+
// uriConnect {
32+
it("should work with database URI", async () => {
33+
const container = await new CockroachDbContainer().start();
34+
35+
const client = new Client({
36+
connectionString: container.getConnectionUri(),
37+
});
38+
await client.connect();
39+
40+
const result = await client.query("SELECT 1");
41+
expect(result.rows[0]).toEqual({ "?column?": "1" });
42+
43+
await client.end();
44+
await container.stop();
45+
});
46+
// }
47+
48+
// setDatabase {
49+
it("should set database", async () => {
50+
const container = await new CockroachDbContainer().withDatabase("custom_database").start();
51+
52+
const client = new Client({
53+
host: container.getHost(),
54+
port: container.getPort(),
55+
database: container.getDatabase(),
56+
user: container.getUsername(),
57+
});
58+
await client.connect();
59+
60+
const result = await client.query("SELECT current_database()");
61+
expect(result.rows[0]).toEqual({ current_database: "custom_database" });
62+
63+
await client.end();
64+
await container.stop();
65+
});
66+
// }
67+
68+
// setUsername {
69+
it("should set username", async () => {
70+
const container = await new CockroachDbContainer().withUsername("custom_username").start();
71+
72+
const client = new Client({
73+
host: container.getHost(),
74+
port: container.getPort(),
75+
database: container.getDatabase(),
76+
user: container.getUsername(),
77+
});
78+
await client.connect();
79+
80+
const result = await client.query("SELECT current_user");
81+
expect(result.rows[0]).toEqual({ current_user: "custom_username" });
82+
83+
await client.end();
84+
await container.stop();
85+
});
86+
// }
87+
88+
// workWithRestartedContainer {
89+
it("should work with restarted container", async () => {
90+
const container = await new CockroachDbContainer().start();
91+
const port = container.getFirstMappedPort();
92+
await container.restart();
93+
expect(port).not.toEqual(container.getFirstMappedPort());
94+
95+
const client = new Client({
96+
host: container.getHost(),
97+
port: container.getPort(),
98+
database: container.getDatabase(),
99+
user: container.getUsername(),
100+
});
101+
await client.connect();
102+
103+
const result = await client.query("SELECT 1");
104+
expect(result.rows[0]).toEqual({ "?column?": "1" });
105+
106+
await client.end();
107+
await container.stop();
108+
});
109+
// }
110+
});

‎packages/modules/cockroachdb/src/cockroachdb-container.ts‎

Lines changed: 79 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,79 @@
1+
import { AbstractStartedContainer, GenericContainer, StartedTestContainer, Wait } from "testcontainers";
2+
3+
const COCKROACH_PORT = 26257;
4+
const COCKROACH_HTTP_PORT = 26258;
5+
6+
export class CockroachDbContainer extends GenericContainer {
7+
private database = "test";
8+
private username = "test";
9+
10+
constructor(image = "cockroachdb/cockroach:v24.3.5") {
11+
super(image);
12+
this.withExposedPorts(COCKROACH_PORT, COCKROACH_HTTP_PORT)
13+
.withCommand(["start-single-node", "--insecure", "--http-addr=0.0.0.0:" + COCKROACH_HTTP_PORT])
14+
.withHealthCheck({
15+
test: ["CMD-SHELL", "curl -f http://localhost:" + COCKROACH_HTTP_PORT + " || exit 1"],
16+
interval: 1000,
17+
timeout: 3000,
18+
retries: 5,
19+
startPeriod: 1000,
20+
})
21+
.withWaitStrategy(Wait.forHealthCheck());
22+
}
23+
24+
public withDatabase(database: string): this {
25+
this.database = database;
26+
return this;
27+
}
28+
29+
public withUsername(username: string): this {
30+
this.username = username;
31+
return this;
32+
}
33+
34+
public override async start(): Promise<StartedCockroachDbContainer> {
35+
this.withEnvironment({
36+
COCKROACH_DATABASE: this.database,
37+
COCKROACH_USER: this.username,
38+
});
39+
return new StartedCockroachDbContainer(await super.start(), this.database, this.username);
40+
}
41+
}
42+
43+
export class StartedCockroachDbContainer extends AbstractStartedContainer {
44+
private port: number;
45+
46+
constructor(
47+
startedTestContainer: StartedTestContainer,
48+
private readonly database: string,
49+
private readonly username: string
50+
) {
51+
super(startedTestContainer);
52+
this.port = startedTestContainer.getMappedPort(COCKROACH_PORT);
53+
}
54+
55+
public getPort(): number {
56+
this.port = this.startedTestContainer.getMappedPort(COCKROACH_PORT);
57+
return this.port;
58+
}
59+
60+
public getDatabase(): string {
61+
return this.database;
62+
}
63+
64+
public getUsername(): string {
65+
return this.username;
66+
}
67+
68+
/**
69+
* @returns A connection URI in the form of `postgres[ql]://[username[:password]@][host[:port],]/database`
70+
*/
71+
public getConnectionUri(): string {
72+
const url = new URL("", "postgres://");
73+
url.hostname = this.getHost();
74+
url.port = this.getPort().toString();
75+
url.pathname = this.getDatabase();
76+
url.username = this.getUsername();
77+
return url.toString();
78+
}
79+
}

‎packages/modules/cockroachdb/src/index.ts‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
export { CockroachDbContainer, StartedCockroachDbContainer } from "./cockroachdb-container";

‎packages/modules/cockroachdb/tsconfig.build.json‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
{
2+
"extends": "./tsconfig.json",
3+
"exclude": [
4+
"build",
5+
"jest.config.ts",
6+
"src/**/*.test.ts"
7+
],
8+
"references": [
9+
{
10+
"path": "../../testcontainers"
11+
}
12+
]
13+
}

‎packages/modules/cockroachdb/tsconfig.json‎

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
{
2+
"extends": "../../../tsconfig.base.json",
3+
"compilerOptions": {
4+
"rootDir": "src",
5+
"outDir": "build",
6+
"paths": {
7+
"testcontainers": [
8+
"../../testcontainers/src"
9+
]
10+
}
11+
},
12+
"exclude": [
13+
"build",
14+
"jest.config.ts"
15+
],
16+
"references": [
17+
{
18+
"path": "../../testcontainers"
19+
}
20+
]
21+
}

0 commit comments

Comments
 (0)