From 0d5fbb6b936b3c01bfd19cb3ee3986527312bfc8 Mon Sep 17 00:00:00 2001
From: "Kent C. Dodds" <me+github@kentcdodds.com>
Date: Wed, 4 Mar 2020 12:00:19 -0700
Subject: [PATCH] feat(debug): accept the same arguments as prettyDOM

---
 src/__tests__/debug.js | 13 +++++++++++++
 src/pure.js            |  6 +++---
 2 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/src/__tests__/debug.js b/src/__tests__/debug.js
index ad3b8591..48411d88 100644
--- a/src/__tests__/debug.js
+++ b/src/__tests__/debug.js
@@ -36,4 +36,17 @@ test('debug pretty prints multiple containers', () => {
   )
 })
 
+test('allows same arguments as prettyDOM', () => {
+  const HelloWorld = () => <h1>Hello World</h1>
+  const {debug, container} = render(<HelloWorld />)
+  debug(container, 6, {highlight: false})
+  expect(console.log).toHaveBeenCalledTimes(1)
+  expect(console.log.mock.calls[0]).toMatchInlineSnapshot(`
+    Array [
+      "<div>
+    ...",
+    ]
+  `)
+})
+
 /* eslint no-console:0 */
diff --git a/src/pure.js b/src/pure.js
index 40588797..9ed80e1e 100644
--- a/src/pure.js
+++ b/src/pure.js
@@ -60,12 +60,12 @@ function render(
   return {
     container,
     baseElement,
-    debug: (el = baseElement) =>
+    debug: (el = baseElement, maxLength, options) =>
       Array.isArray(el)
         ? // eslint-disable-next-line no-console
-          el.forEach(e => console.log(prettyDOM(e)))
+          el.forEach(e => console.log(prettyDOM(e, maxLength, options)))
         : // eslint-disable-next-line no-console,
-          console.log(prettyDOM(el)),
+          console.log(prettyDOM(el, maxLength, options)),
     unmount: () => ReactDOM.unmountComponentAtNode(container),
     rerender: rerenderUi => {
       render(wrapUiIfNeeded(rerenderUi), {container, baseElement})