"""Test module for the @-decorator syntax, which is version-specific"""

import cherrypy
from cherrypy import expose, tools
from cherrypy._cpcompat import ntob


class ExposeExamples(object):

    @expose
    def no_call(self):
        return "Mr E. R. Bradshaw"

    @expose()
    def call_empty(self):
        return "Mrs. B.J. Smegma"

    @expose("call_alias")
    def nesbitt(self):
        return "Mr Nesbitt"

    @expose(["alias1", "alias2"])
    def andrews(self):
        return "Mr Ken Andrews"

    @expose(alias="alias3")
    def watson(self):
        return "Mr. and Mrs. Watson"


class ToolExamples(object):

    @expose
    # This is here to demonstrate that using the config decorator
    # does not overwrite other config attributes added by the Tool
    # decorator (in this case response_headers).
    @cherrypy.config(**{'response.stream': True})
    @tools.response_headers(headers=[('Content-Type', 'application/data')])
    def blah(self):
        yield ntob("blah")
