osmclient/v1/tests/test_vnfd.py

   1 # Copyright 2017 Sandvine
   2 #
   3 # All Rights Reserved.
   4 #
   5 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
   6 #    not use this file except in compliance with the License. You may obtain
   7 #    a copy of the License at
   8 #
   9 #         http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 #    Unless required by applicable law or agreed to in writing, software
  12 #    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  13 #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  14 #    License for the specific language governing permissions and limitations
  15 #    under the License.
  16
  17 import unittest
  18 from mock import Mock
  19 from osmclient.v1 import vnfd
  20 from osmclient.v1 import client
  21 from osmclient.common.exceptions import NotFound
  22
  23
  24 class TestVnfd(unittest.TestCase):
  25     def test_list_empty(self):
  26         mock = Mock()
  27         mock.get_cmd.return_value = list()
  28         assert len(vnfd.Vnfd(mock, client=client.Client(host='127.0.0.1')).list()) == 0
  29
  30     def test_get_notfound(self):
  31         mock = Mock()
  32         mock.get_cmd.return_value = 'foo'
  33         self.assertRaises(NotFound, vnfd.Vnfd(mock, client=client.Client(host='127.0.0.1')).get, 'bar')
  34
  35     def test_get_found(self):
  36         mock = Mock()
  37         if client.Client(host='127.0.0.1')._so_version == 'v3':
  38             mock.get_cmd.return_value = {'project-vnfd:vnfd': [{'name': 'foo'}]}
  39         else:
  40             # Backwards Compatibility
  41             mock.get_cmd.return_value = {'vnfd:vnfd': [{'name': 'foo'}]}
  42
  43         assert vnfd.Vnfd(mock, client=client.Client(host='127.0.0.1')).get('foo')