provide a new add_cubes() migration function for cases where the new cubes are linked together by new relations
In this case, we need to add all new cubes at once.
from logilab.common.testlib import TestCase, unittest_main, mock_object
from cubicweb.server.session import _make_description
class Variable:
def __init__(self, name):
self.name = name
self.children = []
def get_type(self, solution, args=None):
return solution[self.name]
def as_string(self):
return self.name
class Function:
def __init__(self, name, varname):
self.name = name
self.children = [Variable(varname)]
def get_type(self, solution, args=None):
return 'Int'
class MakeDescriptionTC(TestCase):
def test_known_values(self):
solution = {'A': 'Int', 'B': 'EUser'}
self.assertEquals(_make_description((Function('max', 'A'), Variable('B')), {}, solution),
['Int','EUser'])
if __name__ == '__main__':
unittest_main()