def test_validate_flavor_with_none_flavor(self):
mock_flavor = mock.MagicMock()
mock_flavor.name = 'test_flavor'
mock_flavor.id = 'test_flavor_id'
mock_flavors = [mock_flavor]
mock_nova = mock.MagicMock()
mock_nova.flavors.list.return_value = mock_flavors
mock_os_cli = mock.MagicMock()
mock_os_cli.nova.return_value = mock_nova
attr_validator.validate_flavor(mock_os_cli, None)
self.assertEqual(False, mock_nova.flavors.list.called)
def test_validate_flavor_with_vaild_flavor(self):
mock_flavor = mock.MagicMock()
mock_flavor.name = 'test_flavor'
mock_flavor.id = 'test_flavor_id'
mock_flavors = [mock_flavor]
mock_nova = mock.MagicMock()
mock_nova.flavors.list.return_value = mock_flavors
mock_os_cli = mock.MagicMock()
mock_os_cli.nova.return_value = mock_nova
attr_validator.validate_flavor(mock_os_cli, 'test_flavor')
self.assertTrue(mock_nova.flavors.list.called)
def test_validate_flavor_with_none_flavor(self):
mock_flavor = mock.MagicMock()
mock_flavor.name = 'test_flavor'
mock_flavor.id = 'test_flavor_id'
mock_flavors = [mock_flavor]
mock_nova = mock.MagicMock()
mock_nova.flavors.list.return_value = mock_flavors
mock_os_cli = mock.MagicMock()
mock_os_cli.nova.return_value = mock_nova
attr_validator.validate_flavor(mock_os_cli, None)
self.assertFalse(mock_nova.flavors.list.called)
