lodel2_mirror/tests/datahandlers/test_reference.py

import unittest

from lodel.leapi.datahandlers.base_classes import Reference, MultipleRef
from leapi.query.utils import init_dyncode
from lodel.exceptions import *
from lodel.leapi.leobject import LeObject

dyncode = init_dyncode()
obj1 = dyncode.Person(
lodel_id = '1',
lastname = "Foo",
firstname = "Bar",
alias = "Foobar")

obj2 = dyncode.Collection(
lodel_id = '3',
title = "Foo)")

obj3 = dyncode.Collection(
lodel_id = '4',
title = "Foo")

class ReferenceTestCase(unittest.TestCase):

    def test_init_reference_class(self):
        reference = None
        try:
            reference = Reference()
        except NotImplementedError:
            self.assertNotIsInstance(reference, DataHandler)
            self.assertIsNone(datahandler)

    def test_reference_check_bad_data_value(self):
        test_ref = Reference((obj1, obj2))
        for test_value in ['toto']:
            with self.assertRaises(FieldValidationError):
                test_ref._check_data_value(test_value)

    def test_reference_check_good_data_value(self):
        test_ref = Reference((obj1,))
        for test_value in [obj3, 15]:
            value = test_ref._check_data_value(test_value)
            self.assertEqual(test_value, value)


class MultipleRefTestCase(unittest.TestCase):

    def test_multiref_check_data_value_not_iter(self):
        test_multiref = MultipleRef(3)
        for test_value in [obj3]:
            with self.assertRaises(FieldValidationError):
                test_multiref._check_data_value(test_value)
    
    def test_multiref_check_data_multi_bad_value_error(self):
        test_multiref = MultipleRef(3)
        for test_value in [(obj3, 15, 'toto')]:
            with self.assertRaises(FieldValidationError) as cm:
                test_multiref._check_data_value(test_value)
            the_exception = cm.exception
            self.assertEqual(the_exception.args, ("MultipleRef have for invalid values [15,'toto']  :",))
    
    def test_multiref_check_data_too_max_lenght_iter_error(self):
        test_multiref = MultipleRef(3)
        for test_value in [(obj3, obj2, obj1, obj3)]:
            with self.assertRaises(FieldValidationError):
                test_multiref._check_data_value(test_value)


    def test_multiref_check_data_uid_multi_bad_value_error(self):
        test_multiref = MultipleRef(5, **{'allowed_classes' : [dyncode.Person, dyncode.Collection]})
        for test_value in [(obj3, obj2, 1, 15, 'toto')]:
            with self.assertRaises(FieldValidationError) as cm:
                test_multiref._check_data_value(test_value)
            the_exception = cm.exception
            self.assertEqual(the_exception.args, ("MultipleRef have for invalid values ['toto']  :",))
    
    def test_multiref_check_data_object_uid_multi_good_value_error(self):
        test_multiref = MultipleRef(5, **{'allowed_classes' : [dyncode.Person, dyncode.Collection]})
        for test_value in [(obj3, obj2, 1.2, 15)]:
                value = test_multiref._check_data_value(test_value)
                self.assertEqual(value, [obj3, obj2, 1, 15])