def testDelete(self, getStubMock): stubMock = mock.Mock() stubMock.Delete.return_value = filter_pb2.FilterDeleteResponse() getStubMock.return_value = stubMock filterToDelete = opencue.wrappers.filter.Filter(filter_pb2.Filter(name=TEST_FILTER_NAME)) filterToDelete.delete() stubMock.Delete.assert_called_with( filter_pb2.FilterDeleteRequest(filter=filterToDelete.data), timeout=mock.ANY)
def delete(self): """Deletes the filter""" self.stub.Delete(filter_pb2.FilterDeleteRequest(filter=self.data), timeout=Cuebot.Timeout)