def test_put_multi_no_batch_w_partial_key(self):
from gcloud.datastore.test_batch import _CommitResult
from gcloud.datastore.test_batch import _Entity
from gcloud.datastore.test_batch import _Key
entity = _Entity(foo=u'bar')
key = entity.key = _Key(self.DATASET_ID)
key._id = None
creds = object()
client = self._makeOne(credentials=creds)
client.connection._commit.append(_CommitResult(key))
result = client.put_multi([entity])
self.assertTrue(result is None)
self.assertEqual(len(client.connection._commit_cw), 1)
dataset_id, mutation, transaction_id = client.connection._commit_cw[0]
self.assertEqual(dataset_id, self.DATASET_ID)
inserts = list(mutation.insert_auto_id)
self.assertEqual(len(inserts), 1)
self.assertEqual(inserts[0].key, key.to_protobuf())
properties = list(inserts[0].property)
self.assertEqual(properties[0].name, 'foo')
self.assertEqual(properties[0].value.string_value, u'bar')
self.assertTrue(transaction_id is None)
def test_put_multi_no_batch_w_partial_key(self):
from gcloud.datastore.test_batch import _CommitResult
from gcloud.datastore.test_batch import _Entity
from gcloud.datastore.test_batch import _Key
entity = _Entity(foo=u'bar')
key = entity.key = _Key(self.DATASET_ID)
key._id = None
creds = object()
client = self._makeOne(credentials=creds)
client.connection._commit.append(_CommitResult(key))
result = client.put_multi([entity])
self.assertTrue(result is None)
self.assertEqual(len(client.connection._commit_cw), 1)
dataset_id, mutation, transaction_id = client.connection._commit_cw[0]
self.assertEqual(dataset_id, self.DATASET_ID)
inserts = list(mutation.insert_auto_id)
self.assertEqual(len(inserts), 1)
self.assertEqual(inserts[0].key, key.to_protobuf())
properties = list(inserts[0].property)
self.assertEqual(properties[0].name, 'foo')
self.assertEqual(properties[0].value.string_value, u'bar')
self.assertTrue(transaction_id is None)
def test_delete_multi_no_batch(self):
from gcloud.datastore.test_batch import _CommitResult
from gcloud.datastore.test_batch import _Key
key = _Key(self.DATASET_ID)
creds = object()
client = self._makeOne(credentials=creds)
client.connection._commit.append(_CommitResult())
result = client.delete_multi([key])
self.assertEqual(result, None)
self.assertEqual(len(client.connection._commit_cw), 1)
dataset_id, mutation, transaction_id = client.connection._commit_cw[0]
self.assertEqual(dataset_id, self.DATASET_ID)
self.assertEqual(list(mutation.delete), [key.to_protobuf()])
self.assertTrue(transaction_id is None)
def test_delete_multi_no_batch(self):
from gcloud.datastore.test_batch import _CommitResult
from gcloud.datastore.test_batch import _Key
key = _Key(self.DATASET_ID)
creds = object()
client = self._makeOne(credentials=creds)
client.connection._commit.append(_CommitResult())
result = client.delete_multi([key])
self.assertEqual(result, None)
self.assertEqual(len(client.connection._commit_cw), 1)
dataset_id, mutation, transaction_id = client.connection._commit_cw[0]
self.assertEqual(dataset_id, self.DATASET_ID)
self.assertEqual(list(mutation.delete), [key.to_protobuf()])
self.assertTrue(transaction_id is None)
