LBRYCommunity/comment-server
1
0
Fork 0
Code Issues 18 Pull requests 1 Projects Releases Packages Wiki Activity

Updates unittests to be compliant with batched hide comments refactor

Browse source
This commit is contained in:
Oleg Silkin 2019-08-09 01:20:00 -04:00
parent 7f9e475324
commit 332fa2bcd0
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/database_test.py
View file

@ -10,7 +10,7 @@ from src.database.queries import get_comment_ids
from src.database.queries import get_claim_comments from src.database.queries import get_claim_comments
from src.database.queries import get_claim_hidden_comments from src.database.queries import get_claim_hidden_comments
from src.database.writes import create_comment_or_error from src.database.writes import create_comment_or_error
from src.database.queries import hide_comment_by_id from src.database.queries import hide_comments_by_id
from src.database.queries import delete_comment_by_id from src.database.queries import delete_comment_by_id
from tests.testcase import DatabaseTestCase from tests.testcase import DatabaseTestCase
@ -199,11 +199,11 @@ class TestDatabaseOperations(DatabaseTestCase):
comm = create_comment_or_error(self.conn, 'Comment #1', self.claimId, '1'*40, '@Doge123', 'a'*128, '123') comm = create_comment_or_error(self.conn, 'Comment #1', self.claimId, '1'*40, '@Doge123', 'a'*128, '123')
comment = get_comments_by_id(self.conn, [comm['comment_id']]).pop() comment = get_comments_by_id(self.conn, [comm['comment_id']]).pop()
self.assertFalse(comment['is_hidden']) self.assertFalse(comment['is_hidden'])
success = hide_comment_by_id(self.conn, comm['comment_id']) success = hide_comments_by_id(self.conn, [comm['comment_id']])
self.assertTrue(success) self.assertTrue(success)
comment = get_comments_by_id(self.conn, [comm['comment_id']]).pop() comment = get_comments_by_id(self.conn, [comm['comment_id']]).pop()
self.assertTrue(comment['is_hidden']) self.assertTrue(comment['is_hidden'])
success = hide_comment_by_id(self.conn, comm['comment_id']) success = hide_comments_by_id(self.conn, [comm['comment_id']])
self.assertTrue(success) self.assertTrue(success)
comment = get_comments_by_id(self.conn, [comm['comment_id']]).pop() comment = get_comments_by_id(self.conn, [comm['comment_id']]).pop()
self.assertTrue(comment['is_hidden']) self.assertTrue(comment['is_hidden'])
@ -211,11 +211,13 @@ class TestDatabaseOperations(DatabaseTestCase):
def test08DeleteComments(self): def test08DeleteComments(self):
comm = create_comment_or_error(self.conn, 'Comment #1', self.claimId, '1'*40, '@Doge123', 'a'*128, '123') comm = create_comment_or_error(self.conn, 'Comment #1', self.claimId, '1'*40, '@Doge123', 'a'*128, '123')
comments = get_claim_comments(self.conn, self.claimId) comments = get_claim_comments(self.conn, self.claimId)
self.assertIn(comm, comments['items']) match = list(filter(lambda x: comm['comment_id'] == x['comment_id'], comments['items']))
self.assertTrue(match)
deleted = delete_comment_by_id(self.conn, comm['comment_id']) deleted = delete_comment_by_id(self.conn, comm['comment_id'])
self.assertTrue(deleted) self.assertTrue(deleted)
comments = get_claim_comments(self.conn, self.claimId) comments = get_claim_comments(self.conn, self.claimId)
self.assertNotIn(comm, comments['items']) match = list(filter(lambda x: comm['comment_id'] == x['comment_id'], comments['items']))
self.assertFalse(match)
deleted = delete_comment_by_id(self.conn, comm['comment_id']) deleted = delete_comment_by_id(self.conn, comm['comment_id'])
self.assertFalse(deleted) self.assertFalse(deleted)
@ -261,7 +263,7 @@ class ListDatabaseTest(DatabaseTestCase):
self.assertEqual(len(comments), comment_list['total_items']) self.assertEqual(len(comments), comment_list['total_items'])
self.assertIn('has_hidden_comments', comment_list) self.assertIn('has_hidden_comments', comment_list)
self.assertFalse(comment_list['has_hidden_comments']) self.assertFalse(comment_list['has_hidden_comments'])
hide_comment_by_id(self.conn, comm2['comment_id']) hide_comments_by_id(self.conn, [comm2['comment_id']])
default_comments = get_claim_hidden_comments(self.conn, claim_id) default_comments = get_claim_hidden_comments(self.conn, claim_id)
self.assertIn('has_hidden_comments', default_comments) self.assertIn('has_hidden_comments', default_comments)