All files / modules/85-cv/hooks/useErrorBudgetRestHook useErrorBudgetRestHook.tsx

100% Statements 17/17
100% Branches 2/2
100% Functions 4/4
100% Lines 16/16

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53              10x 10x 10x 10x   10x     10x 58x 58x 58x   2x                     2x 2x                   58x 2x 2x     58x          
/*
 * Copyright 2022 Harness Inc. All rights reserved.
 * Use of this source code is governed by the PolyForm Shield 1.0.0 license
 * that can be found in the licenses directory at the root of this repository, also available at
 * https://polyformproject.org/wp-content/uploads/2020/06/PolyForm-Shield-1.0.0.txt.
 */
 
import React, { useState } from 'react'
import { Dialog } from '@harness/uicore'
import { useModalHook } from '@harness/use-modal'
import { useStrings } from 'framework/strings'
import type { SLODashboardWidget } from 'services/cv'
import ErrorBudgetResetForm from './views/ErrorBudgetResetForm'
import type { UseErrorBudgetRestHookProps, UseErrorBudgetRestHookReturn } from './useErrorBudgetRestHook.types'
 
export const useErrorBudgetRestHook = ({ onSuccess }: UseErrorBudgetRestHookProps): UseErrorBudgetRestHookReturn => {
  const { getString } = useStrings()
  const [serviceLevelObjective, setServiceLevelObjective] = useState<SLODashboardWidget>()
  const [showModal, hideModal] = useModalHook(
    () => (
      <Dialog
        isOpen
        enforceFocus={false}
        title={getString('cv.resetErrorBudget')}
        onClose={hideModal}
        style={{ minWidth: 'max-content' }}
      >
        {serviceLevelObjective && (
          <ErrorBudgetResetForm
            serviceLevelObjective={serviceLevelObjective}
            onSubmit={formData => {
              onSuccess(formData)
              hideModal()
            }}
            onCancel={hideModal}
          />
        )}
      </Dialog>
    ),
    [serviceLevelObjective]
  )
 
  const open = (_serviceLevelObjective: SLODashboardWidget): void => {
    setServiceLevelObjective(_serviceLevelObjective)
    showModal()
  }
 
  return {
    openErrorBudgetReset: open,
    closeErrorBudgetReset: hideModal
  }
}