@@ -2,12 +2,12 @@ import { action } from "@storybook/addon-actions"
22import { Story } from "@storybook/react"
33import { WatchAgentMetadataContext } from "components/Resources/AgentMetadata"
44import { ProvisionerJobLog } from "api/typesGenerated"
5- import * as Mocks from "../../ testHelpers/entities"
5+ import * as Mocks from "testHelpers/entities"
66import { Workspace , WorkspaceErrors , WorkspaceProps } from "./Workspace"
77import { withReactContext } from "storybook-react-context"
88import EventSource from "eventsourcemock"
99import { ProxyContext , getPreferredProxy } from "contexts/ProxyContext"
10- import { MockProxyLatencies } from "../../testHelpers/entities "
10+ import { DashboardProviderContext } from "components/Dashboard/DashboardProvider "
1111
1212export default {
1313title :"components/Workspace" ,
@@ -24,21 +24,37 @@ export default {
2424] ,
2525}
2626
27+ const MockedAppearance = {
28+ config :Mocks . MockAppearance ,
29+ preview :false ,
30+ setPreview :( ) => { } ,
31+ save :( ) => { } ,
32+ }
33+
2734const Template :Story < WorkspaceProps > = ( args ) => (
28- < ProxyContext . Provider
35+ < DashboardProviderContext . Provider
2936value = { {
30- proxyLatencies :MockProxyLatencies ,
31- proxy :getPreferredProxy ( [ ] , undefined ) ,
32- proxies :[ ] ,
33- isLoading :false ,
34- isFetched :true ,
35- setProxy :( ) => {
36- return
37- } ,
37+ buildInfo :Mocks . MockBuildInfo ,
38+ entitlements :Mocks . MockEntitlementsWithScheduling ,
39+ experiments :Mocks . MockExperiments ,
40+ appearance :MockedAppearance ,
3841} }
3942>
40- < Workspace { ...args } />
41- </ ProxyContext . Provider >
43+ < ProxyContext . Provider
44+ value = { {
45+ proxyLatencies :Mocks . MockProxyLatencies ,
46+ proxy :getPreferredProxy ( [ ] , undefined ) ,
47+ proxies :[ ] ,
48+ isLoading :false ,
49+ isFetched :true ,
50+ setProxy :( ) => {
51+ return
52+ } ,
53+ } }
54+ >
55+ < Workspace { ...args } />
56+ </ ProxyContext . Provider >
57+ </ DashboardProviderContext . Provider >
4258)
4359
4460export const Running = Template . bind ( { } )