def get_data():
"""
Return some ParameterData
"""
from aiida.orm import DataFactory
data = DataFactory('parameter')(dict={'data': 'test'})
data.store()
return data
def test_create_use_destroy_profile(self):
"""
Test temporary test profile creation
* The profile gets created, the dbenv loaded
* Data can be stored in the db
* reset_db deletes all data added after profile creation
* destroy_all removes all traces of the test run
"""
from aiida.common import setup as aiida_cfg
from aiida import is_dbenv_loaded
with Capturing() as output:
self.fixture_manager.create_profile()
self.assertTrue(self.fixture_manager.root_dir_ok, msg=output)
self.assertTrue(self.fixture_manager.config_dir_ok, msg=output)
self.assertTrue(self.fixture_manager.repo_ok, msg=output)
self.assertEqual(aiida_cfg.AIIDA_CONFIG_FOLDER,
self.fixture_manager.config_dir,
msg=output)
self.assertTrue(is_dbenv_loaded())
from aiida.orm import DataFactory, load_node
data = DataFactory('parameter')(dict={'key': 'value'})
data.store()
data_pk = data.pk
self.assertTrue(load_node(data_pk))
with self.assertRaises(FixtureError):
self.test_create_aiida_db()
self.fixture_manager.reset_db()
with self.assertRaises(Exception):
load_node(data_pk)
temp_dir = self.fixture_manager.root_dir
self.fixture_manager.destroy_all()
with self.assertRaises(Exception):
self.fixture_manager.postgres.db_exists(
self.fixture_manager.db_name)
self.assertFalse(os.path.exists(temp_dir))
self.assertIsNone(self.fixture_manager.root_dir)
self.assertIsNone(self.fixture_manager.pg_cluster)
