如何在单元测试时异步地从db中获取记录？

const mysql = require('mysql');

// 创建数据库连接池
const pool = mysql.createPool({
  host: 'your-db-host',
  user: 'your-db-user',
  password: 'your-db-password',
  database: 'your-db-name',
});

// 异步获取记录的函数
function getRecordFromDB(id, callback) {
  pool.query('SELECT * FROM your_table WHERE id = ?', [id], (error, results) => {
    if (error) {
      callback(error, null);
    } else {
      callback(null, results);
    }
  });
}

// 异步测试用例
async function testGetRecordFromDB() {
  const record = await new Promise((resolve, reject) => {
    getRecordFromDB(1, (error, results) => {
      if (error) {
        reject(error);
      } else {
        resolve(results);
      }
    });
  });

  // 断言验证获取的记录是否符合预期
  // ...
}

// 执行测试用例
testGetRecordFromDB()
  .then(() => console.log('测试通过'))
  .catch(error => console.error('测试失败', error));