def test_add_props(self): old_node = Node(ElementTree.fromstring(NODE_XML)) new_node = Node(ElementTree.fromstring(NODE_XML)) new_node.props['quota'] = '1000' new_node.create = Mock(return_value=new_node.node) data = str(new_node) headers = {'size': str(len(data))} client = Client() client.get_node = Mock(return_value=old_node) client.get_node_url = Mock(return_value='http://foo.com/bar') client.conn = Mock() with patch('vos.Client', client) as mock: mock.add_props(new_node) mock.conn.session.post.assert_called_with('http://foo.com/bar', headers=headers, data=data)
def test_add_props(self): old_node = Node(ElementTree.fromstring(NODE_XML)) old_node.uri = 'vos:sometest' new_node = Node(ElementTree.fromstring(NODE_XML)) new_node.props['quota'] = '1000' new_node.create = Mock(return_value=new_node.node) data = str(new_node) headers = {'size': str(len(data))} client = Client() client.get_node = Mock(return_value=old_node) client.get_node_url = Mock(return_value='http://foo.com/bar') mock_session = Mock() client.get_session = Mock(return_value=mock_session) client.add_props(new_node) client.get_session.assert_called_with('vos://foo.com!vospace/bar') mock_session.post.assert_called_with('http://foo.com/bar', headers=headers, data=data)