def test_recursive(self):
Options = collections.namedtuple('args',
'recursive force command_args')
options = Options(recursive=True, force=True,
command_args=['/some_path'])
command = Rm()
command.prepare(options)
client = mock.Mock()
token = Token(version=10,
name='/some_path/some_token',
owner='some_owner',
expirationTime=10,
data='some_data')
query_response = QueryResponse(tokens=[[token]])
client.query.return_value = query_response
modify_response = ModifyResponse()
client.modify.return_value = modify_response
output = command.execute(client, None)
query = Query(namePrefix='/some_path')
query_request = QueryRequest(queries=[query])
client.query.assert_called_once_with(query_request)
modify_request = ModifyRequest(deletes=[token])
client.modify.assert_called_once_with(modify_request)
self.assertEqual('removed 1 token(s)\n', output)
def test_recursive(self):
Options = collections.namedtuple('args',
'recursive force command_args')
options = Options(recursive=True, force=True,
command_args=['/some_path'])
command = Rm()
command.prepare(options)
client = mock.Mock()
token = Token(version=10,
name='/some_path/some_token',
owner='some_owner',
expirationTime=10,
data='some_data')
query_response = QueryResponse(tokens=[[token]])
client.query.return_value = query_response
modify_response = ModifyResponse()
client.modify.return_value = modify_response
output = command.execute(client, None)
query = Query(namePrefix='/some_path')
query_request = QueryRequest(queries=[query])
client.query.assert_called_once_with(query_request)
modify_request = ModifyRequest(deletes=[token])
client.modify.assert_called_once_with(modify_request)
self.assertEqual('removed 1 token(s)\n', output)
def test_empty(self):
Options = collections.namedtuple("args", "recursive force command_args")
options = Options(recursive=False, force=True, command_args=["/some_path"])
command = Rm()
command.prepare(options)
client = mock.Mock()
response = QueryResponse()
client.query.return_value = response
output = command.execute(client, None)
query = Query(namePrefix="/some_path")
request = QueryRequest(queries=[query])
client.query.assert_called_once_with(request)
self.assertEqual("no tokens found\nremoved 0 token(s)\n", output)
def test_empty(self):
Options = collections.namedtuple('args',
'recursive force command_args')
options = Options(recursive=False, force=True,
command_args=['/some_path'])
command = Rm()
command.prepare(options)
client = mock.Mock()
response = QueryResponse()
client.query.return_value = response
output = command.execute(client, None)
query = Query(namePrefix='/some_path')
request = QueryRequest(queries=[query])
client.query.assert_called_once_with(request)
self.assertEqual('no tokens found\nremoved 0 token(s)\n', output)
