add support for search prop (#120)

.changeset/sour-buttons-check.md

@@ -0,0 +1,5 @@
+---
+"next-router-mock": patch
+---
+
+Add support for parsing `search` parameter for URLs

src/MemoryRouter.test.tsx

@@ -417,6 +417,55 @@ describe("MemoryRouter", () => {
           });
         });
 
+        describe("search parameter", () => {
+          it("happy path", async () => {
+            await memoryRouter.push({
+              pathname: "/path",
+              search: "foo=FOO&bar=BAR",
+            });
+            expectMatch(memoryRouter, {
+              asPath: "/path?foo=FOO&bar=BAR",
+              pathname: "/path",
+              query: {
+                foo: "FOO",
+                bar: "BAR",
+              },
+            });
+          });
+
+          it("multiple values can be specified for a query parameter", async () => {
+            await memoryRouter.push({
+              pathname: "/path",
+              search: "foo=FOO&foo=BAR",
+            });
+            expectMatch(memoryRouter, {
+              asPath: "/path?foo=FOO&foo=BAR",
+              pathname: "/path",
+              query: {
+                foo: ["FOO", "BAR"],
+              },
+            });
+          });
+
+          it("if search and query are both provided preference is given to search", async () => {
+            await memoryRouter.push({
+              pathname: "/path",
+              search: "foo=FOO&bar=BAR",
+              query: {
+                baz: "BAZ",
+              },
+            });
+            expectMatch(memoryRouter, {
+              asPath: "/path?foo=FOO&bar=BAR",
+              pathname: "/path",
+              query: {
+                foo: "FOO",
+                bar: "BAR",
+              },
+            });
+          });
+        });
+
         describe("with different paths", () => {
           it("the real path and query are used", async () => {
             await memoryRouter.push("/real-path", "/as-path");

src/MemoryRouter.tsx

@@ -1,12 +1,13 @@
 import type { NextRouter, RouterEvent } from "next/router";
 import mitt, { MittEmitter } from "./lib/mitt";
-import { parseUrl, stringifyQueryString } from "./urls";
+import { parseUrl, parseQueryString, stringifyQueryString } from "./urls";
 
 export type Url = string | UrlObject;
 export type UrlObject = {
   pathname?: string | null | undefined;
   query?: NextRouter["query"];
   hash?: string;
+  search?: string;
 };
 export type UrlObjectComplete = {
   pathname: string;
@@ -210,9 +211,13 @@ export class MemoryRouter extends BaseRouter {
  */
 function parseUrlToCompleteUrl(url: Url, currentPathname: string): UrlObjectComplete {
   const parsedUrl = typeof url === "object" ? url : parseUrl(url);
+
+  const queryFromSearch = parsedUrl.search ? parseQueryString(parsedUrl.search) : undefined;
+  const query = queryFromSearch ?? parsedUrl.query ?? {};
+
   return {
     pathname: normalizeTrailingSlash(parsedUrl.pathname ?? currentPathname),
-    query: parsedUrl.query || {},
+    query,
     hash: parsedUrl.hash || "",
     routeParams: {},
   };

src/urls.ts

@@ -26,3 +26,8 @@ export function stringifyQueryString(query: NextRouter["query"]): string {
   });
   return params.toString();
 }
+export function parseQueryString(query: string): NextRouter["query"] | undefined {
+  const parsedUrl = parseUrl(`?${query}`);
+
+  return parsedUrl.query;
+}