osmclient/v1/tests/test_nsd.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 nsd
  20 from osmclient.common.exceptions import NotFound
  21
  22 class TestNsd(unittest.TestCase):
  23
  24    def test_list_empty(self):
  25        mock=Mock()
  26        mock.get_cmd.return_value=list()
  27        assert len(nsd.Nsd(mock).list()) == 0
  28
  29    def test_get_notfound(self):
  30        mock=Mock()
  31        mock.get_cmd.return_value='foo'
  32        self.assertRaises(NotFound,nsd.Nsd(mock).get,'bar')
  33
  34    def test_get_found(self):
  35        mock=Mock()
  36        mock.get_cmd.return_value={'nsd:nsd': [{'name': 'foo' }]}
  37        assert nsd.Nsd(mock).get('foo')