def test_suspend_vendor_ok_response(self, mock_vendor_repo_update,
                                        mock_vendor_repo_get):
        '''Test suspend_vendor OK response.
        '''
        # Arrange
        with self.app.app_context():
            mock_vendor_repo_get.return_value = self.mock_vendor
            mock_vendor_repo_update.return_value = self.mock_vendor
            vendor_controller = VendorController(self.request_context)

            # Act
            result = vendor_controller.suspend_vendor(1)

            # Assert
            assert result.status_code == 200
            assert result.get_json()['msg'] == 'OK'
    def test_suspend_vendor_when_vendor_doesnot_exist(self,
                                                      mock_vendor_repo_get):
        '''Test suspend_vendor when vendor doesnot exist.
        '''
        # Arrange
        with self.app.app_context():
            mock_vendor_repo_get.return_value = None
            vendor_controller = VendorController(self.request_context)

            # Act
            result = vendor_controller.suspend_vendor(1)

            # Assert
            assert result.status_code == 400
            assert result.get_json()['msg'] == 'Invalid or incorrect ' \
                'vendor_id provided'