def test_comment_show(self):
""" Test showing an existing comment """
from aiida.cmdline.commands.cmd_comment import show
from aiida.orm import Node
node = Node()
node.store()
node.add_comment(COMMENT)
result = CliRunner().invoke(show, [str(node.pk)],
catch_exceptions=False)
self.assertNotEqual(result.output.find(COMMENT), -1)
self.assertEqual(result.exit_code, 0)
def test_comment_remove_all(self):
""" Test removing all comments from a node """
from aiida.cmdline.commands.cmd_comment import remove
from aiida.orm import Node
node = Node()
node.store()
for _ in range(10):
node.add_comment(COMMENT)
self.assertEqual(len(node.get_comments()), 10)
result = CliRunner().invoke(remove, [str(node.pk), '--all', '--force'],
catch_exceptions=False)
self.assertEqual(result.exit_code, 0)
self.assertEqual(len(node.get_comments()), 0)
def test_comment_remove(self):
""" Test removing a comment """
from aiida.cmdline.commands.cmd_comment import remove
from aiida.orm import Node
node = Node()
node.store()
comment_id = node.add_comment(COMMENT)
self.assertEquals(len(node.get_comments()), 1)
result = CliRunner().invoke(
remove, [str(node.pk), str(comment_id), '--force'],
catch_exceptions=False)
self.assertEqual(result.exit_code, 0)
self.assertEquals(len(node.get_comments()), 0)