EmptyOperator(task_id=task_id, dag=dag2)
        DAG.bulk_write_to_db([dag1, dag2], session=session)
        session.commit()
        session.expunge_all()
        stored_datasets = {x.uri: x for x in session.query(DatasetModel).all()}
        d1_orm = stored_datasets[d1.uri]
        d2_orm = stored_datasets[d2.uri]
        assert [x.dag_id for x in d1_orm.consuming_dags] == []
        assert set(
            session.query(
                TaskOutletDatasetReference.task_id,
                TaskOutletDatasetReference.dag_id,
                TaskOutletDatasetReference.dataset_id,
            )
            .filter(TaskOutletDatasetReference.dag_id.in_((dag_id1, dag_id2)))
            .all()
        ) == {(task_id, dag_id1, d2_orm.id)}
    def test_bulk_write_to_db_unorphan_datasets(self):
        """
        Datasets can lose their last reference and be orphaned, but then if a reference to them reappears, we
        need to un-orphan those datasets
        """
        with create_session() as session:
            
            
            dataset1 = Dataset(uri="ds1")
            dataset2 = Dataset(uri="ds2")
            session.add(DatasetModel(uri=dataset2.uri, is_orphaned=True))
            dataset3 = Dataset(uri="ds3")
            dataset4 = Dataset(uri="ds4")