def _backend_setup(self, server=True, *args, **kwargs):
"""
Args:
server (bool): Whether this is the client or a server
Raises:
TurbiniaException: When there are errors creating PSQ Queue
"""
log.debug(
'Setting up PSQ Task Manager requirements on project {0:s}'.format(
config.TURBINIA_PROJECT))
self.server_pubsub = turbinia_pubsub.TurbiniaPubSub(
config.PUBSUB_TOPIC)
if server:
self.server_pubsub.setup_subscriber()
else:
self.server_pubsub.setup_publisher()
psq_publisher = pubsub.PublisherClient()
psq_subscriber = pubsub.SubscriberClient()
datastore_client = datastore.Client(project=config.TURBINIA_PROJECT)
try:
self.psq = psq.Queue(
psq_publisher,
psq_subscriber,
config.TURBINIA_PROJECT,
name=config.PSQ_TOPIC,
storage=psq.DatastoreStorage(datastore_client))
except exceptions.GoogleCloudError as e:
msg = 'Error creating PSQ Queue: {0:s}'.format(str(e))
log.error(msg)
raise turbinia.TurbiniaException(msg)
def setUp(self):
request = getTurbiniaRequest()
self.pubsub = pubsub.TurbiniaPubSub('fake_topic')
pub_sub_message = MockPubSubMessage(request.to_json(), 'msg id')
# pylint: disable=protected-access
self.pubsub._queue.put(pub_sub_message)
self.pubsub.topic_path = 'faketopicpath'
def setUp(self):
request = getTurbiniaRequest()
self.pubsub = pubsub.TurbiniaPubSub('fake_topic')
results = MockPubSubResults(ack_id='1234',
message=MockPubSubMessage(
request.to_json(), 'msg id'))
self.pubsub.subscription = mock.MagicMock()
self.pubsub.subscription.pull.return_value = results
def _backend_setup(self):
log.debug(
'Setting up PSQ Task Manager requirements on project {0:s}'.format(
config.PROJECT))
self.server_pubsub = turbinia_pubsub.TurbiniaPubSub(config.PUBSUB_TOPIC)
self.server_pubsub.setup()
psq_pubsub_client = pubsub.Client(project=config.PROJECT)
datastore_client = datastore.Client(project=config.PROJECT)
try:
self.psq = psq.Queue(
psq_pubsub_client,
config.PSQ_TOPIC,
storage=psq.DatastoreStorage(datastore_client))
except GaxError as e:
msg = 'Error creating PSQ Queue: {0:s}'.format(str(e))
log.error(msg)
raise turbinia.TurbiniaException(msg)
