Test related fix

katyukha · Sep 5, 2016 · edc3805 · edc3805
1 parent 61fca44
commit edc3805
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/openerp_proxy/session.py b/openerp_proxy/session.py
@@ -21,8 +21,8 @@ class SessionClientExt(Client):
         used in session
     """
     def __init__(self, *args, **kwargs):
+        self._no_save = kwargs.pop('no_save', False)
         super(SessionClientExt, self).__init__(*args, **kwargs)
-        self._no_save = kwargs.get('no_save', False)
 
     def get_init_args(self):
         res = super(SessionClientExt, self).get_init_args()

diff --git a/openerp_proxy/tests/test_session.py b/openerp_proxy/tests/test_session.py
@@ -157,12 +157,6 @@ def test_20_connect_save_connect_no_save(self):
         self.assertNotIn(cl.get_url(), session.db_list)
         self.assertEqual(len(session.db_list), 0)
 
-        with self.assertRaises(ValueError):
-            session.get_db(cl.get_url())
-
-        with self.assertRaises(KeyError):
-            session[cl.get_url()]
-
     def test_25_aliases(self):
         session = Session(self._session_file_path)