From 409db4f2d121486d8a76ad726ff949725fdcef82 Mon Sep 17 00:00:00 2001
From: Ward Oosterlijnck <wardoosterlijnck@gmail.com>
Date: Thu, 11 Apr 2019 13:39:05 +1000
Subject: [PATCH] Improve useBeforeUnload story, #220

---
 src/__stories__/useBeforeUnload.story.tsx | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/__stories__/useBeforeUnload.story.tsx b/src/__stories__/useBeforeUnload.story.tsx
index 989ee946..28abe94d 100644
--- a/src/__stories__/useBeforeUnload.story.tsx
+++ b/src/__stories__/useBeforeUnload.story.tsx
@@ -1,14 +1,16 @@
 import {storiesOf} from '@storybook/react';
 import * as React from 'react';
-import {useBeforeUnload} from '..';
+import {useToggle, useBeforeUnload} from '..';
 import ShowDocs from './util/ShowDocs';
 
 const Demo = () => {
-  useBeforeUnload();
+  const [dirty, toggleDirty] = useToggle(false);
+  useBeforeUnload(dirty, 'You have unsaved changes, are you sure?');
 
   return (
     <div>
-      Try to reload or close tab
+      {dirty && 'Try to reload or close tab'}
+      <button onClick={() => toggleDirty()}>{dirty ? 'Disable' : 'Enable'}</button>
     </div>
   );
 };