def enableBooking(self, value):
"""Enable booking on the show
@type: Boolean
@param: Whether or not to enable booking
@rtype: show_pb2.ShowEnableBookingResponse
@return: Response is empty
"""
response = self.stub.EnableBooking(show_pb2.ShowEnableBookingRequest(
show=self.data, enabled=value),
timeout=Cuebot.Timeout)
return response
def testEnableBooking(self, getStubMock):
stubMock = mock.Mock()
stubMock.EnableBooking.return_value = show_pb2.ShowEnableBookingResponse()
getStubMock.return_value = stubMock
show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME))
show.enableBooking(TEST_ENABLE_VALUE)
stubMock.EnableBooking.assert_called_with(
show_pb2.ShowEnableBookingRequest(show=show.data, enabled=TEST_ENABLE_VALUE),
timeout=mock.ANY)
def enableBooking(self, value):
"""Enables or disables booking on the show.
:type value: bool
:param value: whether to enable booking
:rtype: show_pb2.ShowEnableBookingResponse
:return: response is empty
"""
response = self.stub.EnableBooking(show_pb2.ShowEnableBookingRequest(
show=self.data, enabled=value),
timeout=Cuebot.Timeout)
return response
