mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-18 11:20:21 +00:00
110 lines
3.1 KiB
Python
110 lines
3.1 KiB
Python
|
# BlenderBIM Add-on - OpenBIM Blender Add-on
|
||
|
|
# Copyright (C) 2021 Dion Moult <dion@thinkmoult.com>
|
||
|
|
#
|
||
|
|
# This file is part of BlenderBIM Add-on.
|
||
|
|
#
|
||
|
|
# BlenderBIM Add-on 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.
|
||
|
|
#
|
||
|
|
# BlenderBIM Add-on 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 BlenderBIM Add-on. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
import json
|
||
|
|
import pytest
|
||
|
|
import blenderbim.core.tool
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def ifc():
|
||
|
|
prophet = Prophecy(blenderbim.core.tool.Ifc)
|
||
|
|
yield prophet
|
||
|
|
prophet.verify()
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def blender():
|
||
|
|
prophet = Prophecy(blenderbim.core.tool.Blender)
|
||
|
|
yield prophet
|
||
|
|
prophet.verify()
|
||
|
|
|
||
|
|
|
||
|
|
def subject(sus):
|
||
|
|
def decorate(cls):
|
||
|
|
cls.sus = sus
|
||
|
|
return cls
|
||
|
|
|
||
|
|
return decorate
|
||
|
|
|
||
|
|
|
||
|
|
class Spec:
|
||
|
|
@pytest.fixture(autouse=True)
|
||
|
|
def setup(self):
|
||
|
|
self.subject = None
|
||
|
|
|
||
|
|
def construct_with(self, *args, **kwargs):
|
||
|
|
self.subject = self.sus(*args, **kwargs)
|
||
|
|
return self.subject
|
||
|
|
|
||
|
|
def predict(self, cls):
|
||
|
|
return Prophecy(cls)
|
||
|
|
|
||
|
|
|
||
|
|
class Prophecy:
|
||
|
|
def __init__(self, cls):
|
||
|
|
self.subject = cls
|
||
|
|
self.predictions = []
|
||
|
|
self.calls = []
|
||
|
|
self.return_values = {}
|
||
|
|
self.should_call = None
|
||
|
|
|
||
|
|
def __getattr__(self, attr):
|
||
|
|
if not hasattr(self.subject, attr):
|
||
|
|
raise AttributeError(f"Prophecy has no attribute {attr}")
|
||
|
|
|
||
|
|
def decorate(*args, **kwargs):
|
||
|
|
call = {"name": attr, "args": args, "kwargs": kwargs}
|
||
|
|
try:
|
||
|
|
key = json.dumps(call, sort_keys=True)
|
||
|
|
self.calls.append(call)
|
||
|
|
if key in self.return_values:
|
||
|
|
return self.return_values[key]
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
return self
|
||
|
|
|
||
|
|
return decorate
|
||
|
|
|
||
|
|
def should(self):
|
||
|
|
self.should_call = self.calls.pop()
|
||
|
|
return self
|
||
|
|
|
||
|
|
def be_called(self, number=None):
|
||
|
|
self.predictions.append({"type": "SHOULD_BE_CALLED", "number": number, "call": self.should_call})
|
||
|
|
return self
|
||
|
|
|
||
|
|
def return_with(self, value):
|
||
|
|
key = json.dumps(self.should_call, sort_keys=True)
|
||
|
|
self.return_values[key] = value
|
||
|
|
return self
|
||
|
|
|
||
|
|
def verify(self):
|
||
|
|
for prediction in self.predictions:
|
||
|
|
if prediction["type"] == "SHOULD_BE_CALLED":
|
||
|
|
self.verify_should_be_called(prediction)
|
||
|
|
|
||
|
|
def verify_should_be_called(self, prediction):
|
||
|
|
if prediction["number"]:
|
||
|
|
count = self.calls.count(prediction["call"])
|
||
|
|
if count != prediction["number"]:
|
||
|
|
raise Exception(f"Called {count}: {prediction}")
|
||
|
|
else:
|
||
|
|
if prediction["call"] not in self.calls:
|
||
|
|
raise Exception("Not called", prediction)
|