fix(firstMile): save bucket selection to session (#4501)

Author: ChitlangeSahas

src/homepageExperience/components/steps/go/WriteData.tsx

@@ -11,9 +11,7 @@ import {
 import {SafeBlankLink} from 'src/utils/SafeBlankLink'
 import WriteDataHelperBuckets from 'src/writeData/components/WriteDataHelperBuckets'
 import CodeSnippet from 'src/shared/components/CodeSnippet'
-import WriteDataDetailsContextProvider, {
-  WriteDataDetailsContext,
-} from 'src/writeData/components/WriteDataDetailsContext'
+import {WriteDataDetailsContext} from 'src/writeData/components/WriteDataDetailsContext'
 
 import {getOrg} from 'src/organizations/selectors'
 import DataListening from 'src/homepageExperience/components/DataListening'
@@ -130,9 +128,5 @@ for value := 0; value < 5; value++ {
 }
 
 export const WriteData = props => {
-  return (
-    <WriteDataDetailsContextProvider>
-      <WriteDataComponent onSelectBucket={props.onSelectBucket} />
-    </WriteDataDetailsContextProvider>
-  )
+  return <WriteDataComponent onSelectBucket={props.onSelectBucket} />
 }

src/homepageExperience/components/steps/nodejs/WriteData.tsx

@@ -11,9 +11,7 @@ import {
 import {SafeBlankLink} from 'src/utils/SafeBlankLink'
 import WriteDataHelperBuckets from 'src/writeData/components/WriteDataHelperBuckets'
 import CodeSnippet from 'src/shared/components/CodeSnippet'
-import WriteDataDetailsContextProvider, {
-  WriteDataDetailsContext,
-} from 'src/writeData/components/WriteDataDetailsContext'
+import {WriteDataDetailsContext} from 'src/writeData/components/WriteDataDetailsContext'
 
 import {getOrg} from 'src/organizations/selectors'
 import DataListening from 'src/homepageExperience/components/DataListening'
@@ -129,9 +127,5 @@ for (let i = 0; i < 5; i++) {
 }
 
 export const WriteData = props => {
-  return (
-    <WriteDataDetailsContextProvider>
-      <WriteDataComponent onSelectBucket={props.onSelectBucket} />
-    </WriteDataDetailsContextProvider>
-  )
+  return <WriteDataComponent onSelectBucket={props.onSelectBucket} />
 }

src/homepageExperience/components/steps/python/WriteData.tsx

@@ -11,9 +11,7 @@ import {
 import {SafeBlankLink} from 'src/utils/SafeBlankLink'
 import WriteDataHelperBuckets from 'src/writeData/components/WriteDataHelperBuckets'
 import CodeSnippet from 'src/shared/components/CodeSnippet'
-import WriteDataDetailsContextProvider, {
-  WriteDataDetailsContext,
-} from 'src/writeData/components/WriteDataDetailsContext'
+import {WriteDataDetailsContext} from 'src/writeData/components/WriteDataDetailsContext'
 
 import {getOrg} from 'src/organizations/selectors'
 import DataListening from 'src/homepageExperience/components/DataListening'
@@ -124,9 +122,5 @@ for value in range(5):
 }
 
 export const WriteData = props => {
-  return (
-    <WriteDataDetailsContextProvider>
-      <WriteDataComponent onSelectBucket={props.onSelectBucket} />
-    </WriteDataDetailsContextProvider>
-  )
+  return <WriteDataComponent onSelectBucket={props.onSelectBucket} />
 }

src/homepageExperience/containers/GoWizard.tsx

@@ -18,6 +18,7 @@ import {WriteData} from 'src/homepageExperience/components/steps/go/WriteData'
 import {ExecuteQuery} from 'src/homepageExperience/components/steps/go/ExecuteQuery'
 import {Finish} from 'src/homepageExperience/components/steps/Finish'
 import {ExecuteAggregateQuery} from 'src/homepageExperience/components/steps/go/ExecuteAggregateQuery'
+import WriteDataDetailsContextProvider from 'src/writeData/components/WriteDataDetailsContext'
 
 import {GoIcon} from 'src/homepageExperience/components/HomepageIcons'
 
@@ -167,7 +168,9 @@ export class GoWizard extends PureComponent<null, State> {
                   }
                 )}
               >
-                {this.renderStep()}
+                <WriteDataDetailsContextProvider>
+                  {this.renderStep()}
+                </WriteDataDetailsContextProvider>{' '}
               </div>
 
               <div className="homepage-wizard-container-footer">

src/homepageExperience/containers/NodejsWizard.tsx

@@ -18,6 +18,7 @@ import {WriteData} from 'src/homepageExperience/components/steps/nodejs/WriteDat
 import {ExecuteQuery} from 'src/homepageExperience/components/steps/nodejs/ExecuteQuery'
 import {Finish} from 'src/homepageExperience/components/steps/Finish'
 import {ExecuteAggregateQuery} from 'src/homepageExperience/components/steps/nodejs/ExecuteAggregateQuery'
+import WriteDataDetailsContextProvider from 'src/writeData/components/WriteDataDetailsContext'
 
 import {NodejsIcon} from 'src/homepageExperience/components/HomepageIcons'
 
@@ -168,7 +169,9 @@ export class NodejsWizard extends PureComponent<null, State> {
                   }
                 )}
               >
-                {this.renderStep()}
+                <WriteDataDetailsContextProvider>
+                  {this.renderStep()}
+                </WriteDataDetailsContextProvider>
               </div>
 
               <div className="homepage-wizard-container-footer">

src/homepageExperience/containers/PythonWizard.tsx

@@ -10,6 +10,8 @@ import {
   SubwayNav,
 } from '@influxdata/clockface'
 
+import WriteDataDetailsContextProvider from 'src/writeData/components/WriteDataDetailsContext'
+
 import {InstallDependencies} from 'src/homepageExperience/components/steps/python/InstallDependencies'
 import {Overview} from 'src/homepageExperience/components/steps/Overview'
 import {Tokens} from 'src/homepageExperience/components/steps/Tokens'
@@ -169,7 +171,9 @@ export class PythonWizard extends PureComponent<null, State> {
                   }
                 )}
               >
-                {this.renderStep()}
+                <WriteDataDetailsContextProvider>
+                  {this.renderStep()}
+                </WriteDataDetailsContextProvider>
               </div>
 
               <div className="homepage-wizard-container-footer">