diff --git a/lib/Account.js b/lib/Account.js index d63b03a..7534330 100644 --- a/lib/Account.js +++ b/lib/Account.js @@ -1,6 +1,9 @@ /* eslint-disable no-empty-function, getter-return */ 'use strict'; +const DatabaseConnectionPool = require('./DatabaseConnectionPool'); + +const TestTemplate = require('./TestTemplate'); const User = require('./User'); class Account extends User { @@ -8,12 +11,26 @@ class Account extends User { super('account', id); } - get classes() { + async getTestTemplates() { + const sql = ` + select + testTemplateId as id + from + testTemplate + where + accountId = ?; + `; - } + const conn = await new DatabaseConnectionPool(); + const records = await conn.runQuery(sql, [ this.id ]); - get testTemplates() { + const promises = records.map(record => { + return new TestTemplate(record.id); + }); + const objects = await Promise.all(promises); + + return objects; } createTestTemplate() {