diff --git a/integrationtests/test_data_model.py b/integrationtests/test_data_model.py index 6f530719a810d76e5cc5a2c59fcd2d0325ff5268..2949fa81727a6c61a8646a48c249204fa87542d8 100644 --- a/integrationtests/test_data_model.py +++ b/integrationtests/test_data_model.py @@ -33,13 +33,6 @@ class DataModelTest(unittest.TestCase): rt = db.execute_query("FIND RECORDTYPE TestRecord", unique=True) assert rt.get_property("test") is not None - def tearDown(self): - try: - tests = db.execute_query("FIND test*") - tests.delete() - except Exception: - pass - def test_missing(self): # Test sync with missing prop # insert propt @@ -52,3 +45,19 @@ class DataModelTest(unittest.TestCase): dm.sync_data_model(noquestion=True) rt = db.execute_query("FIND RECORDTYPE TestRecord", unique=True) assert rt.get_property("testproperty") is not None + + def test_get_existing_entities(self): + db.RecordType(name="TestRecord").insert() + c = db.Container().extend([ + db.Property(name="test"), + db.RecordType(name="TestRecord")]) + exist = DataModel.get_existing_entities(c) + assert len(exist) == 1 + assert exist[0].name == "TestRecord" + + def tearDown(self): + try: + tests = db.execute_query("FIND test*") + tests.delete() + except Exception: + pass diff --git a/unittests/test_data_model.py b/unittests/test_data_model.py index 9be23c3188e3c3244112c4195c40ebd475f4d492..159adfca1d589bb092b6f59110828b5868401e25 100644 --- a/unittests/test_data_model.py +++ b/unittests/test_data_model.py @@ -1,19 +1,11 @@ import unittest import caosdb as db -import pytest from caosadvancedtools.models.data_model import DataModel class DataModelTest(unittest.TestCase): - def tearDown(self): - try: - tests = db.execute_query("FIND test*") - tests.delete() - except Exception: - pass - def test_collecting(self): maintained = {"one": db.RecordType(name="TestRecord").add_property( name="testproperty"), @@ -24,17 +16,6 @@ class DataModelTest(unittest.TestCase): assert "TestRecord" in names assert "testproperty" in names - # TODO this seems to require integration test - @pytest.mark.xfail(reason="This test will need an integration test. See issue #51") - def test_get_existing_entities(self): - db.RecordType(name="TestRecord").insert() - c = db.Container().extend([ - db.Property(name="testproperty"), - db.RecordType(name="TestRecord")]) - exist = DataModel.get_existing_entities(c) - assert len(exist) == 1 - assert exist[0].name == "TestRecord" - def test_sync_ids_by_name(self): container = db.Container().extend([db.RecordType(name="TestRecord"), db.RecordType(name="TestRecord2"),