feat(basemodel): compute refs from the added columns

adonisjs · Oct 7, 2019 · dc79df7 · dc79df7
1 parent 212b5e9
commit dc79df7
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/src/Orm/BaseModel/index.ts b/src/Orm/BaseModel/index.ts
@@ -184,6 +184,7 @@ export class BaseModel implements ModelContract {
     Object.defineProperty(this, '$columns', { value: new Map() })
     Object.defineProperty(this, '$computed', { value: new Map() })
     Object.defineProperty(this, '$relations', { value: new Map() })
+    Object.defineProperty(this, '$refs', { value: {} })
     Object.defineProperty(this, '_mappings', {
       value: {
         cast: new Map(),
@@ -219,6 +220,7 @@ export class BaseModel implements ModelContract {
 
     this.$columns.set(name, column)
     this._mappings.cast.set(column.castAs!, name)
+    this.$refs[name] = column.castAs
    }
 
   /**

diff --git a/test-helpers/tmp/db.sqlite b/test-helpers/tmp/db.sqlite
diff --git a/test/orm/base-model.spec.ts b/test/orm/base-model.spec.ts
@@ -99,6 +99,22 @@ test.group('Base model | boot', (group) => {
     assert.deepEqual(mapToObj(User.$columns), {})
     assert.deepEqual(mapToObj(User.$relations), {})
     assert.deepEqual(mapToObj(User.$computed), {})
+    assert.deepEqual(User.$refs, {})
+  })
+
+  test('compute refs from the added columns', async (assert) => {
+    class User extends BaseModel {
+      public static $increments = false
+
+      @column({ primary: true })
+      public id: number
+
+      @column()
+      public userName: string
+    }
+
+    User.$boot()
+    assert.deepEqual(User.$refs, { id: 'id', userName: 'user_name' })
   })
 })