1 files changed, 119 insertions, 0 deletions
diff --git a/packages/db/test/unit/remote-info.test.js b/packages/db/test/unit/remote-info.test.js
new file mode 100644
index 000000000..2c58f28b7
--- /dev/null
+++ b/packages/db/test/unit/remote-info.test.js
@@ -0,0 +1,119 @@
+import assert from 'node:assert';
+import test, { after, beforeEach, describe } from 'node:test';
+import { getManagedRemoteToken, getRemoteDatabaseInfo } from '../../dist/core/utils.js';
+import { clearEnvironment } from '../test-utils.js';
+
+describe('RemoteDatabaseInfo', () => {
+	beforeEach(() => {
+		clearEnvironment();
+	});
+
+	test('default remote info', () => {
+		const dbInfo = getRemoteDatabaseInfo();
+
+		assert.deepEqual(dbInfo, {
+			type: 'studio',
+			url: 'https://db.services.astro.build',
+		});
+	});
+
+	test('configured Astro Studio remote', () => {
+		process.env.ASTRO_STUDIO_REMOTE_DB_URL = 'https://studio.astro.build';
+		const dbInfo = getRemoteDatabaseInfo();
+
+		assert.deepEqual(dbInfo, {
+			type: 'studio',
+			url: 'https://studio.astro.build',
+		});
+	});
+
+	test('configured libSQL remote', () => {
+		process.env.ASTRO_DB_REMOTE_URL = 'libsql://libsql.self.hosted';
+		const dbInfo = getRemoteDatabaseInfo();
+
+		assert.deepEqual(dbInfo, {
+			type: 'libsql',
+			url: 'libsql://libsql.self.hosted',
+		});
+	});
+
+	test('configured both libSQL and Studio remote', () => {
+		process.env.ASTRO_DB_REMOTE_URL = 'libsql://libsql.self.hosted';
+		process.env.ASTRO_STUDIO_REMOTE_DB_URL = 'https://studio.astro.build';
+		const dbInfo = getRemoteDatabaseInfo();
+
+		assert.deepEqual(dbInfo, {
+			type: 'studio',
+			url: 'https://studio.astro.build',
+		});
+	});
+});
+
+describe('RemoteManagedToken', () => {
+	// Avoid conflicts with other tests
+	beforeEach(() => {
+		clearEnvironment();
+		process.env.ASTRO_STUDIO_APP_TOKEN = 'studio token';
+		process.env.ASTRO_DB_APP_TOKEN = 'db token';
+	});
+	after(() => {
+		clearEnvironment();
+	});
+
+	test('given token for default remote', async () => {
+		const { token } = await getManagedRemoteToken('given token');
+		assert.equal(token, 'given token');
+	});
+
+	test('token for default remote', async () => {
+		const { token } = await getManagedRemoteToken();
+
+		assert.equal(token, 'studio token');
+	});
+
+	test('given token for configured Astro Studio remote', async () => {
+		process.env.ASTRO_STUDIO_REMOTE_DB_URL = 'https://studio.astro.build';
+		const { token } = await getManagedRemoteToken('given token');
+		assert.equal(token, 'given token');
+	});
+
+	test('token for configured Astro Studio remote', async () => {
+		process.env.ASTRO_STUDIO_REMOTE_DB_URL = 'https://studio.astro.build';
+		const { token } = await getManagedRemoteToken();
+
+		assert.equal(token, 'studio token');
+	});
+
+	test('given token for configured libSQL remote', async () => {
+		process.env.ASTRO_DB_REMOTE_URL = 'libsql://libsql.self.hosted';
+		const { token } = await getManagedRemoteToken('given token');
+		assert.equal(token, 'given token');
+	});
+
+	test('token for configured libSQL remote', async () => {
+		process.env.ASTRO_DB_REMOTE_URL = 'libsql://libsql.self.hosted';
+		const { token } = await getManagedRemoteToken();
+
+		assert.equal(token, 'db token');
+	});
+
+	test('token for given Astro Studio remote', async () => {
+		process.env.ASTRO_DB_REMOTE_URL = 'libsql://libsql.self.hosted';
+		const { token } = await getManagedRemoteToken(undefined, {
+			type: 'studio',
+			url: 'https://studio.astro.build',
+		});
+
+		assert.equal(token, 'studio token');
+	});
+
+	test('token for given libSQL remote', async () => {
+		process.env.ASTRO_STUDIO_REMOTE_URL = 'libsql://libsql.self.hosted';
+		const { token } = await getManagedRemoteToken(undefined, {
+			type: 'libsql',
+			url: 'libsql://libsql.self.hosted',
+		});
+
+		assert.equal(token, 'db token');
+	});
+});