# Copyright 2014 DreamHost, LLC
#
# Author: DreamHost, LLC
#
# 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 mock
import uuid

import unittest2 as unittest

from akanda.rug import scheduler


class TestScheduler(unittest.TestCase):

    def test_invalid_num_workers(self):
        try:
            scheduler.Scheduler(0, lambda x: x)
        except ValueError:
            pass
        else:
            self.fail('Should have raised ValueError')

    @mock.patch('multiprocessing.Process')
    def test_creating_workers(self, process):
        s = scheduler.Scheduler(2, mock.Mock)
        self.assertEqual(2, len(s.workers))

    @mock.patch('multiprocessing.Process')
    @mock.patch('multiprocessing.JoinableQueue')
    def test_stop(self, process, queue):
        s = scheduler.Scheduler(2, mock.Mock)
        s.stop()
        for w in s.workers:
            w['queue'].put.assert_called_once(None)
            w['queue'].close.assert_called_once()
            w['worker'].join.assert_called_once()


class TestDispatcher(unittest.TestCase):

    def setUp(self):
        super(TestDispatcher, self).setUp()
        self.workers = range(5)
        self.d = scheduler.Dispatcher(self.workers)

    def _mk_uuid(self, i):
        # Creates a well-known UUID
        return str(uuid.UUID(fields=(1, 2, 3, 4, 5, i)))

    def test_pick(self):
        for i in range(len(self.workers)):
            router_id = self._mk_uuid(i)
            self.assertEqual(
                [i],
                self.d.pick_workers(router_id),
                'Incorrect index for %s' % router_id,
            )

    def test_wildcard(self):
        self.assertEqual(
            self.workers,
            self.d.pick_workers('*'),
            'wildcard dispatch failed',
        )