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)
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_add(self): """ Test adding a comment """ from aiida.cmdline.commands.cmd_comment import add from aiida.orm import Node node = Node() node.store() result = CliRunner().invoke( add, ['-c{}'.format(COMMENT), str(node.pk)], catch_exceptions=False) self.assertEqual(result.exit_code, 0) comment = node.get_comments() self.assertEquals(len(comment), 1) self.assertEqual(comment[0]['content'], COMMENT)