def _data_process_client(_mock_bigquery_obj): from rudra.data_store.bigquery.pypi_bigquery import PyPiBigQuery, PyPiBigQueryDataProcessing _pypi_ins = PyPiBigQuery() s3_client = MockS3(tempfile.mkdtemp()) _pypi_ins.query = "select id, name, content from manifests\ where name like '%requirements.txt'" _client = PyPiBigQueryDataProcessing(_pypi_ins, s3_client=s3_client) return _client, s3_client
def _pypi_bigquery_client(_mock_bigquery_obj): from rudra.data_store.bigquery.pypi_bigquery import PyPiBigQuery _client = PyPiBigQuery() _client.query = "select id, name, content from manifests where name like '%requirements.txt'" return _client