def test_createdb(): url = 'postgresql://*****:*****@localhost:5432/createdb' db = SQLAlchemyORM(url) db.createdb(True) assert db.existsdb() db.dropdb() assert not db.existsdb()
def db_connection(): db_url = 'postgresql://*****:*****@localhost:5432/flask_test' db = SQLAlchemyORM(db_url) if db.existsdb(): # Comment it for create database yield db_connection return db.create_all(True) session = db.create_session() repo1 = Repository('repo/terrame') sys1 = System('TerraME', repo1) session.add(repo1) session.add(sys1) miner = RepositoryMiner(repo1, sys1) miner.add_ignore_dir_with('test') miner.add_ignore_dir_with('examples') miner.add_ignore_dir_with('ide') miner.add_ignore_dir_with('data') miner.extract_last_commits(session) repo2 = Repository('repo/ca') sys2 = System('CA', repo2) miner = RepositoryMiner(repo2, sys2) miner.extract_last_commits(session) ecosystem = Ecosystem() ecolyzer = EcosystemAnalyzer(ecosystem) ecolyzer.make_relations(sys2, sys1, session) session.close() yield db_connection db.drop_all()