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())