def testValidate_Passes(self):
"""Test validate function accepts valid parameters."""
params = {
"entity_kind": ENTITY_KIND,
}
mapper_spec = model.MapperSpec(
"FooHandler",
"djangoappengine.mapreduce.input_readers.DjangoModelInputReader",
params, 1)
DjangoModelInputReader.validate(mapper_spec)
def testValidate_Passes(self):
"""Test validate function accepts valid parameters."""
params = {
"entity_kind": ENTITY_KIND,
}
mapper_spec = model.MapperSpec(
"FooHandler",
"djangoappengine.mapreduce.input_readers.DjangoModelInputReader",
params, 1)
DjangoModelInputReader.validate(mapper_spec)
def testGeneratorWithKeyRange(self):
"""Test DjangoModelInputReader as generator using KeyRanges."""
expected_entities = []
for i in range(0, 100):
entity = TestModel(test_property=i)
entity.save()
expected_entities.append(entity)
params = {
"entity_kind": ENTITY_KIND,
}
mapper_spec = model.MapperSpec(
"FooHandler",
"djangoappengine.mapreduce.input_readers.DjangoModelInputReader",
params, 1)
input_ranges = DjangoModelInputReader.split_input(mapper_spec)
entities = []
for query_range in input_ranges:
for entity in query_range:
entities.append(entity)
self.assertEquals(100, len(entities))
self.assertEquals(expected_entities, entities)
def testGeneratorWithKeyRange(self):
"""Test DjangoModelInputReader as generator using KeyRanges."""
expected_entities = []
for i in range(0, 100):
entity = TestModel(test_property=i)
entity.save()
expected_entities.append(entity)
params = {
"entity_kind": ENTITY_KIND,
}
mapper_spec = model.MapperSpec(
"FooHandler",
"djangoappengine.mapreduce.input_readers.DjangoModelInputReader",
params, 1)
input_ranges = DjangoModelInputReader.split_input(mapper_spec)
entities = []
for query_range in input_ranges:
for entity in query_range:
entities.append(entity)
self.assertEquals(100, len(entities))
self.assertEquals(expected_entities, entities)
