import test, { after, before } from "node:test";
import assert from "node:assert";
import mongoose from "mongoose";

const childSchema = new mongoose.Schema({
  name: { type: String, required: true },
});

const Child = mongoose.model("Child", childSchema);

const parentSchema = new mongoose.Schema({
  name: { type: String, required: true },
  childId: { type: mongoose.Schema.Types.ObjectId },
});

parentSchema.virtual("child", {
  ref: "Child",
  localField: "childId",
  foreignField: "_id",
  justOne: true,
});

const Parent = mongoose.model("Parent", parentSchema);

let parent;
let child;
before(async () => {
  console.log("Connecting");
  await mongoose.connect(
    "mongodb://localhost:27018/mongoose-bug?directConnection=true"
  );

  console.log("Creating child");
  child = await Child.create({
    name: "child",
  });

  parent = await Parent.create({
    name: "parent",
    childId: child._id,
  });
});

after(async () => {
  await mongoose.disconnect();
});

test("should populate when document is already found and .parent() should work on populated child", async () => {
  await parent.populate(["child"]);

  assert.equal(parent.child.parent()._id, parent._id);
});

test("should populate via findOne and .parent() should work on populated child", async () => {
  const foundParent = await Parent.findOne({ _id: parent._id }).populate([
    "child",
  ]);
  assert(foundParent, "parent not found");

  assert.equal(foundParent.child.parent()._id, foundParent._id);
});

test("should populate via find with no cursor and .parent() should work on populated child", async () => {
  const foundParents = await Parent.find({
    _id: { $in: [parent._id] },
  }).populate(["child"]);
  assert(foundParents[0], "parent not found");

  for (const parent of foundParents) {
    assert.equal(parent.child.parent()._id, parent._id);
  }
});

test("should populate via find with a cursor and .parent() should work on populated child", async () => {
  const cursor = Parent.find({
    _id: { $in: [parent._id] },
  })
    .populate(["child"])
    .cursor();

  for await (const parent of cursor) {
    assert.equal(parent.child.parent()._id, parent._id);
  }
});