personal-document-generator/__test__/utils/number-generator.test.js

A simple DNI (spanish ID document number) and NIE (id number for foreing people in spain) generators.

import { expect, test } from 'vitest';

import { getEightOrSevenDigitsNumber, generateNumber } from '../../generators/utils/number-generator.js';

test('should generate a number', () => {
  expect(typeof getEightOrSevenDigitsNumber()).toBe('number');
});

test('should generate a number of 8 digits by default', () => {
  expect(getEightOrSevenDigitsNumber().toString().length).toBe(8);
});

test('should generate a number with the passed number of digits', () => {
  expect(getEightOrSevenDigitsNumber(8).toString().length).toBe(8);
  expect(getEightOrSevenDigitsNumber(7).toString().length).toBe(7);
});

test('getEightOrSevenDigitsNumber should return a number of 7 digits', () => {
  let n = 8;
  while (n > 0) {
    expect(getEightOrSevenDigitsNumber(7).toString().length).toBe(7);
    n--;
  }
});
test('generate number should return a value between 0 an 2', () => {
  let n = 5;
  while (n > 0) {
    const num = generateNumber();
    expect(num >= 0 || num < 3).toBe(true);
    n--;
  }
});