X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osmclient%2Fv1%2Ftests%2Ftest_ns.py;h=19fc58f8942710363f276a22b0e0c6dca56c7b5e;hb=c009e0adcaaa63f0a998f47020472679fd295d38;hp=d4a6012eec939b80a4a16d66c6d11d0046b37627;hpb=e84eb31aded0cb62a5523422ddc4524f6e615209;p=osm%2Fosmclient.git

diff --git a/osmclient/v1/tests/test_ns.py b/osmclient/v1/tests/test_ns.py
index d4a6012..19fc58f 100644
--- a/osmclient/v1/tests/test_ns.py
+++ b/osmclient/v1/tests/test_ns.py
@@ -17,27 +17,28 @@
 import unittest
 from mock import Mock
 from osmclient.v1 import ns
+from osmclient.v1 import client
 from osmclient.common.exceptions import NotFound
 
 
-class TestNs(unittest.TestCase):
+class TestNs(unittest.TestCase): 
+    def test_list_empty(self):
+        mock = Mock()
+        mock.get_cmd.return_value = list()
+        assert len(ns.Ns(mock, client=client.Client(host='127.0.0.1')).list()) == 0
 
-   def test_list_empty(self):
-       mock=Mock()
-       mock.get_cmd.return_value=list()
-       assert len(ns.Ns(mock).list()) == 0
+    def test_get_notfound(self):
+        mock = Mock()
+        mock.get_cmd.return_value = 'foo'
+        self.assertRaises(NotFound, ns.Ns(mock, client=client.Client(host='127.0.0.1')).get, 'bar')
 
-   def test_get_notfound(self):
-       mock=Mock()
-       mock.get_cmd.return_value='foo'
-       self.assertRaises(NotFound,ns.Ns(mock).get,'bar')
+    def test_get_found(self):
+        mock = Mock()
+        mock.get_cmd.return_value = {'nsr:ns-instance-config':
+                                     {'nsr': [{'name': 'foo'}]}}
+        assert ns.Ns(mock, client=client.Client(host='127.0.0.1')).get('foo')
 
-   def test_get_found(self):
-       mock=Mock()
-       mock.get_cmd.return_value={'nsr:ns-instance-config': { 'nsr': [{'name': 'foo' }]}}
-       assert ns.Ns(mock).get('foo')
-
-   def test_get_monitoring_notfound(self):
-       mock=Mock()
-       mock.get_cmd.return_value='foo'
-       self.assertRaises(NotFound,ns.Ns(mock).get_monitoring,'bar')
+    def test_get_monitoring_notfound(self):
+        mock = Mock()
+        mock.get_cmd.return_value = 'foo'
+        self.assertRaises(NotFound, ns.Ns(mock, client=client.Client(host='127.0.0.1')).get_monitoring, 'bar')