feb919a7b0
Makes tests a bit faster if requirements are not present as the whole class is skipped (rather than running each test, etc.).
31 lines
868 B
Python
31 lines
868 B
Python
from pytest import mark
|
|
import six
|
|
import sqlalchemy as sa
|
|
from sqlalchemy_utils.types import ip_address
|
|
from tests import TestCase
|
|
|
|
|
|
@mark.skipif('ip_address.ipaddress is None')
|
|
class TestIPAddressType(TestCase):
|
|
def create_models(self):
|
|
class Visitor(self.Base):
|
|
__tablename__ = 'document'
|
|
id = sa.Column(sa.Integer, primary_key=True)
|
|
ip_address = sa.Column(ip_address.IPAddressType)
|
|
|
|
def __repr__(self):
|
|
return 'Visitor(%r)' % self.id
|
|
|
|
self.Visitor = Visitor
|
|
|
|
def test_parameter_processing(self):
|
|
visitor = self.Visitor(
|
|
ip_address=u'111.111.111.111'
|
|
)
|
|
|
|
self.session.add(visitor)
|
|
self.session.commit()
|
|
|
|
visitor = self.session.query(self.Visitor).first()
|
|
assert six.text_type(visitor.ip_address) == u'111.111.111.111'
|