Add/update tests

test/models/ModelWithQueryMethod.ts

@@ -14,7 +14,7 @@ export class ModelWithQueryMethod extends Model<ModelWithQueryMethod>() {
   password?: string
 
   static findByEmail(email: string){
-    return ModelWithQueryMethod.find({email: email} as any).one()
+    return this.find({email: email} as any).one()
   }
 
 }

test/models/ModelWithRef.ts

@@ -12,6 +12,21 @@ export class Address extends Model<Address>(){
 }
 
 
+@model()
+export class Product extends Model<Product>(){
+  @field()
+  title?: string
+}
+
+@model()
+export class User extends Model<User>(){
+  @field()
+  name?: string
+
+  @field({ref: [Product]})
+  products?: Ref<Product>[]
+}
+
 @model()
 export class ModelWithRef extends Model<ModelWithRef>() {
 
@@ -21,7 +36,13 @@ export class ModelWithRef extends Model<ModelWithRef>() {
   @field()
   name?: string
 
-  @field({type: Address})
+  @field({ref: Address})
   address?: Ref<Address>
 
+  @field({ref: [Product]})
+  products?: Ref<Product>[]
+
+  @field({type: [User]})
+  users?: User[]
+
 }

test/tests/populate.test.ts

@@ -0,0 +1,53 @@
+import {Address, ModelWithRef, Product, User} from "../models/ModelWithRef";
+
+
+it('should populate 1 level', async () => {
+  const address = new Address({city: "Saint Martins2"})
+  await address.save()
+  const ob = await ModelWithRef.create({email: "pop3@gmail.com", address})
+  const retrieved = await ModelWithRef.find({email: "pop3@gmail.com"}).one().orThrow().populate({
+    path: "address"
+  })
+  expect(address._id.toString()).toBe((<Address>retrieved.address)._id.toString())
+});
+
+
+it('should populate array', async () => {
+  const product1 = new Product({title: "Apple"})
+  await product1.save()
+
+  const product2 = new Product({title: "Mango"})
+  await product2.save()
+
+  const ob = await ModelWithRef.create({email: "pop4@gmail.com", products: [product1, product2]})
+
+  const retrieved = await ModelWithRef.find({email: "pop4@gmail.com"}).one().populate({
+    path: "products"
+  })
+  // @ts-ignore
+  expect(["Mango", "Apple"]).toContain((<Product>retrieved.products![0]).title)
+
+});
+
+
+it('should populate level 2 array', async () => {
+  const product1 = new Product({title: "Apple"})
+  await product1.save()
+
+  const product2 = new Product({title: "Mango"})
+  await product2.save()
+
+  const user1 = new User({name: "User1", products: [product1]})
+  const user2 = new User({name: "User1", products: [product2]})
+
+  const ob = await ModelWithRef.create({email: "pop5@gmail.com", users: [user2, user1]})
+
+  const retrieved = await ModelWithRef.find({email: "pop5@gmail.com"}).one().populate({
+    path: "users.products"
+  }).orThrow()
+  expect(["Mango", "Apple"]).toContain((<Product[]>retrieved.users![0].products)![0].title)
+
+});
+
+
+

test/tests/save.test.ts

@@ -13,10 +13,13 @@ it('should save models properly', async () => {
 
 
 it('should have same _id after saving', async () => {
-  const staff = new User({lastName: "Shamim", firstName: "Model", email: "ddx@example.com"})
+  const staff = new User({lastName: "Shamim", firstName: "Model", email: "ddx2@example.com"})
   const _id = staff._id
-  await staff.save()
+  const saved = await staff.save()
   expect(_id).toBe(staff._id)
+  expect(_id).toBe(saved._id)
+  const retrieved = await User.find({email: "ddx2@example.com"}).one().orThrow()
+  expect(_id!.toString()).toBe(retrieved._id!.toString())
 });
 
 
@@ -49,4 +52,15 @@ it('should save and retrieve date properly', async () => {
   await datable.save()
   const retrieved = await BasicDate.find().one().orThrow()
   expect(date).toEqual(retrieved.created)
+});
+
+it('should not save extra properties', async () => {
+  const p = {lastName: "Doe", email: "withextra@example.com", extra: "extra value"}
+  const b = new Basic(p)
+  await b.save()
+  // @ts-ignore
+  const getExtra = (ob) => ob.extra
+
+  const retrieved = Basic.find({email: "withextra@example.com"}).one().orThrow()
+  expect(getExtra(retrieved)).toBeUndefined()
 });