refactor(location)!: zip code state

src/modules/location/index.ts

@@ -1,4 +1,5 @@
 import type { Faker } from '../..';
+import { FakerError } from '../..';
 import { deprecated } from '../../internal/deprecated';
 
 /**
@@ -23,8 +24,12 @@ export class LocationModule {
    * the locale's zip format is used.
    *
    * @param options The format used to generate the the zip code or an options object. Defaults to `{}`.
+   * @param options.state The state to generate the zip code for.
+   * Only works for locales with postcode_by_state definition. If a locale does not
+   * have a postcode_by_state definition an error is thrown.
    * @param options.format The optional format used to generate the the zip code.
    * By default, a random format is used from the locale zip formats.
+   * This wont be used if the state option is specified.
    *
    * @see faker.helpers.replaceSymbols()
    *
@@ -38,9 +43,18 @@ export class LocationModule {
     options:
       | string
       | {
+          /**
+           * The state to generate the zip code for.
+           *
+           * Only works for locales with postcode_by_state definition. If a locale does not
+           * have a postcode_by_state definition an error is thrown.
+           */
+          state?: string;
           /**
            * The optional format used to generate the the zip code.
            *
+           * This wont be used if the state option is specified.
+           *
            * @default faker.definitions.location.postcode
            */
           format?: string;
@@ -50,6 +64,19 @@ export class LocationModule {
       options = { format: options };
     }
 
+    const { state } = options;
+
+    if (state) {
+      const zipRange =
+        this.faker.definitions.location.postcode_by_state?.[state];
+
+      if (zipRange) {
+        return String(this.faker.number.int(zipRange));
+      }
+
+      throw new FakerError(`No zip code range found for state "${state}"`);
+    }
+
     let { format = this.faker.definitions.location.postcode } = options;
     if (typeof format === 'string') {
       format = [format];
@@ -64,8 +91,7 @@ export class LocationModule {
    * Generates random zip code from state abbreviation.
    *
    * Only works for locales with postcode_by_state definition. If a locale does not
-   * have a postcode_by_state definition, a random zip code is generated according
-   * to the locale's zip format.
+   * have a postcode_by_state definition an error is thrown.
    *
    * @param options A state abbreviation or an options object. Defaults to `{}`.
    * @param options.state The abbreviation of the state to generate the zip code for.
@@ -76,6 +102,8 @@ export class LocationModule {
    * fakerEN_US.location.zipCodeByState("??") // '47683-9880'
    *
    * @since 8.0.0
+   *
+   * @deprecated Use `faker.location.zipCode({ state })` instead.
    */
   zipCodeByState(
     options:
@@ -88,18 +116,20 @@ export class LocationModule {
           state?: string;
         } = {}
   ): string {
+    deprecated({
+      deprecated: 'faker.location.zipCodeByState',
+      proposed: 'faker.location.zipCode({ state })',
+      since: '8.0',
+      until: '9.0',
+    });
+
     if (typeof options === 'string') {
       options = { state: options };
     }
 
     const { state } = options;
 
-    const zipRange = this.faker.definitions.location.postcode_by_state?.[state];
-    if (zipRange) {
-      return String(this.faker.number.int(zipRange));
-    }
-
-    return this.zipCode();
+    return this.zipCode({ state });
   }
 
   /**

test/__snapshots__/location.spec.ts.snap

@@ -152,12 +152,6 @@ exports[`location > 42 > zipCode > with string 1`] = `"379"`;
 
 exports[`location > 42 > zipCodeByState > noArgs 1`] = `"79177"`;
 
-exports[`location > 42 > zipCodeByState > with state options 1`] = `"79177"`;
-
-exports[`location > 42 > zipCodeByState > with string 1 1`] = `"79177"`;
-
-exports[`location > 42 > zipCodeByState > with string 2 1`] = `"79177"`;
-
 exports[`location > 1211 > buildingNumber 1`] = `"487"`;
 
 exports[`location > 1211 > cardinalDirection > noArgs 1`] = `"West"`;
@@ -310,12 +304,6 @@ exports[`location > 1211 > zipCode > with string 1`] = `"948"`;
 
 exports[`location > 1211 > zipCodeByState > noArgs 1`] = `"48721-9061"`;
 
-exports[`location > 1211 > zipCodeByState > with state options 1`] = `"48721-9061"`;
-
-exports[`location > 1211 > zipCodeByState > with string 1 1`] = `"48721-9061"`;
-
-exports[`location > 1211 > zipCodeByState > with string 2 1`] = `"48721-9061"`;
-
 exports[`location > 1337 > buildingNumber 1`] = `"51225"`;
 
 exports[`location > 1337 > cardinalDirection > noArgs 1`] = `"East"`;
@@ -467,9 +455,3 @@ exports[`location > 1337 > zipCode > with format option 1`] = `"251-225"`;
 exports[`location > 1337 > zipCode > with string 1`] = `"251"`;
 
 exports[`location > 1337 > zipCodeByState > noArgs 1`] = `"51225"`;
-
-exports[`location > 1337 > zipCodeByState > with state options 1`] = `"51225"`;
-
-exports[`location > 1337 > zipCodeByState > with string 1 1`] = `"51225"`;
-
-exports[`location > 1337 > zipCodeByState > with string 2 1`] = `"51225"`;

test/location.spec.ts

@@ -1,5 +1,5 @@
 import { describe, expect, it } from 'vitest';
-import { faker, fakerEN_CA, fakerEN_US } from '../src';
+import { faker, fakerEN_CA, fakerEN_US, FakerError } from '../src';
 import { seededTests } from './support/seededRuns';
 import { times } from './support/times';
 
@@ -113,7 +113,9 @@ describe('location', () => {
     t.describe('zipCode', (t) => {
       t.it('noArgs')
         .it('with string', '###')
-        .it('with format option', { format: '###-###' });
+        .it('with format option', { format: '###-###' })
+        .it('with state option', { state: 'CA' })
+        .it('with options', { state: 'CA', format: '###-###' });
     });
 
     t.describe('zipCodeByState', (t) => {
@@ -156,6 +158,22 @@ describe('location', () => {
 
           expect(zipCode).toMatch(/^[A-Za-z]\d[A-Za-z]\s?\d[A-Za-z]\d$/);
         });
+
+        it.each([
+          ['IL', 60001, 62999],
+          ['GA', 30001, 31999],
+          ['WA', 98001, 99403],
+        ])('returns zipCode valid for state %s', (state, lower, upper) => {
+          const zipCode1 = +fakerEN_US.location.zipCode({ state });
+          expect(zipCode1).toBeGreaterThanOrEqual(lower);
+          expect(zipCode1).toBeLessThanOrEqual(upper);
+        });
+
+        it('should throw when definitions.location.postcode_by_state not set', () => {
+          expect(() => faker.location.zipCode({ state: 'XX' })).toThrow(
+            new FakerError('No zip code range found for state "XX"')
+          );
+        });
       });
 
       describe('zipCodeByState()', () => {