snaps/config/tests/volume_tests.py

   1 # Copyright (c) 2017 Cable Television Laboratories, Inc. ("CableLabs")
   2 #                    and others.  All rights reserved.
   3 #
   4 # Licensed under the Apache License, Version 2.0 (the "License");
   5 # you may not use this file except in compliance with the License.
   6 # You may obtain a copy of the License at:
   7 #
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15 import unittest
  16
  17 from snaps.config.volume import VolumeConfigError, VolumeConfig
  18
  19
  20 class VolumeConfigUnitTests(unittest.TestCase):
  21     """
  22     Tests the construction of the VolumeConfig class
  23     """
  24
  25     def test_no_params(self):
  26         with self.assertRaises(VolumeConfigError):
  27             VolumeConfig()
  28
  29     def test_empty_config(self):
  30         with self.assertRaises(VolumeConfigError):
  31             VolumeConfig(**dict())
  32
  33     def test_name_only(self):
  34         settings = VolumeConfig(name='foo')
  35         self.assertEqual('foo', settings.name)
  36         self.assertIsNone(settings.project_name)
  37         self.assertIsNone(settings.description)
  38         self.assertEquals(1, settings.size)
  39         self.assertIsNone(settings.image_name)
  40         self.assertIsNone(settings.type_name)
  41         self.assertIsNone(settings.availability_zone)
  42         self.assertFalse(settings.multi_attach)
  43
  44     def test_config_with_name_only(self):
  45         settings = VolumeConfig(**{'name': 'foo'})
  46         self.assertEqual('foo', settings.name)
  47         self.assertIsNone(settings.project_name)
  48         self.assertIsNone(settings.description)
  49         self.assertEquals(1, settings.size)
  50         self.assertIsNone(settings.image_name)
  51         self.assertIsNone(settings.type_name)
  52         self.assertIsNone(settings.availability_zone)
  53         self.assertFalse(settings.multi_attach)
  54
  55     def test_all_strings(self):
  56         settings = VolumeConfig(
  57             name='foo', project_name='proj-foo', description='desc', size='2',
  58             image_name='image', type_name='type', availability_zone='zone1',
  59             multi_attach='true')
  60
  61         self.assertEqual('foo', settings.name)
  62         self.assertEqual('proj-foo', settings.project_name)
  63         self.assertEqual('desc', settings.description)
  64         self.assertEqual(2, settings.size)
  65         self.assertEqual('image', settings.image_name)
  66         self.assertEqual('type', settings.type_name)
  67         self.assertEqual('zone1', settings.availability_zone)
  68         self.assertTrue(settings.multi_attach)
  69
  70     def test_all_correct_type(self):
  71         settings = VolumeConfig(
  72             name='foo', project_name='proj-foo', description='desc', size=2,
  73             image_name='image', type_name='bar', availability_zone='zone1',
  74             multi_attach=True)
  75
  76         self.assertEqual('foo', settings.name)
  77         self.assertEqual('proj-foo', settings.project_name)
  78         self.assertEqual('desc', settings.description)
  79         self.assertEqual(2, settings.size)
  80         self.assertEqual('image', settings.image_name)
  81         self.assertEqual('bar', settings.type_name)
  82         self.assertEqual('zone1', settings.availability_zone)
  83         self.assertTrue(settings.multi_attach)
  84
  85     def test_config_all(self):
  86         settings = VolumeConfig(
  87             **{'name': 'foo', 'project_name': 'proj-foo',
  88                'description': 'desc', 'size': '2',
  89                'image_name': 'foo', 'type_name': 'bar',
  90                'availability_zone': 'zone1', 'multi_attach': 'true'})
  91
  92         self.assertEqual('foo', settings.name)
  93         self.assertEqual('proj-foo', settings.project_name)
  94         self.assertEqual('desc', settings.description)
  95         self.assertEqual(2, settings.size)
  96         self.assertEqual('foo', settings.image_name)
  97         self.assertEqual('bar', settings.type_name)
  98         self.assertEqual('zone1', settings.availability_zone)
  99         self.assertTrue(settings.multi_attach)