mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-13 10:57:49 +00:00
Dion Moult
127 lines
5.3 KiB
Python
127 lines
5.3 KiB
Python
# Bonsai - OpenBIM Blender Add-on
|
|
# Copyright (C) 2021 Dion Moult <dion@thinkmoult.com>
|
|
#
|
|
# This file is part of Bonsai.
|
|
#
|
|
# Bonsai is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Bonsai is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Bonsai. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import bonsai.core.library as subject
|
|
from test.core.bootstrap import ifc, library
|
|
|
|
|
|
class TestAddLibrary:
|
|
def test_run(self, ifc):
|
|
ifc.run("library.add_library", name="Unnamed").should_be_called().will_return("library")
|
|
assert subject.add_library(ifc) == "library"
|
|
|
|
|
|
class TestRemoveLibrary:
|
|
def test_run(self, ifc):
|
|
ifc.run("library.remove_library", library="library").should_be_called()
|
|
subject.remove_library(ifc, library="library")
|
|
|
|
|
|
class TestEnableEditingLibraryReferences:
|
|
def test_run(self, library):
|
|
library.set_editing_mode("REFERENCES").should_be_called()
|
|
library.set_active_library("library").should_be_called()
|
|
library.import_references("library").should_be_called()
|
|
subject.enable_editing_library_references(library, library="library")
|
|
|
|
|
|
class TestDisableEditingLibraryReferences:
|
|
def test_run(self, library):
|
|
library.clear_editing_mode().should_be_called()
|
|
library.set_active_library(None).should_be_called()
|
|
subject.disable_editing_library_references(library)
|
|
|
|
|
|
class TestEnableEditingLibrary:
|
|
def test_run(self, library):
|
|
library.set_editing_mode("LIBRARY").should_be_called()
|
|
library.get_active_library().should_be_called().will_return("library")
|
|
library.import_library_attributes("library").should_be_called()
|
|
subject.enable_editing_library(library)
|
|
|
|
|
|
class TestDisableEditingLibrary:
|
|
def test_run(self, library):
|
|
library.set_editing_mode("REFERENCES").should_be_called()
|
|
subject.disable_editing_library(library)
|
|
|
|
|
|
class TestEditLibrary:
|
|
def test_run(self, ifc, library):
|
|
library.set_editing_mode("REFERENCES").should_be_called()
|
|
library.get_active_library().should_be_called().will_return("library")
|
|
library.export_library_attributes().should_be_called().will_return("attributes")
|
|
ifc.run("library.edit_library", library="library", attributes="attributes").should_be_called()
|
|
library.import_references("library").should_be_called()
|
|
subject.edit_library(ifc, library)
|
|
|
|
|
|
class TestAddLibraryReference:
|
|
def test_run(self, ifc, library):
|
|
library.get_active_library().should_be_called().will_return("library")
|
|
ifc.run("library.add_reference", library="library").should_be_called()
|
|
library.import_references("library").should_be_called()
|
|
subject.add_library_reference(ifc, library)
|
|
|
|
|
|
class TestRemoveLibraryReference:
|
|
def test_run(self, ifc, library):
|
|
ifc.run("library.remove_reference", reference="reference").should_be_called()
|
|
library.get_active_library().should_be_called().will_return("library")
|
|
library.import_references("library").should_be_called()
|
|
subject.remove_library_reference(ifc, library, reference="reference")
|
|
|
|
|
|
class TestEnableEditingLibraryReference:
|
|
def test_run(self, library):
|
|
library.set_editing_mode("REFERENCE").should_be_called()
|
|
library.set_active_reference("reference").should_be_called()
|
|
library.import_reference_attributes("reference").should_be_called()
|
|
subject.enable_editing_library_reference(library, reference="reference")
|
|
|
|
|
|
class TestDisableEditingLibraryReference:
|
|
def test_run(self, library):
|
|
library.set_editing_mode("REFERENCES").should_be_called()
|
|
subject.disable_editing_library_reference(library)
|
|
|
|
|
|
class TestEditLibraryReference:
|
|
def test_run(self, ifc, library):
|
|
library.set_editing_mode("REFERENCES").should_be_called()
|
|
library.get_active_reference().should_be_called().will_return("reference")
|
|
library.export_reference_attributes().should_be_called().will_return("attributes")
|
|
ifc.run("library.edit_reference", reference="reference", attributes="attributes").should_be_called()
|
|
library.get_active_library().should_be_called().will_return("library")
|
|
library.import_references("library").should_be_called()
|
|
subject.edit_library_reference(ifc, library)
|
|
|
|
|
|
class TestAssignLibraryReference:
|
|
def test_run(self, ifc):
|
|
ifc.get_entity("obj").should_be_called().will_return("product")
|
|
ifc.run("library.assign_reference", products=["product"], reference="reference").should_be_called()
|
|
subject.assign_library_reference(ifc, obj="obj", reference="reference")
|
|
|
|
|
|
class TestUnassignLibraryReference:
|
|
def test_run(self, ifc):
|
|
ifc.get_entity("obj").should_be_called().will_return("product")
|
|
ifc.run("library.unassign_reference", products=["product"], reference="reference").should_be_called()
|
|
subject.unassign_library_reference(ifc, obj="obj", reference="reference")
|