mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
Add unit tests for imgadm beacon
This commit is contained in:
Jorge Schrauwen
parent
f513b9d7ec
commit
4892c0f9dc
214
tests/unit/beacons/test_smartos_imgadm_beacon.py
Normal file
214
tests/unit/beacons/test_smartos_imgadm_beacon.py
Normal file
@ -0,0 +1,214 @@
|
||||
# coding: utf-8
|
||||
|
||||
# Python libs
|
||||
from __future__ import absolute_import
|
||||
|
||||
# Salt testing libs
|
||||
from tests.support.unit import skipIf, TestCase
|
||||
from tests.support.mock import NO_MOCK, NO_MOCK_REASON, patch, MagicMock
|
||||
from tests.support.mixins import LoaderModuleMockMixin
|
||||
|
||||
# Salt libs
|
||||
import salt.beacons.smartos_imgadm as imgadm
|
||||
|
||||
|
||||
# Mock Results
|
||||
MOCK_CLEAN_STATE = {'first_run': True, 'images': []}
|
||||
MOCK_IMAGE_NONE = {}
|
||||
MOCK_IMAGE_ONE = {
|
||||
'00000000-0000-0000-0000-000000000001': {
|
||||
'description': 'Example Image 1',
|
||||
'name': 'example-1',
|
||||
'os': 'smartos',
|
||||
'published': '2018-01-01T00:42:00Z',
|
||||
'source': 'https://images.joyent.com',
|
||||
'version': '18.1.0',
|
||||
},
|
||||
}
|
||||
MOCK_IMAGE_TWO = {
|
||||
'00000000-0000-0000-0000-000000000001': {
|
||||
'description': 'Example Image 1',
|
||||
'name': 'example-1',
|
||||
'os': 'smartos',
|
||||
'published': '2018-01-01T00:42:00Z',
|
||||
'source': 'https://images.joyent.com',
|
||||
'version': '18.1.0',
|
||||
},
|
||||
'00000000-0000-0000-0000-000000000002': {
|
||||
'description': 'Example Image 2',
|
||||
'name': 'example-2',
|
||||
'os': 'smartos',
|
||||
'published': '2018-01-01T00:42:00Z',
|
||||
'source': 'https://images.joyent.com',
|
||||
'version': '18.2.0',
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
||||
class SmartOSImgAdmBeaconTestCase(TestCase, LoaderModuleMockMixin):
|
||||
'''
|
||||
Test case for salt.beacons.imgadm
|
||||
'''
|
||||
|
||||
def setup_loader_modules(self):
|
||||
return {
|
||||
imgadm: {
|
||||
'__context__': {},
|
||||
'__salt__': {},
|
||||
}
|
||||
}
|
||||
|
||||
def test_non_list_config(self):
|
||||
'''
|
||||
We only have minimal validation so we test that here
|
||||
'''
|
||||
config = {}
|
||||
|
||||
ret = imgadm.validate(config)
|
||||
|
||||
self.assertEqual(ret, (False, 'Configuration for imgadm beacon must be a list!'))
|
||||
|
||||
def test_imported_startup(self):
|
||||
'''
|
||||
Test with one image and startup_import_event
|
||||
'''
|
||||
# NOTE: this should yield 1 imported event
|
||||
with patch.dict(imgadm.IMGADM_STATE, MOCK_CLEAN_STATE), \
|
||||
patch.dict(imgadm.__salt__, {'imgadm.list': MagicMock(return_value=MOCK_IMAGE_ONE)}):
|
||||
|
||||
config = [{'startup_import_event': True}]
|
||||
|
||||
ret = imgadm.validate(config)
|
||||
self.assertEqual(ret, (True, 'Valid beacon configuration'))
|
||||
|
||||
ret = imgadm.beacon(config)
|
||||
res = [{'description': 'Example Image 1',
|
||||
'name': 'example-1',
|
||||
'os': 'smartos',
|
||||
'published': '2018-01-01T00:42:00Z',
|
||||
'source': 'https://images.joyent.com',
|
||||
'tag': 'imported/00000000-0000-0000-0000-000000000001',
|
||||
'version': '18.1.0'}]
|
||||
self.assertEqual(ret, res)
|
||||
|
||||
def test_imported_nostartup(self):
|
||||
'''
|
||||
Test with one image and startup_import_event unset/false
|
||||
'''
|
||||
# NOTE: this should yield 0 imported event
|
||||
with patch.dict(imgadm.IMGADM_STATE, MOCK_CLEAN_STATE), \
|
||||
patch.dict(imgadm.__salt__, {'imgadm.list': MagicMock(return_value=MOCK_IMAGE_ONE)}):
|
||||
|
||||
config = []
|
||||
|
||||
ret = imgadm.validate(config)
|
||||
self.assertEqual(ret, (True, 'Valid beacon configuration'))
|
||||
|
||||
ret = imgadm.beacon(config)
|
||||
res = []
|
||||
|
||||
self.assertEqual(ret, res)
|
||||
|
||||
def test_imported(self):
|
||||
'''
|
||||
Test with one image and a new image added on the 2nd pass
|
||||
'''
|
||||
# NOTE: this should yield 1 imported event
|
||||
with patch.dict(imgadm.IMGADM_STATE, MOCK_CLEAN_STATE), \
|
||||
patch.dict(imgadm.__salt__, {'imgadm.list': MagicMock(side_effect=[MOCK_IMAGE_ONE, MOCK_IMAGE_TWO])}):
|
||||
|
||||
config = []
|
||||
|
||||
ret = imgadm.validate(config)
|
||||
self.assertEqual(ret, (True, 'Valid beacon configuration'))
|
||||
|
||||
# Initial pass (Initialized state and do not yield imported imagesd at startup)
|
||||
ret = imgadm.beacon(config)
|
||||
|
||||
# Second pass (After importing a new image)
|
||||
ret = imgadm.beacon(config)
|
||||
res = [{'description': 'Example Image 2',
|
||||
'name': 'example-2',
|
||||
'os': 'smartos',
|
||||
'published': '2018-01-01T00:42:00Z',
|
||||
'source': 'https://images.joyent.com',
|
||||
'tag': 'imported/00000000-0000-0000-0000-000000000002',
|
||||
'version': '18.2.0'}]
|
||||
|
||||
self.assertEqual(ret, res)
|
||||
|
||||
def test_deleted(self):
|
||||
'''
|
||||
Test with two images and one gets deletes
|
||||
'''
|
||||
# NOTE: this should yield 1 deleted event
|
||||
with patch.dict(imgadm.IMGADM_STATE, MOCK_CLEAN_STATE), \
|
||||
patch.dict(imgadm.__salt__, {'imgadm.list': MagicMock(side_effect=[MOCK_IMAGE_TWO, MOCK_IMAGE_ONE])}):
|
||||
|
||||
config = []
|
||||
|
||||
ret = imgadm.validate(config)
|
||||
self.assertEqual(ret, (True, 'Valid beacon configuration'))
|
||||
|
||||
# Initial pass (Initialized state and do not yield imported imagesd at startup)
|
||||
ret = imgadm.beacon(config)
|
||||
|
||||
# Second pass (After deleting one image)
|
||||
ret = imgadm.beacon(config)
|
||||
res = [{'description': 'Example Image 2',
|
||||
'name': 'example-2',
|
||||
'os': 'smartos',
|
||||
'published': '2018-01-01T00:42:00Z',
|
||||
'source': 'https://images.joyent.com',
|
||||
'tag': 'deleted/00000000-0000-0000-0000-000000000002',
|
||||
'version': '18.2.0'}]
|
||||
|
||||
self.assertEqual(ret, res)
|
||||
|
||||
def test_complex(self):
|
||||
'''
|
||||
Test with one image, delete both, import 2
|
||||
'''
|
||||
# NOTE: this should yield 1 delete and 2 import events
|
||||
with patch.dict(imgadm.IMGADM_STATE, MOCK_CLEAN_STATE), \
|
||||
patch.dict(imgadm.__salt__, {'imgadm.list': MagicMock(side_effect=[MOCK_IMAGE_ONE, MOCK_IMAGE_NONE, MOCK_IMAGE_TWO])}):
|
||||
|
||||
config = []
|
||||
|
||||
ret = imgadm.validate(config)
|
||||
self.assertEqual(ret, (True, 'Valid beacon configuration'))
|
||||
|
||||
# Initial pass (Initialized state and do not yield imported imagesd at startup)
|
||||
ret = imgadm.beacon(config)
|
||||
|
||||
# Second pass (After deleting one image)
|
||||
ret = imgadm.beacon(config)
|
||||
res = [{'description': 'Example Image 1',
|
||||
'name': 'example-1',
|
||||
'os': 'smartos',
|
||||
'published': '2018-01-01T00:42:00Z',
|
||||
'source': 'https://images.joyent.com',
|
||||
'tag': 'deleted/00000000-0000-0000-0000-000000000001',
|
||||
'version': '18.1.0'}]
|
||||
|
||||
self.assertEqual(ret, res)
|
||||
|
||||
# Third pass (After importing two images)
|
||||
ret = imgadm.beacon(config)
|
||||
res = [{'description': 'Example Image 1',
|
||||
'name': 'example-1',
|
||||
'os': 'smartos',
|
||||
'published': '2018-01-01T00:42:00Z',
|
||||
'source': 'https://images.joyent.com',
|
||||
'tag': 'imported/00000000-0000-0000-0000-000000000001',
|
||||
'version': '18.1.0'},
|
||||
{'description': 'Example Image 2',
|
||||
'name': 'example-2',
|
||||
'os': 'smartos',
|
||||
'published': '2018-01-01T00:42:00Z',
|
||||
'source': 'https://images.joyent.com',
|
||||
'tag': 'imported/00000000-0000-0000-0000-000000000002',
|
||||
'version': '18.2.0'}]
|
||||
|
||||
self.assertEqual(ret, res)
|
||||
Loading…
Reference in New Issue
Block a user