Align query API types with runtime behavior (#25)

Author: justjavac

src/index.ts

@@ -28,11 +28,11 @@ export function getByCountry(country: string): readonly string[] {
   return byCountry[country] ?? emptyIps;
 }
 
-export function has(ip: string): boolean {
+export function has(ip: unknown): boolean {
   return typeof ip === "string" && ipToCountry.has(ip);
 }
 
-export function findCountry(ip: string): string | undefined {
+export function findCountry(ip: unknown): string | undefined {
   if (typeof ip !== "string") {
     return undefined;
   }

test/index.test.ts

@@ -7,6 +7,7 @@ import {
   getByCountry,
   has,
   ips,
+  random,
 } from "../src";
 
 test("exports the generated IP list", () => {
@@ -29,8 +30,23 @@ test("groups IPs by country", () => {
 test("looks up IP membership and country", () => {
   assert.equal(has("93.123.23.1"), true);
   assert.equal(has("127.0.0.1"), false);
+  assert.equal(has(127001), false);
   assert.equal(findCountry("118.174.25.251"), "Thailand");
   assert.equal(findCountry("127.0.0.1"), undefined);
+  assert.equal(findCountry(null), undefined);
+});
+
+test("returns random IPs from all data or a country", () => {
+  const originalRandom = Math.random;
+  Math.random = () => 0;
+
+  try {
+    assert.equal(random(), "93.123.23.1");
+    assert.equal(random("Bulgaria"), "93.123.23.1");
+    assert.equal(random("Unknown"), undefined);
+  } finally {
+    Math.random = originalRandom;
+  }
 });
 
 test("returns immutable data", () => {
@@ -41,3 +57,9 @@ test("returns immutable data", () => {
     (byCountry.Bulgaria as string[]).push("127.0.0.1");
   }, TypeError);
 });
+
+test("validates country input", () => {
+  assert.throws(() => {
+    getByCountry(123 as unknown as string);
+  }, TypeError);
+});