class TestReliureAPIMultiInputs(unittest.TestCase):
maxDiff = None
def setUp(self):
self.engine = Engine("op1", "op2")
self.engine.op1.setup(in_name="in", out_name="middle", required=False)
self.engine.op2.setup(in_name="middle", out_name="out")
self.engine.op1.set(OptProductEx())
foisdouze = OptProductEx("foisdouze")
foisdouze.force_option_value("factor", 12)
self.engine.op2.set(foisdouze, OptProductEx())
egn_view = EngineView(self.engine, name="my_egn")
egn_view.add_input("in", Numeric(vtype=int, min=-5, max=5))
egn_view.add_input("middle", Numeric(vtype=int))
print(self.engine.needed_inputs())
egn_view.add_output("in")
egn_view.add_output("middle")
egn_view.add_output("out")
api = ReliureAPI()
api.register_view(egn_view)
app = Flask(__name__)
app.config['TESTING'] = True
app.register_blueprint(api, url_prefix="/api")
self.app = app.test_client()
def test_routes(self):
resp = self.app.get('api/')
data = json.loads(resp.data.decode("utf-8"))
# test routes
routes = data["routes"]
routes = {route['name']: route for route in routes}
pprint(routes)
# routes GET api/ for this entry point
assert "api.routes" in routes
assert routes["api.routes"]["path"] == u'/api/'
assert sorted(routes["api.routes"]["methods"]) == [u'GET', u'HEAD', u'OPTIONS']
# routes GET api/egn for options
assert "api.my_egn_options" in routes
assert routes["api.my_egn_options"]["path"] == u'/api/my_egn'
assert sorted(routes["api.my_egn_options"]["methods"]) == [u'GET', u'HEAD', u'OPTIONS']
# routes POST api/egn for play
assert "api.my_egn" in routes
assert routes["api.my_egn"]["path"] == u'/api/my_egn'
assert sorted(routes["api.my_egn"]["methods"]) == [u'OPTIONS', u'POST']
def test_options(self):
resp = self.app.get('api/my_egn')
data = json.loads(resp.data.decode("utf-8"))
# check we have the same than in engine
assert data["blocks"] == self.engine.as_dict()["blocks"]
assert data["args"] == ["in", "middle"]
assert data["returns"] == ["in", "middle", "out"]
def test_play_simple(self):
# it should be possible to play the full engine
# prepare query
rdata = {'in': 2}
rdata["options"] = {
'op1': [{
'name': 'mult_opt',
}]
}
json_data = json.dumps(rdata)
resp = self.app.post('api/my_egn', data=json_data, content_type='application/json')
# load the results
resp_data = json.loads(resp.data.decode("utf-8"))
# check it
assert "results" in resp_data
results = resp_data["results"]
assert "out" in results
assert len(results) == 3 # in, middle, out
assert results["out"] == 2*5*12
def test_play_skip_op1(self):
# it should be possible to play the only the block op2
# prepare query
rdata = {'middle': 2}
rdata["options"] = {
'op1': []
}
json_data = json.dumps(rdata)
resp = self.app.post('api/my_egn', data=json_data, content_type='application/json')
# load the results
resp_data = json.loads(resp.data.decode("utf-8"))
# check it
assert "results" in resp_data
results = resp_data["results"]
assert "out" in results
assert len(results) == 2 # middle, out
assert results["out"] == 2*12
class TestReliureAPISimple(unittest.TestCase):
maxDiff = None
def setUp(self):
self.engine = Engine("op1", "op2")
self.engine.op1.setup(in_name="in")
self.engine.op2.setup(out_name="out")
self.engine.op1.set(OptProductEx())
foisdouze = OptProductEx("foisdouze")
foisdouze.force_option_value("factor", 12)
self.engine.op2.set(foisdouze, OptProductEx())
egn_view = EngineView(self.engine)
egn_view.set_input_type(Numeric(vtype=int, min=-5, max=5))
egn_view.add_output("out")
api = ReliureAPI()
api.register_view(egn_view, url_prefix="egn")
app = Flask(__name__)
self.appp = app
app.config['TESTING'] = True
app.register_blueprint(api, url_prefix="/api")
self.app = app.test_client()
def test_engine_view_init(self):
_egn_view = EngineView(self.engine)
with pytest.raises(ValueError):
_egn_view.add_output("existe_pas")
def test_routes(self):
resp = self.app.get('api/')
data = json.loads(resp.data.decode("utf-8"))
assert data["api"] == "api"
assert data["url_root"] == "http://localhost/"
# test routes
routes = data["routes"]
routes = {route['name']: route for route in routes}
pprint(routes)
# routes GET api/ for this entry point
assert "api.routes" in routes
assert routes["api.routes"]["path"] == u'/api/'
assert sorted(routes["api.routes"]["methods"]) == [u'GET', u'HEAD', u'OPTIONS']
# routes GET api/egn for options
assert "api.egn_options" in routes
assert routes["api.egn_options"]["path"] == u'/api/egn'
assert sorted(routes["api.egn_options"]["methods"]) == [u'GET', u'HEAD', u'OPTIONS']
# routes POST api/egn for play
assert "api.egn" in routes
assert routes["api.egn"]["path"] == u'/api/egn'
assert sorted(routes["api.egn"]["methods"]) == [u'OPTIONS', u'POST']
def test_options(self):
resp = self.app.get('api/egn')
data = json.loads(resp.data.decode("utf-8"))
# check we have the same than in engine
assert data["blocks"] == self.engine.as_dict()["blocks"]
assert data["args"] == ["in"]
assert data["returns"] == ["out"]
def test_play_simple(self):
# prepare query
rdata = {'in': '2'}
json_data = json.dumps(rdata)
resp = self.app.post('api/egn', data=json_data, content_type='application/json')
# load the results
resp_data = json.loads(resp.data.decode("utf-8"))
# check it
assert "results" in resp_data
results = resp_data["results"]
assert "out" in results
assert len(results) == 1
assert results["out"] == 2*5*12
def test_play_nojson(self):
# prepare query
data = {u'in': 3}
resp = self.app.post('api/egn', data=data)
# load the results
resp_data = json.loads(resp.data.decode("utf-8"))
# check it
assert "results" in resp_data
results = resp_data["results"]
assert "out" in results
assert len(results) == 1
assert results["out"] == 3*5*12
def test_play_simple_options(self):
# prepare query
rdata = {'in': '2'}
rdata["options"] = {
'op2': [{
'name': 'mult_opt',
'options': {
'factor': '2'
}
}]
}
json_data = json.dumps(rdata)
resp = self.app.post('api/egn', data=json_data, content_type='application/json')
# load the results
resp_data = json.loads(resp.data.decode("utf-8"))
# check it
assert "results" in resp_data
results = resp_data["results"]
assert "out" in results
assert len(results) == 1
assert results["out"] == 2*2*5
def test_as_dict(self):
engine = Engine("op1", "op2")
engine.set("op1", self.plus_comp, self.mult_opt, self.max_comp)
engine.set("op2", self.plus_comp)
assert engine.as_dict() == {
'args': ['input'],
'blocks': [
{
'components': [
{
'default': True, #This is the default NOT the selected value
'name': 'plus_comp',
'options': None
},
{
'default': False,
'name': 'mult_opt',
'options': [
{
'name': 'factor',
'type': 'value',
'value': 5,
'otype': {
'choices': None,
'default': 5,
'help': 'multipliation factor',
'max': None,
'min': None,
'multi': False,
'type': 'Numeric',
'uniq': False,
'vtype': 'int'
}
}
]
},
{
'name': 'max20',
'default': False,
'options': None
}
],
'args': None,
'multiple': False,
'name': 'op1',
'returns': 'op1',
'required': True
},
{
'components': [
{
'name': 'plus_comp',
'default': True,
'options': None
}
],
'args': None,
'multiple': False,
'name': 'op2',
'returns': 'op2',
'required': True
}
]
}
