NodePollNowTestCase

Complexity

Total Complexity

4

Size/Duplication

Total Lines	29
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	test_run_node_not_exist()	0	6	1
A	test_run_polled_text()	0	7	1
A	test_run_polled()	0	3	1
A	test_run_connect_fail()	0	6	1

# Licensed to the StackStorm, Inc ('StackStorm') under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You 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

from mock import MagicMock

from orion_base_action_test_case import OrionBaseActionTestCase

from node_pollnow import NodePollNow

__all__ = [
    'NodePollNowTestCase'
]


class NodePollNowTestCase(OrionBaseActionTestCase):
    __test__ = True
    action_cls = NodePollNow

    def test_run_connect_fail(self):
        action = self.setup_connect_fail()
        self.assertRaises(ValueError,
                          action.run,
                          "orion",
                          "router1")

    def test_run_node_not_exist(self):
        action = self.setup_query_blank_results()
        self.assertRaises(ValueError,
                          action.run,
                          "orion",
                          "router1")

    def test_run_polled(self):
        action = self.setup_node_exists()
        self.assertTrue(action.run("router1", "orion"))

    def test_run_polled_text(self):
        expected = "fake"

        action = self.setup_node_exists()
        action.invoke = MagicMock(return_value="fake")
        result = action.run("router1", "orion")
        self.assertEqual(result, expected)