def test_open(self): # Invalid mode raises OSError with self.assertRaises(OSError): client = Client() client.open('vos://foo/bar', mode=-1) with self.assertRaises(OSError): client = Client() client.get_node_url = Mock(return_value=None) client.open(None, url=None) conn = Connection() mock_vofile = VOFile(['http://foo.com/bar'], conn, 'GET') client = Client() client.get_node_url = Mock(return_value=mock_vofile) vofile = client.open(None, url=None) self.assertEquals(vofile.url.URLs[0], 'http://foo.com/bar')
def test_open(self): # Invalid mode raises OSError with self.assertRaises(OSError): client = Client() client.open('vos://foo/bar', mode=-10000) with self.assertRaises(OSError): client = Client() client.get_node_url = Mock(return_value=None) client.open(None, url=None) conn = Connection() mock_vofile = VOFile(['http://foo.com/bar'], conn, 'GET') client = Client() client.get_node_url = Mock(return_value=mock_vofile) vofile = client.open(None, url=None) self.assertEquals(vofile.url.URLs[0], 'http://foo.com/bar')
def test_open(self): # Invalid mode raises OSError with self.assertRaises(OSError): client = Client() client.open('vos://foo/bar', mode=-10000) with self.assertRaises(OSError): client = Client() client.get_node_url = Mock(return_value=None) client.open(None, url=None) conn = Connection(resource_id='ivo://cadc.nrc.ca/vault') mock_vofile = VOFile(['http://foo.com/bar'], conn, 'GET') client = Client() client.get_node_url = Mock(return_value=mock_vofile) endpoints_mock = Mock(conn=conn) client.get_endpoints = Mock(return_value=endpoints_mock) vofile = client.open(None, url=None) self.assertEqual(vofile.url.URLs[0], 'http://foo.com/bar')
def test_getNode(self): """ @return: """ uri = "vos://foo.com!vospace/bar" nodes = (' <vos:nodes>\n' '<vos:node uri="vos://cadc.nrc.ca!vospace/mydir/file123" ' 'xs:type="vos:DataNode">\n' ' <vos:properties>\n' ' <vos:property ' 'uri="ivo://ivoa.net/vospace/core#date">2016-05-10T09:52:13' '</vos:property>\n' ' </vos:properties>\n' '</vos:node>\n' '<vos:node uri="vos://cadc.nrc.ca!vospace/mydir/file456" ' 'xs:type="vos:DataNode">\n' ' <vos:properties>\n' ' <vos:property uri="ivo://ivoa.net/vospace/core#date">' '2016-05-19T09:52:14</vos:property>\n' ' </vos:properties>\n' '</vos:node>\n' '</vos:nodes>\n') mock_vofile = Mock() client = Client() client.open = Mock(return_value=mock_vofile) mock_vofile.read = Mock( return_value=NODE_XML.format(uri, '').encode('UTF-8')) my_node = client.get_node(uri, limit=0, force=False) self.assertEqual(uri, my_node.uri) self.assertEqual(len(my_node.node_list), 0) mock_vofile.read = Mock( return_value=NODE_XML.format(uri, nodes).encode('UTF-8')) my_node = client.get_node(uri, limit=2, force=True) self.assertEqual(uri, my_node.uri) self.assertEqual(len(my_node.node_list), 2) my_node = client.get_node(uri, limit=2, force=False) self.assertEqual(uri, my_node.uri) self.assertEqual(len(my_node.node_list), 2)