def test_session_max_retires(self):
# Arrange
template_retriever = TemplateRetriever()
httpretty.register_uri('GET',
TEMPLATES_YML,
body=self.mock_get_templates_from_github())
# Act
templates = template_retriever.get_templates()
# Assert
self.assertEqual(templates['router'].name, 'router')
self.assertEqual(templates['router'].description,
'Basic router template')
self.assertEqual(
templates['router'].repository,
'https://github.com/QualiSystems/shellfoundry-router-template')
self.assertEqual(templates['switch'].name, 'switch')
self.assertEqual(templates['switch'].description,
'Basic switch template')
self.assertEqual(
templates['switch'].repository,
'https://github.com/QualiSystems/shellfoundry-switch-template')
def test_get_templates_from_data_folder(self):
# Arrange
self.fs.CreateFile('/data/templates.yml',
contents="""
templates:
- name : gen1/resource
description : 1st generation shell template for basic inventory resources
repository : https://github.com/QualiSystems/shell-resource-standard
params:
project_name :
min_cs_ver: 7.0
- name : gen2/software-asset
params:
project_name :
family_name :
description : 2nd generation shell template for software assets
repository : https://github.com/QualiSystems/shellfoundry-tosca-software_asset-template
min_cs_ver: 8.1
""")
template_retriever = TemplateRetriever()
# Act
templates = template_retriever.get_templates(
alternative='/data/templates.yml')
# Assert
self.assertEqual(len(templates), 2)
self.assertTrue('gen1/resource' in templates)
self.assertTrue('gen2/software-asset' in templates)
self.assertEqual(templates['gen1/resource'].standard, None)
self.assertEqual(templates['gen2/software-asset'].standard,
'software-asset')
def test_empty_templates_should_return_when_not_templates_found_on_github(self):
# Arrange
template_retriever = TemplateRetriever()
# Act
templates = template_retriever.get_templates()
# Assert
self.assertEqual(templates, {})
def test_get_templates(self):
# Arrange
template_retriever = TemplateRetriever()
# Act
templates = template_retriever.get_templates()
# Assert
self.assertEqual(templates['router'].name, 'router')
self.assertEqual(templates['router'].description, 'Basic router template')
self.assertEqual(templates['router'].repository, 'https://github.com/QualiSystems/shellfoundry-router-template')
self.assertEqual(templates['switch'].name, 'switch')
self.assertEqual(templates['switch'].description, 'Basic switch template')
self.assertEqual(templates['switch'].repository, 'https://github.com/QualiSystems/shellfoundry-switch-template')