예제 #1
0
 def test_update_network_system_properties_failure(self):
     net = NiceCXNetwork()
     net.set_name('foo')
     loader = NDExNciPidLoader(None)
     loader._ndex = MagicMock()
     loader._networksystemproperty_retry = 1
     loader._networksystemproperty_wait = 0
     loader._ndex.set_network_system_properties = MagicMock(return_value='error')
     perm_dict = {'index_level': 'ALL',
                  'showcase': True}
     res = loader._update_network_system_properties('http://public.'
                                                    'ndexbio.org'
                                                    '/v2/network/someid')
     self.assertTrue('After 1 retries' in res)
     loader._ndex.set_network_system_properties.assert_called_with('someid',
                                                                   perm_dict)
예제 #2
0
    def test_update_network_system_properties_exception_then_okay(self):
        net = NiceCXNetwork()
        net.set_name('foo')
        loader = NDExNciPidLoader(None)
        loader._ndex = MagicMock()
        loader._networksystemproperty_wait = 0
        loader._ndex.set_network_system_properties = MagicMock(side_effect=[Exception('foo'), ''])
        perm_dict = {'index_level': 'ALL',
                     'showcase': True}
        res = loader._update_network_system_properties('http://public.'
                                                       'ndexbio.org'
                                                       '/v2/network/someid')
        self.assertEqual(None, res)
        loader._ndex.set_network_system_properties.assert_called_with('someid',
                                                                      perm_dict)

        cnt = loader._ndex.set_network_system_properties.call_count
        self.assertEqual(2, cnt)