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