Update findById method

data_mapper.ts

@@ -63,16 +63,26 @@ export class DataMapper<Def extends Definition> {
     });
   }
 
-  async findById(args: DataMapperIdQueryArgs): Promise<GetResponse<Def>> {
-    const datastore = args.datastore ?? this.#defaultDatastore;
+  async findById(
+    args: DataMapperIdQueryArgs | string,
+  ): Promise<GetResponse<Def>> {
+    let datastore = this.#defaultDatastore;
+    let id = undefined;
+    if (typeof args === "string") {
+      id = args;
+    } else {
+      id = args.id;
+      datastore = args.datastore ?? this.#defaultDatastore;
+    }
     if (!datastore) {
       throw new ConfigurationError(this.#datastoreMissingError);
     }
+    const [client, logger] = [this.#client, this.#logger];
     return await func.findById<Def>({
-      client: this.#client,
+      client,
       datastore,
-      id: args.id,
-      logger: this.#logger,
+      logger,
+      id,
     });
   }
 

data_mapper_test.ts

@@ -49,6 +49,26 @@ mf.mock("POST@/api/apps.datastore.query", () => {
   );
 });
 
+mf.mock("POST@/api/apps.datastore.get", () => {
+  return new Response(
+    JSON.stringify({
+      "ok": true,
+      "datastore": "suveys",
+      "item": {
+        "id": "123",
+        "title": "Off-site event ideas",
+        "questions": [
+          "Can you share a fun idea for our off-site event in December?",
+        ],
+        "closed": false,
+      },
+    }),
+    {
+      status: 200,
+    },
+  );
+});
+
 export const Surveys = DefineDatastore(
   {
     name: "surveys",
@@ -108,6 +128,9 @@ Deno.test("Run a query", async () => {
     logLevel: "DEBUG",
   });
 
+  await dataMapper.findById("123");
+  await dataMapper.findById({ id: "123" });
+
   await dataMapper.findAll();
 
   await dataMapper.findAllBy({