From c0b6319ccf49e2ac3107bfb906456489ab30f9d4 Mon Sep 17 00:00:00 2001
From: cwmoo740 <cwmoo740@gmail.com>
Date: Thu, 9 Mar 2017 15:22:54 -0500
Subject: [PATCH] react-day-picker update tests to include new use of Element
 or React.SFC for captionElement

---
 react-day-picker/react-day-picker-tests.tsx | 23 ++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/react-day-picker/react-day-picker-tests.tsx b/react-day-picker/react-day-picker-tests.tsx
index deecb3e21a..8c2755f9b5 100644
--- a/react-day-picker/react-day-picker-tests.tsx
+++ b/react-day-picker/react-day-picker-tests.tsx
@@ -8,7 +8,7 @@ function isSunday(day: Date) {
 function MyComponent() {
     return (
         <DayPicker
-            ref="foo" initialMonth={ new Date(2016, 1) } localeUtils={DayPicker.LocaleUtils} modifiers={{ isSunday }}
+            initialMonth={ new Date(2016, 1) } localeUtils={DayPicker.LocaleUtils} modifiers={{ isSunday }}
         />
     );
 }
@@ -26,10 +26,27 @@ class Caption extends React.Component<MyCaptionProps, {}> {
         }
 
         return (
-            <div className="DayPicker-Caption" onClick={ (e) => onClick(date, e) }>
+            <div className="DayPicker-Caption" onClick={ onClick }>
                 { localeUtils.formatMonthTitle(date, locale) }
             </div>
         );
     }
 }
-<DayPicker captionElement={<Caption/>}/>
+<DayPicker captionElement={ Caption }/>
+
+
+type CaptionElementProps = Partial<DayPicker.CaptionElementProps>;
+class CaptionElement extends React.Component<CaptionElementProps, {}> {
+    render() {
+        const { date, locale, localeUtils, onClick } = this.props;
+        if (!date || !locale || !localeUtils || !onClick) {
+            return <div/>;
+        }
+        return (
+            <div onClick={ onClick }>
+                { localeUtils.formatMonthTitle(date, locale) }
+            </div>
+        )
+    }
+}
+<DayPicker captionElement={ <CaptionElement /> }/>