# -*- coding: utf-8 -*-
from __future__ import print_function
from zope import schema
from zope.component import adapter
from zope.interface import implementer
from zope.interface import Interface

import doctest
import re
import sys
import unittest
import zope.component.testing


# Simple adapter behavior - no context restrictions
class IAdapterBehavior(Interface):
    pass


@implementer(IAdapterBehavior)
class AdapterBehavior(object):
    def __init__(self, context):
        self.context = context


# Adapter behavior with explicit context restriction
class IRestrictedAdapterBehavior(Interface):
    pass


@implementer(IRestrictedAdapterBehavior)
class RestrictedAdapterBehavior(object):
    def __init__(self, context):
        self.context = context


class IMinimalContextRequirements(Interface):
    pass


# Behavior with interface and for_ implied by factory
class IImpliedRestrictionAdapterBehavior(Interface):
    pass


class ISomeContext(Interface):
    pass


@implementer(IImpliedRestrictionAdapterBehavior)
@adapter(ISomeContext)
class ImpliedRestrictionAdapterBehavior(object):

    def __init__(self, context):
        self.context = context


# Behavior with marker
class IMarkerBehavior(Interface):
    pass


# Behavior to be registered with name only
class INameOnlyBehavior(Interface):
    pass


# For test of the annotation factory
class IAnnotationStored(Interface):
    some_field = schema.TextLine(title=u'Some field', default=u'default value')


# Behavior and marker
class IMarkerAndAdapterBehavior(Interface):
    pass


class IMarkerAndAdapterBehavior2(Interface):
    pass


class IMarkerAndAdapterMarker(Interface):
    pass


class IMarkerAndAdapterMarker2(Interface):
    pass


class DummyBehaviorImpl(object):

    def __init__(self, context):
        self.context = context


class Py23DocChecker(doctest.OutputChecker):
    def check_output(self, want, got, optionflags):
        if sys.version_info[0] > 2:
            want = re.sub("u'(.*?)'", "'\\1'", want)
            want = re.sub('u"(.*?)"', '"\\1"', want)
            got = re.sub(
                'plone.behavior.registration.BehaviorRegistrationNotFound',
                'BehaviorRegistrationNotFound', got)
        return doctest.OutputChecker.check_output(self, want, got, optionflags)


def test_suite():
    return unittest.TestSuite((

        doctest.DocFileSuite(
            'behaviors.rst',
            tearDown=zope.component.testing.tearDown,
            globs={
                'print_function': print_function,
            },
            checker=Py23DocChecker(),
        ),
        doctest.DocFileSuite(
            'directives.rst',
            setUp=zope.component.testing.setUp,
            tearDown=zope.component.testing.tearDown,
            checker=Py23DocChecker(),
        ),
        doctest.DocFileSuite(
            'annotation.rst',
            setUp=zope.component.testing.setUp,
            tearDown=zope.component.testing.tearDown,
            checker=Py23DocChecker(),
        ),
    ))