358 lines
9.0 KiB
Python
358 lines
9.0 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# This file is part of facho. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
|
|
"""Tests for `facho` package."""
|
|
|
|
import pytest
|
|
|
|
import facho.model
|
|
import facho.model.fields as fields
|
|
|
|
def test_model_to_element():
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
person = Person()
|
|
|
|
assert "<Person/>" == person.to_xml()
|
|
|
|
def test_model_to_element_with_attribute():
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
id = fields.Attribute('id')
|
|
|
|
person = Person()
|
|
person.id = 33
|
|
|
|
personb = Person()
|
|
personb.id = 44
|
|
|
|
assert "<Person id=\"33\"/>" == person.to_xml()
|
|
assert "<Person id=\"44\"/>" == personb.to_xml()
|
|
|
|
def test_model_to_element_with_attribute_as_element():
|
|
class ID(facho.model.Model):
|
|
__name__ = 'ID'
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
id = fields.Many2One(ID)
|
|
|
|
person = Person()
|
|
person.id = 33
|
|
assert "<Person><ID>33</ID></Person>" == person.to_xml()
|
|
|
|
def test_many2one_with_custom_attributes():
|
|
class TaxAmount(facho.model.Model):
|
|
__name__ = 'TaxAmount'
|
|
|
|
currencyID = fields.Attribute('currencyID')
|
|
|
|
class TaxTotal(facho.model.Model):
|
|
__name__ = 'TaxTotal'
|
|
|
|
amount = fields.Many2One(TaxAmount)
|
|
|
|
tax_total = TaxTotal()
|
|
tax_total.amount = 3333
|
|
tax_total.amount.currencyID = 'COP'
|
|
assert '<TaxTotal><TaxAmount currencyID="COP">3333</TaxAmount></TaxTotal>' == tax_total.to_xml()
|
|
|
|
def test_many2one_with_custom_setter():
|
|
|
|
class PhysicalLocation(facho.model.Model):
|
|
__name__ = 'PhysicalLocation'
|
|
|
|
id = fields.Attribute('ID')
|
|
|
|
class Party(facho.model.Model):
|
|
__name__ = 'Party'
|
|
|
|
location = fields.Many2One(PhysicalLocation, setter='location_setter')
|
|
|
|
def location_setter(self, field, value):
|
|
field.id = value
|
|
|
|
party = Party()
|
|
party.location = 99
|
|
assert '<Party><PhysicalLocation ID="99"/></Party>' == party.to_xml()
|
|
|
|
def test_many2one_auto_create():
|
|
class TaxAmount(facho.model.Model):
|
|
__name__ = 'TaxAmount'
|
|
|
|
currencyID = fields.Attribute('currencyID')
|
|
|
|
class TaxTotal(facho.model.Model):
|
|
__name__ = 'TaxTotal'
|
|
|
|
amount = fields.Many2One(TaxAmount)
|
|
|
|
tax_total = TaxTotal()
|
|
tax_total.amount.currencyID = 'COP'
|
|
tax_total.amount = 3333
|
|
assert '<TaxTotal><TaxAmount currencyID="COP">3333</TaxAmount></TaxTotal>' == tax_total.to_xml()
|
|
|
|
def test_field_model():
|
|
class ID(facho.model.Model):
|
|
__name__ = 'ID'
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
id = fields.Many2One(ID)
|
|
|
|
person = Person()
|
|
person.id = ID()
|
|
person.id = 33
|
|
assert "<Person><ID>33</ID></Person>" == person.to_xml()
|
|
|
|
def test_field_multiple_model():
|
|
class ID(facho.model.Model):
|
|
__name__ = 'ID'
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
id = fields.Many2One(ID)
|
|
id2 = fields.Many2One(ID)
|
|
|
|
person = Person()
|
|
person.id = 33
|
|
person.id2 = 44
|
|
assert "<Person><ID>33</ID><ID>44</ID></Person>" == person.to_xml()
|
|
|
|
def test_field_model_failed_initialization():
|
|
class ID(facho.model.Model):
|
|
__name__ = 'ID'
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
id = fields.Many2One(ID)
|
|
|
|
|
|
person = Person()
|
|
person.id = 33
|
|
assert "<Person><ID>33</ID></Person>" == person.to_xml()
|
|
|
|
def test_field_model_with_custom_name():
|
|
class ID(facho.model.Model):
|
|
__name__ = 'ID'
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
id = fields.Many2One(ID, name='DID')
|
|
|
|
|
|
person = Person()
|
|
person.id = 33
|
|
assert "<Person><DID>33</DID></Person>" == person.to_xml()
|
|
|
|
def test_field_model_default_initialization_with_attributes():
|
|
class ID(facho.model.Model):
|
|
__name__ = 'ID'
|
|
|
|
reference = fields.Attribute('REFERENCE')
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
id = fields.Many2One(ID)
|
|
|
|
person = Person()
|
|
person.id = 33
|
|
person.id.reference = 'haber'
|
|
assert '<Person><ID REFERENCE="haber">33</ID></Person>' == person.to_xml()
|
|
|
|
def test_model_with_xml_namespace():
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
__namespace__ = {
|
|
'facho': 'http://lib.facho.cyou'
|
|
}
|
|
|
|
person = Person()
|
|
assert '<Person xmlns:facho="http://lib.facho.cyou"/>'
|
|
|
|
def test_model_with_xml_namespace_nested():
|
|
class ID(facho.model.Model):
|
|
__name__ = 'ID'
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
__namespace__ = {
|
|
'facho': 'http://lib.facho.cyou'
|
|
}
|
|
|
|
id = fields.Many2One(ID, namespace='facho')
|
|
|
|
person = Person()
|
|
person.id = 33
|
|
assert '<Person xmlns:facho="http://lib.facho.cyou"><facho:ID>33</facho:ID></Person>' == person.to_xml()
|
|
|
|
def test_field_model_with_namespace():
|
|
class ID(facho.model.Model):
|
|
__name__ = 'ID'
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
__namespace__ = {
|
|
"facho": "http://lib.facho.cyou"
|
|
}
|
|
id = fields.Many2One(ID, namespace="facho")
|
|
|
|
|
|
person = Person()
|
|
person.id = 33
|
|
assert '<Person xmlns:facho="http://lib.facho.cyou"><facho:ID>33</facho:ID></Person>' == person.to_xml()
|
|
|
|
def test_field_hook_before_xml():
|
|
class Hash(facho.model.Model):
|
|
__name__ = 'Hash'
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
hash = fields.Many2One(Hash)
|
|
|
|
def __before_xml__(self):
|
|
self.hash = "calculate"
|
|
|
|
person = Person()
|
|
assert "<Person><Hash>calculate</Hash></Person>" == person.to_xml()
|
|
|
|
|
|
def test_field_function_with_attribute():
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
hash = fields.Function(fields.Attribute('hash'), getter='get_hash')
|
|
|
|
def get_hash(self, name, field):
|
|
return 'calculate'
|
|
|
|
person = Person()
|
|
assert '<Person hash="calculate"/>'
|
|
|
|
def test_field_function_with_model():
|
|
class Hash(facho.model.Model):
|
|
__name__ = 'Hash'
|
|
|
|
id = fields.Attribute('id')
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
hash = fields.Function(fields.Many2One(Hash), getter='get_hash')
|
|
|
|
def get_hash(self, name, field):
|
|
field.id = 'calculate'
|
|
|
|
|
|
person = Person()
|
|
assert person.hash.id == 'calculate'
|
|
assert '<Person/>'
|
|
|
|
|
|
def test_field_function_setter():
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
hash = fields.Attribute('hash')
|
|
password = fields.Virtual(setter='set_hash')
|
|
|
|
def set_hash(self, name, value):
|
|
self.hash = "%s+2" % (value)
|
|
|
|
person = Person()
|
|
person.password = 'calculate'
|
|
assert '<Person hash="calculate+2"/>' == person.to_xml()
|
|
|
|
def test_field_function_only_setter():
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
hash = fields.Attribute('hash')
|
|
password = fields.Virtual(setter='set_hash')
|
|
|
|
def set_hash(self, name, value):
|
|
self.hash = "%s+2" % (value)
|
|
|
|
person = Person()
|
|
person.password = 'calculate'
|
|
assert '<Person hash="calculate+2"/>' == person.to_xml()
|
|
|
|
def test_model_set_default_setter():
|
|
class Hash(facho.model.Model):
|
|
__name__ = 'Hash'
|
|
|
|
def __default_set__(self, value):
|
|
return "%s+3" % (value)
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
hash = fields.Many2One(Hash)
|
|
|
|
person = Person()
|
|
person.hash = 'hola'
|
|
assert '<Person><Hash>hola+3</Hash></Person>' == person.to_xml()
|
|
|
|
|
|
def test_field_virtual():
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
age = fields.Virtual()
|
|
|
|
person = Person()
|
|
person.age = 55
|
|
assert person.age == 55
|
|
assert "<Person/>" == person.to_xml()
|
|
|
|
|
|
def test_field_inserted_default_attribute():
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
hash = fields.Attribute('hash', default='calculate')
|
|
|
|
|
|
person = Person()
|
|
assert '<Person hash="calculate"/>' == person.to_xml()
|
|
|
|
def test_field_inserted_default_many2one():
|
|
class ID(facho.model.Model):
|
|
__name__ = 'ID'
|
|
|
|
key = fields.Attribute('key')
|
|
|
|
def __default_set__(self, value):
|
|
self.key = value
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
id = fields.Many2One(ID, default="oe")
|
|
|
|
person = Person()
|
|
assert '<Person><ID key="oe"/></Person>' == person.to_xml()
|
|
|
|
def test_field_inserted_default_nested_many2one():
|
|
class ID(facho.model.Model):
|
|
__name__ = 'ID'
|
|
|
|
class Person(facho.model.Model):
|
|
__name__ = 'Person'
|
|
|
|
id = fields.Many2One(ID, default="ole")
|
|
|
|
person = Person()
|
|
assert '<Person><ID>ole</ID></Person>' == person.to_xml()
|
|
|