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