# Copyright 2011 OpenStack LLC.
# All Rights Reserved.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

import json

import nova

from nova import flags
from nova import notifier
from nova.notifier import no_op_notifier
from nova import test

import stubout

class NotifierTestCase(test.TestCase):
    """Test case for notifications"""
    def setUp(self):
        super(NotifierTestCase, self).setUp()
        self.stubs = stubout.StubOutForTesting()

    def tearDown(self):
        self.stubs.UnsetAll()
        super(NotifierTestCase, self).tearDown()

    def test_send_notification(self):
        self.notify_called = False
        def mock_notify(cls, *args):
            self.notify_called = True

        self.stubs.Set(nova.notifier.no_op_notifier.NoopNotifier, 'notify',
                mock_notify)

        class Mock(object):
            pass
        nova.notifier.notify('event_name', 'publisher_id', 'event_type',
                nova.notifier.WARN, dict(a=3))
        self.assertEqual(self.notify_called, True)

    def test_verify_message_format(self):
        """A test to ensure changing the message format is prohibitively
        annoying""" 
        def message_assert(cls, blob):
            message = json.loads(blob)
            fields = [ ('publisher_id', 'publisher_id'),
                       ('event_type', 'event_type'),
                       ('priority', 'WARN'),
                       ('payload', dict(a=3))]
            for k, v in fields:
                self.assertEqual(message[k], v)

        self.stubs.Set(nova.notifier.no_op_notifier.NoopNotifier, 'notify',
                message_assert)
        nova.notifier.notify('event_name', 'publisher_id', 'event_type',
                nova.notifier.WARN, dict(a=3))

    def test_send_rabbit_notification(self):
        self.stubs.Set(nova.flags.FLAGS, 'notification_driver',
                'nova.notifier.rabbit_notifier.RabbitNotifier')
        self.mock_cast = False
        def mock_cast(cls, *args):
            self.mock_cast = True
    
        class Mock(object):
            pass
        self.stubs.Set(nova.rpc, 'cast', mock_cast) 
        nova.notifier.notify('event_name', 'publisher_id', 'event_type',
                nova.notifier.WARN, dict(a=3))

        self.assertEqual(self.mock_cast, True)

    def test_invalid_priority(self):
        self.stubs.Set(nova.flags.FLAGS, 'notification_driver',
                'nova.notifier.rabbit_notifier.RabbitNotifier')
        self.mock_cast = False
        def mock_cast(cls, *args):
            pass
    
        class Mock(object):
            pass

        self.stubs.Set(nova.rpc, 'cast', mock_cast) 
        self.assertRaises(nova.notifier.BadPriorityException, 
                nova.notifier.notify, 'event_name', 'publisher_id',
                'event_type', 'not a priority', dict(a=3))