def test__connection_getter_with_batch(self): from google.cloud.storage.batch import Batch PROJECT = 'PROJECT' CREDENTIALS = _make_credentials() client = self._make_one(project=PROJECT, credentials=CREDENTIALS) batch = Batch(client) client._push_batch(batch) self.assertIsNot(client._connection, client._base_connection) self.assertIs(client._connection, batch) self.assertIs(client.current_batch, batch)
def test_connection_getter_with_batch(self): from google.cloud.storage.batch import Batch PROJECT = 'PROJECT' CREDENTIALS = _Credentials() client = self._makeOne(project=PROJECT, credentials=CREDENTIALS) batch = Batch(client) client._push_batch(batch) self.assertTrue(client.connection is not client._connection) self.assertTrue(client.connection is batch) self.assertTrue(client.current_batch is batch)
def test__push_batch_and__pop_batch(self): from google.cloud.storage.batch import Batch PROJECT = 'PROJECT' CREDENTIALS = _make_credentials() client = self._make_one(project=PROJECT, credentials=CREDENTIALS) batch1 = Batch(client) batch2 = Batch(client) client._push_batch(batch1) self.assertEqual(list(client._batch_stack), [batch1]) self.assertIs(client.current_batch, batch1) client._push_batch(batch2) self.assertIs(client.current_batch, batch2) # list(_LocalStack) returns in reverse order. self.assertEqual(list(client._batch_stack), [batch2, batch1]) self.assertIs(client._pop_batch(), batch2) self.assertEqual(list(client._batch_stack), [batch1]) self.assertIs(client._pop_batch(), batch1) self.assertEqual(list(client._batch_stack), [])
def batch(self): """Factory constructor for batch object. .. note:: This will not make an HTTP request; it simply instantiates a batch object owned by this client. :rtype: :class:`google.cloud.storage.batch.Batch` :returns: The batch object created. """ return Batch(client=self)
def test_duplicate_user_agent(self): # Regression test for issue #565 from google.cloud._http import ClientInfo from google.cloud.storage.batch import Batch from google.cloud.storage import __version__ client_info = ClientInfo(user_agent="test/123") conn = self._make_one(object(), client_info=client_info) expected_user_agent = "test/123 gcloud-python/{} ".format(__version__) self.assertEqual(conn._client_info.user_agent, expected_user_agent) client = mock.Mock(_connection=conn, spec=["_connection"]) batch = Batch(client) self.assertEqual(batch._client_info.user_agent, expected_user_agent)