SaltLocalActionTestCase

Complexity

Total Complexity

4

Size/Duplication

Total Lines	24
Duplicated Lines	0 %

4 Methods

Rating	Name	Duplication	Size	Complexity
A	setUp()	0	7	1
A	test_generic_action_no_args()	0	3	1
A	test_generic_action_one_arg()	0	4	1
A	test_generic_action_multiple_args()	0	4	1

from st2tests.base import BaseActionTestCase

from local import SaltLocal
import requests_mock
from requests_mock.contrib import fixture
import testtools

__all__ = [
    'SaltLocalActionTestCase'
]

no_args = {
    'module': 'this.something',
    'target': '*',
    'expr_form': 'glob',
    'args': []
}

one_arg = {
    'module': 'this.something',
    'target': '*',
    'expr_form': 'glob',
    'args': ['os'],
}

multiple_args = {
    'module': 'this.something',
    'target': '*',
    'expr_form': 'glob',
    'args': ['this', 'that', 'home'],
}

CONFIG_DATA = {
    'api_url': 'https://example.com',
    'username': 'this',
    'password': 'that'
}
requests_mock.Mocker.TEST_PREFIX = 'test'


class SaltLocalActionTestCase(testtools.TestCase, BaseActionTestCase):
    action_cls = SaltLocal

    def setUp(self):
        super(SaltLocalActionTestCase, self).setUp()
        self.m = self.useFixture(fixture.Fixture())
        self.action = self.get_action_instance(config=CONFIG_DATA)
        self.m.register_uri('POST',
                            "{}/run".format(CONFIG_DATA['api_url']),
                            json={})

    def test_generic_action_no_args(self):
        self.action.run(**no_args)
        self.assertNotIn('arg', self.action.data)

    def test_generic_action_one_arg(self):
        self.action.run(**one_arg)
        self.assertIn('arg', self.action.data)
        self.assertIsInstance(self.action.data['arg'], list)

    def test_generic_action_multiple_args(self):
        self.action.run(**multiple_args)
        self.assertIn('arg', self.action.data)
        self.assertIsInstance(self.action.data['arg'], list)