~launchpad-pqm/launchpad/devel

# Copyright 2011 Canonical Ltd.  This software is licensed under the

# GNU Affero General Public License version 3 (see the file LICENSE).

"""Tests for lp.testing.fixture."""

__metaclass__ = type

from textwrap import dedent

from fixtures import EnvironmentVariableFixture

from zope.component import (

    adapts,

    queryAdapter,

    )

from zope.interface import (

    implements,

    Interface,

    )

from canonical.testing.layers import BaseLayer

from lp.testing import TestCase

from lp.testing.fixture import (

    RabbitServer,

    ZopeAdapterFixture,

    )

class TestRabbitServer(TestCase):

    layer = BaseLayer

    def test_service_config(self):

        # Rabbit needs to fully isolate itself: an existing per user

        # .erlange.cookie has to be ignored, and ditto bogus HOME if other

        # tests fail to cleanup.

        self.useFixture(EnvironmentVariableFixture('HOME', '/nonsense/value'))

        # RabbitServer pokes some .ini configuration into its config.

        with RabbitServer() as fixture:

            expected = dedent("""\

                [rabbitmq]

                host: localhost:%d

                userid: guest

                password: guest

                virtual_host: /

                """ % fixture.config.port)

            self.assertEqual(expected, fixture.config.service_config)

class IFoo(Interface):

    pass

class IBar(Interface):

    pass

class Foo:

    implements(IFoo)

class Bar:

    implements(IBar)

class FooToBar:

    adapts(IFoo)

    implements(IBar)

    def __init__(self, foo):

        self.foo = foo

class TestZopeAdapterFixture(TestCase):

    layer = BaseLayer

    def test_register_and_unregister(self):

        # Entering ZopeAdapterFixture's context registers the given adapter,

        # and exiting the context unregisters the adapter again.

        context = Foo()

        # No adapter from Foo to Bar is registered.

        self.assertIs(None, queryAdapter(context, IBar))

        with ZopeAdapterFixture(FooToBar):

            # Now there is an adapter from Foo to Bar.

            adapter = queryAdapter(context, IBar)

            self.assertIsNot(None, adapter)

            self.assertIsInstance(adapter, FooToBar)

        # The adapter is no longer registered.

        self.assertIs(None, queryAdapter(context, IBar))