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)
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)