Source code for gclouddatastore.test_dataset

import unittest2

from gclouddatastore.dataset import Dataset
from gclouddatastore.entity import Entity
from gclouddatastore.query import Query


[docs]class TestDataset(unittest2.TestCase):
[docs] def test_init_id_required(self): with self.assertRaises(Exception): dataset = Dataset() dataset = Dataset('dataset-id-here') self.assertEqual('dataset-id-here', dataset.id()) self.assertEqual(None, dataset.connection())
[docs] def test_query_factory(self): dataset = Dataset('test') query = dataset.query() self.assertIsInstance(query, Query) self.assertEqual(dataset, query.dataset())
[docs] def test_entity_factory(self): dataset = Dataset('test') entity = dataset.entity('TestKind') self.assertIsInstance(entity, Entity) self.assertEqual('TestKind', entity.kind())