How to use the body-scroll-lock.disableBodyScroll.mock function in body-scroll-lock

it('adds body scroll lock when shouldLockBgScroll changes to true', () => {
      component = shallow(<Dialog {...requiredProps} shouldLockBgScroll />);
      component.instance().componentDidUpdate({ shouldLockBgScroll: false });
      expect(disableBodyScroll.mock.calls.length).toBe(1);
    });

it('adds body scroll lock when transitionState changes to "entered" and shouldLockBgScroll is true', () => {
      component = shallow(
        <Dialog
          {...requiredProps}
          shouldLockBgScroll
          transitionState="entered"
        />
      );
      component.instance().componentDidUpdate({
        shouldLockBgScroll: true,
        transitionState: 'entering'
      });
      expect(disableBodyScroll.mock.calls.length).toBe(1);
    });
  });