@tldraw/sync-core 5.3.0-next.fc37bd96825b → 5.3.0

package/src/test/TestServer.ts

@@ -1,12 +1,18 @@
 import { StoreSchema, UnknownRecord } from '@tldraw/store'
 import { InMemorySyncStorage } from '../lib/InMemorySyncStorage'
+import { TLObjectStoreAccess } from '../lib/protocol'
 import { RoomSnapshot, TLSyncRoom } from '../lib/TLSyncRoom'
 import { TestSocketPair } from './TestSocketPair'
 
+type TestRoomOptions<R extends UnknownRecord> = Omit<
+	ConstructorParameters<typeof TLSyncRoom<R, undefined>>[0],
+	'schema' | 'storage'
+>
+
 export class TestServer<R extends UnknownRecord, P = unknown> {
 	room: TLSyncRoom<R, undefined>
 	storage: InMemorySyncStorage<R>
-	constructor(schema: StoreSchema<R, P>, snapshot?: RoomSnapshot) {
+	constructor(schema: StoreSchema<R, P>, snapshot?: RoomSnapshot, roomOpts?: TestRoomOptions<R>) {
 		// Use provided snapshot or create an empty one with the current schema
 		this.storage = new InMemorySyncStorage<R>({
 			snapshot: snapshot ?? {
@@ -16,15 +22,19 @@ export class TestServer<R extends UnknownRecord, P = unknown> {
 				schema: schema.serialize(),
 			},
 		})
-		this.room = new TLSyncRoom<R, undefined>({ schema, storage: this.storage })
+		this.room = new TLSyncRoom<R, undefined>({ schema, storage: this.storage, ...roomOpts })
 	}
 
-	connect(socketPair: TestSocketPair<R>): void {
+	connect(
+		socketPair: TestSocketPair<R>,
+		opts?: { isReadonly?: boolean; objectAccess?: TLObjectStoreAccess }
+	): void {
 		this.room.handleNewSession({
 			sessionId: socketPair.id,
 			socket: socketPair.roomSocket,
 			meta: undefined,
-			isReadonly: false,
+			isReadonly: opts?.isReadonly ?? false,
+			objectAccess: opts?.objectAccess,
 		})
 
 		socketPair.clientSocket.connectionStatus = 'online'