def test__get(self):
foreman = FakeForeman()
a_worker = worker.MappingManagingWorker(foreman,
managed_sqla_map=FakeMapping,
managed_sqla_map_name="fake")
a_worker._get(sqla_obj=FakeMapping())
def test_init_without_id_or_uuid(self):
foreman = FakeForeman()
a_worker = worker.MappingManagingWorker(foreman,
managed_sqla_map=FakeMapping,
managed_sqla_map_name="fake")
self.assertFalse(a_worker._with_id)
self.assertFalse(a_worker._with_uuid)
def test_init_forced(self):
foreman = FakeForeman()
a_worker = worker.MappingManagingWorker(foreman,
managed_sqla_map=FakeMapping,
managed_sqla_map_name="fake",
id_type="uuid")
self.assertTrue(a_worker._with_uuid)
self.assertFalse(a_worker._with_id)
a_second_worker = worker.MappingManagingWorker(
foreman,
managed_sqla_map=FakeMapping,
managed_sqla_map_name="fake",
id_type="id")
self.assertTrue(a_second_worker._with_id)
self.assertFalse(a_second_worker._with_uuid)
with self.assertRaises(AttributeError):
worker.MappingManagingWorker(foreman,
managed_sqla_map=FakeMapping,
managed_sqla_map_name="fake",
id_type="wrong")
def setUp(self):
engine = create_engine("sqlite:///:memory:", echo=False)
Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
dbsession = Session()
a_foreman = foreman.RawForeman(dbsession=dbsession)
self.worker = worker.MappingManagingWorker(
a_foreman,
managed_sqla_map=Mapping,
managed_sqla_map_name="mapping")
def build_query(mapping):
return a_foreman._dbsession.query(mapping)
self.build_query = build_query
def test_init(self):
foreman = FakeForeman()
worker.MappingManagingWorker(foreman,
managed_sqla_map=FakeMapping,
managed_sqla_map_name="fake")