mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
Merge pull request #23447 from jayeshka/grafana-states-unit-test
adding states/grafana unit test case
This commit is contained in:
Justin Findlay
commit
23d7e7ef92
1 changed files with 149 additions and 0 deletions
149
tests/unit/states/grafana_test.py
Normal file
149
tests/unit/states/grafana_test.py
Normal file
|
|
@ -0,0 +1,149 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
'''
|
||||
:codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
|
||||
'''
|
||||
# Import Python libs
|
||||
from __future__ import absolute_import
|
||||
|
||||
# Import Salt Testing Libs
|
||||
from salttesting import skipIf, TestCase
|
||||
from salttesting.mock import (
|
||||
NO_MOCK,
|
||||
NO_MOCK_REASON,
|
||||
MagicMock,
|
||||
patch
|
||||
)
|
||||
|
||||
from salttesting.helpers import ensure_in_syspath
|
||||
from salt.exceptions import SaltInvocationError
|
||||
import json
|
||||
|
||||
ensure_in_syspath('../../')
|
||||
|
||||
# Import Salt Libs
|
||||
from salt.states import grafana
|
||||
|
||||
grafana.__opts__ = {}
|
||||
grafana.__salt__ = {}
|
||||
|
||||
|
||||
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
||||
class GrafanaTestCase(TestCase):
|
||||
'''
|
||||
Test cases for salt.states.grafana
|
||||
'''
|
||||
# 'dashboard_present' function tests: 1
|
||||
|
||||
def test_dashboard_present(self):
|
||||
'''
|
||||
Test to ensure the grafana dashboard exists and is managed.
|
||||
'''
|
||||
name = 'myservice'
|
||||
rows = ['systemhealth', 'requests', 'title']
|
||||
row = [{'panels': [{'id': 'a'}], 'title': 'systemhealth'}]
|
||||
|
||||
ret = {'name': name,
|
||||
'result': None,
|
||||
'changes': {},
|
||||
'comment': ''}
|
||||
|
||||
comt1 = ("Dashboard myservice is set to be updated. The following rows "
|
||||
"set to be updated: ['systemhealth']")
|
||||
self.assertRaises(SaltInvocationError, grafana.dashboard_present, name,
|
||||
profile=False)
|
||||
|
||||
self.assertRaises(SaltInvocationError, grafana.dashboard_present, name,
|
||||
True, True)
|
||||
|
||||
mock = MagicMock(side_effect=[{'hosts': True, 'index': False},
|
||||
{'hosts': True, 'index': True},
|
||||
{'hosts': True, 'index': True},
|
||||
{'hosts': True, 'index': True},
|
||||
{'hosts': True, 'index': True},
|
||||
{'hosts': True, 'index': True},
|
||||
{'hosts': True, 'index': True}])
|
||||
mock_f = MagicMock(side_effect=[False, False, True, True, True, True])
|
||||
mock_t = MagicMock(return_value='')
|
||||
mock_i = MagicMock(return_value=False)
|
||||
source = {'dashboard': '["rows", {"rows":["baz", null, 1.0, 2]}]'}
|
||||
mock_dict = MagicMock(return_value={'_source': source})
|
||||
with patch.dict(grafana.__salt__, {'config.option': mock,
|
||||
'elasticsearch.exists': mock_f,
|
||||
'pillar.get': mock_t,
|
||||
'elasticsearch.get': mock_dict,
|
||||
'elasticsearch.index': mock_i}):
|
||||
self.assertRaises(SaltInvocationError, grafana.dashboard_present,
|
||||
name)
|
||||
|
||||
with patch.dict(grafana.__opts__, {'test': True}):
|
||||
self.assertRaises(SaltInvocationError, grafana.dashboard_present,
|
||||
name)
|
||||
|
||||
comt = ('Dashboard {0} is set to be created.'.format(name))
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(grafana.dashboard_present(name, True), ret)
|
||||
|
||||
mock = MagicMock(return_value={'rows':
|
||||
[{'panels': 'b',
|
||||
'title': 'systemhealth'}]})
|
||||
with patch.object(json, 'loads', mock):
|
||||
ret.update({'comment': comt1, 'result': None})
|
||||
self.assertDictEqual(grafana.dashboard_present(name, True,
|
||||
rows=row),
|
||||
ret)
|
||||
|
||||
with patch.object(json, 'loads',
|
||||
MagicMock(return_value={'rows': {}})):
|
||||
self.assertRaises(SaltInvocationError,
|
||||
grafana.dashboard_present, name,
|
||||
rows_from_pillar=rows)
|
||||
|
||||
comt = ('Dashboard myservice is up to date')
|
||||
ret.update({'comment': comt, 'result': True})
|
||||
self.assertDictEqual(grafana.dashboard_present(name, True), ret)
|
||||
|
||||
mock = MagicMock(return_value={'rows': [{'panels': 'b',
|
||||
'title': 'systemhealth'}]})
|
||||
with patch.dict(grafana.__opts__, {'test': False}):
|
||||
with patch.object(json, 'loads', mock):
|
||||
comt = ('Failed to update dashboard myservice.')
|
||||
ret.update({'comment': comt, 'result': False})
|
||||
self.assertDictEqual(grafana.dashboard_present(name, True,
|
||||
rows=row),
|
||||
ret)
|
||||
|
||||
# 'dashboard_absent' function tests: 1
|
||||
|
||||
def test_dashboard_absent(self):
|
||||
'''
|
||||
Test to ensure the named grafana dashboard is deleted.
|
||||
'''
|
||||
name = 'myservice'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': None,
|
||||
'changes': {},
|
||||
'comment': ''}
|
||||
|
||||
mock = MagicMock(side_effect=[{'hosts': True, 'index': False},
|
||||
{'hosts': True, 'index': True},
|
||||
{'hosts': True, 'index': True}])
|
||||
mock_f = MagicMock(side_effect=[True, False])
|
||||
with patch.dict(grafana.__salt__, {'config.option': mock,
|
||||
'elasticsearch.exists': mock_f}):
|
||||
self.assertRaises(SaltInvocationError, grafana.dashboard_absent,
|
||||
name)
|
||||
|
||||
with patch.dict(grafana.__opts__, {'test': True}):
|
||||
comt = ('Dashboard myservice is set to be removed.')
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
self.assertDictEqual(grafana.dashboard_absent(name), ret)
|
||||
|
||||
comt = ('Dashboard myservice does not exist.')
|
||||
ret.update({'comment': comt, 'result': True})
|
||||
self.assertDictEqual(grafana.dashboard_absent(name), ret)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
from integration import run_tests
|
||||
run_tests(GrafanaTestCase, needs_daemon=False)
|
||||
Loading…
Add table
Reference in a new issue