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)