什么是JavaScript的迭代器与生成器在测试数据生成中的使用，以及它们如何简化大规模测试用例？

yield

function* generateTestUsers(count = Infinity) {
    let id = 1;
    while (id <= count) {
        yield {
            id: id,
            username: `user_${id}`,
            email: `user_${id}@example.com`,
            password: `password${id}`
        };
        id++;
    }
}

// 实际使用时，可以这样按需获取
const userGenerator = generateTestUsers(1000000); // 声明要生成100万用户，但并不会立即占用内存

// 在测试循环中，每次调用next()才生成一个用户
for (let i = 0; i < 5; i++) {
    const user = userGenerator.next().value;
    console.log(user);
}

// 甚至可以直接在for...of循环中使用，它会自动调用next()
// for (const user of generateTestUsers(3)) {
//     console.log(user);
// }

yield

next()

function* generateAdvancedUsers(options = {}) {
    let id = 1;
    const {
        minAge = 18, maxAge = 60,
        roles = ['user'],
        isActiveProbability = 0.8
    } = options;

    while (true) { // 可以是无限流，也可以设定上限
        const age = Math.floor(Math.random() * (maxAge - minAge + 1)) + minAge;
        const role = roles[Math.floor(Math.random() * roles.length)];
        const isActive = Math.random() < isActiveProbability;

        yield {
            id: id++,
            username: `user_${id}_${role}`,
            email: `user_${id}_${role}@example.com`,
            age: age,
            role: role,
            isActive: isActive,
            createdAt: new Date()
        };
    }
}

// 想要10个活跃的管理员用户
const adminUsers = generateAdvancedUsers({
    roles: ['admin'],
    isActiveProbability: 0.9
});
for (let i = 0; i < 10; i++) {
    console.log(adminUsers.next().value);
}

// 想要5个普通用户，年龄在20-30之间
const youngUsers = generateAdvancedUsers({
    minAge: 20,
    maxAge: 30
});
for (let i = 0; i < 5; i++) {
    console.log(youngUsers.next().value);
}

options

yield

[Symbol.iterator]()

next()

for...of

test.each

describe.each

// userGenerator.js
function* generateLoginCredentials(count = Infinity) {
    let id = 1;
    while (id <= count) {
        yield [`user_${id}`, `password${id}`, true]; // [username, password, expectedSuccess]
        id++;
        if (id % 5 === 0) { // 模拟一些失败的用例
            yield [`invalid_user_${id}`, `wrong_password`, false];
            id++;
        }
    }
}

module.exports = { generateLoginCredentials };

// login.test.js
const { generateLoginCredentials } = require('./userGenerator');

// 假设有一个模拟的登录函数
const mockLogin = (username, password) => {
    return new Promise(resolve => {
        setTimeout(() => {
            if (username.startsWith('user_') && password.startsWith('password')) {
                resolve(true);
            } else {
                resolve(false);
            }
        }, 10);
    });
};

describe('用户登录功能', () => {
    // 使用生成器作为数据源，只取前10个用例
    // 注意：Jest的.each通常需要一个数组。这里我们手动从生成器中取出部分数据形成数组。
    // 如果是无限生成器，需要明确限制取出的数量。
    const testCases = Array.from(generateLoginCredentials(10));

    test.each(testCases)(
        '登录用户 %s 密码 %s 应该返回 %s',
        async (username, password, expectedSuccess) => {
            const result = await mockLogin(username, password);
            expect(result).toBe(expectedSuccess);
        }
    );

    // 如果生成器返回的是一个有限序列，或者你希望测试所有生成的数据，
    // 也可以直接将生成器对象传入，但需要确保它能被转换成数组或Jest能直接迭代。
    // 在某些场景下，框架可能需要一个明确的数组。
    // 另一种更灵活的方式是封装一个函数，它返回一个可迭代对象，
    // 然后在测试文件中按需调用并转换为数组。

    // 假设我们想测试一些特定场景，直接从生成器中提取
    test('一个特定的成功登录', async () => {
        const gen = generateLoginCredentials();
        gen.next(); // 跳过第一个
        const [username, password, expectedSuccess] = gen.next().value; // 取第二个
        const result = await mockLogin(username, password);
        expect(result).toBe(expectedSuccess);
    });
});

test.each

Array.from