import unittest
from mock import Mock
from osmclient.v1 import nsd
+from osmclient.v1 import client
from osmclient.common.exceptions import NotFound
class TestNsd(unittest.TestCase):
-
def test_list_empty(self):
mock = Mock()
mock.get_cmd.return_value = list()
- assert len(nsd.Nsd(mock).list()) == 0
+ assert len(nsd.Nsd(mock, client=client.Client(host='127.0.0.1')).list()) == 0
def test_get_notfound(self):
mock = Mock()
mock.get_cmd.return_value = 'foo'
- self.assertRaises(NotFound, nsd.Nsd(mock).get, 'bar')
+ self.assertRaises(NotFound, nsd.Nsd(mock, client=client.Client(host='127.0.0.1')).get, 'bar')
def test_get_found(self):
mock = Mock()
- mock.get_cmd.return_value = {'nsd:nsd': [{'name': 'foo'}]}
- assert nsd.Nsd(mock).get('foo')
+ if client.Client(host='127.0.0.1')._so_version == 'v3':
+ mock.get_cmd.return_value = {'project-nsd:nsd': [{'name': 'foo'}]}
+ else:
+ # Backwards Compatibility
+ mock.get_cmd.return_value = {'nsd:nsd': [{'name': 'foo'}]}
+ assert nsd.Nsd(mock, client=client.Client(host='127.0.0.1')).get('foo')