Source code for gclouddatastore.test_entity

import unittest2

from gclouddatastore.dataset import Dataset
from gclouddatastore.entity import Entity
from gclouddatastore.key import Key


[docs]class TestEntity(unittest2.TestCase):
[docs] def test_init_sets_proper_values(self): dataset = Dataset(id='test-dataset') entity = Entity(dataset, 'TestKind') self.assertEqual('test-dataset', entity.dataset().id()) self.assertEqual('TestKind', entity.kind())
[docs] def test_key(self): dataset = Dataset(id='test-dataset') entity = Entity(dataset, 'TestKind') self.assertIsInstance(entity.key(), Key)
[docs] def test_from_key(self): key = Key(dataset=Dataset('test-dataset')).kind('TestKind').id(1234) entity = Entity.from_key(key, load_properties=False) self.assertEqual('test-dataset', entity.dataset().id()) self.assertEqual('TestKind', entity.key().kind()) self.assertEqual(entity.key().kind(), entity.kind()) raise Exception(entity.key().path()) self.assertEqual(1234, entity.key().id())