def test_process_comment(database, reddit):
created = utils.datetime_now()
username = "******"
comment_id = reddit_test.random_id()
thread_id = reddit_test.random_id()
comment = reddit_test.RedditObject(
body=f"{static.TRIGGER}! 1 day",
author=username,
created=created,
id=comment_id,
link_id="t3_" + thread_id,
permalink=f"/r/test/{thread_id}/_/{comment_id}/",
subreddit="test")
reddit.add_comment(comment)
comments.process_comment(comment.get_pushshift_dict(), reddit, database)
result = comment.get_first_child().body
assert "CLICK THIS LINK" in result
reminders = database.get_all_user_reminders(username)
assert len(reminders) == 1
assert reminders[0].user.name == username
assert reminders[0].message is None
assert reminders[0].source == utils.reddit_link(comment.permalink)
assert reminders[0].requested_date == created
assert reminders[0].target_date == created + timedelta(hours=24)
assert reminders[0].id is not None
assert reminders[0].recurrence is None
def test_comment_in_thread(database, reddit):
comment_id = reddit_test.random_id()
thread_id = reddit_test.random_id()
comment = reddit_test.RedditObject(
body=f"{static.TRIGGER}! 1 day",
author="Watchful1",
created=utils.datetime_now(),
id=comment_id,
link_id="t3_" + thread_id,
permalink=f"/r/test/{thread_id}/_/{comment_id}/",
subreddit="test")
reddit.add_comment(comment)
comments.process_comment(comment.get_pushshift_dict(), reddit, database)
comment_id_2 = reddit_test.random_id()
comment_2 = reddit_test.RedditObject(
body=f"{static.TRIGGER}! 1 day",
author="Watchful1",
created=utils.datetime_now(),
id=comment_id_2,
link_id="t3_" + thread_id,
permalink=f"/r/test/{thread_id}/_/{comment_id_2}/",
subreddit="test")
reddit.add_comment(comment_2)
comments.process_comment(comment_2.get_pushshift_dict(), reddit, database)
assert len(comment_2.children) == 0
assert len(reddit.sent_messages) == 1
assert reddit.sent_messages[0].author.name == static.ACCOUNT_NAME
assert "I've already replied to another comment in this thread" in reddit.sent_messages[
0].body
def test_process_comment_timezone(database, reddit):
user = database.get_or_add_user(user_name="Watchful1")
user.timezone = "America/Los_Angeles"
username = "******"
comment_id = reddit_test.random_id()
thread_id = reddit_test.random_id()
created = utils.datetime_now()
comment = reddit_test.RedditObject(
body=f"{static.TRIGGER}! 1 day",
author=username,
created=created,
id=comment_id,
link_id="t3_" + thread_id,
permalink=f"/r/test/{thread_id}/_/{comment_id}/",
subreddit="test")
reddit.add_comment(comment)
comments.process_comment(comment.get_pushshift_dict(), reddit, database)
result = comment.get_first_child().body
assert "default time zone" in result
assert "`America/Los_Angeles`" in result
reminders = database.get_all_user_reminders(username)
assert reminders[0].target_date == created + timedelta(hours=24)
def test_commenting_deleted(database, reddit):
comment_id = reddit_test.random_id()
thread_id = reddit_test.random_id()
comment = reddit_test.RedditObject(
body=f"{static.TRIGGER}! 1 day",
author="Watchful1",
created=utils.datetime_now(),
id=comment_id,
link_id="t3_" + thread_id,
permalink=f"/r/test/{thread_id}/_/{comment_id}/",
subreddit="test")
comments.process_comment(comment.get_pushshift_dict(), reddit, database)
assert len(comment.children) == 0
assert len(reddit.sent_messages) == 1
assert "it was deleted before I could get to it" in reddit.sent_messages[
0].body
def test_commenting_locked(database, reddit):
thread_id = reddit_test.random_id()
reddit.lock_thread(thread_id)
comment_id = reddit_test.random_id()
comment = reddit_test.RedditObject(
body=f"{static.TRIGGER}! 1 day",
author="Watchful1",
created=utils.datetime_now(),
id=comment_id,
link_id="t3_" + thread_id,
permalink=f"/r/test/{thread_id}/_/{comment_id}/",
subreddit="test")
reddit.add_comment(comment)
comments.process_comment(comment.get_pushshift_dict(), reddit, database)
assert len(comment.children) == 0
assert len(reddit.sent_messages) == 1
assert "the thread is locked" in reddit.sent_messages[0].body
def test_commenting_banned(database, reddit):
reddit.ban_subreddit("test")
comment_id = reddit_test.random_id()
thread_id = reddit_test.random_id()
comment = reddit_test.RedditObject(
body=f"{static.TRIGGER}! 1 day",
author="Watchful1",
created=utils.datetime_now(),
id=comment_id,
link_id="t3_" + thread_id,
permalink=f"/r/test/{thread_id}/_/{comment_id}/",
subreddit=reddit.subreddits["test"])
reddit.add_comment(comment)
comments.process_comment(comment.get_pushshift_dict(), reddit, database)
assert len(comment.children) == 0
assert len(reddit.sent_messages) == 1
assert "I'm not allowed to reply in this subreddit" in reddit.sent_messages[
0].body
def test_process_comment_split_not_start(database, reddit):
created = utils.datetime_now()
username = "******"
comment_id = reddit_test.random_id()
thread_id = reddit_test.random_id()
comment = reddit_test.RedditObject(
body=f"this is a test {static.TRIGGER_SPLIT}! 1 day",
author=username,
created=created,
id=comment_id,
link_id="t3_" + thread_id,
permalink=f"/r/test/{thread_id}/_/{comment_id}/",
subreddit="test")
reddit.add_comment(comment)
comments.process_comment(comment.get_pushshift_dict(), reddit, database)
assert len(comment.children) == 0
reminders = database.get_all_user_reminders(username)
assert len(reminders) == 0
def test_process_cakeday_comment(database, reddit):
username = "******"
user = reddit_test.User(
username,
utils.parse_datetime_string("2015-05-05 15:25:17").timestamp())
reddit.add_user(user)
created = utils.parse_datetime_string("2019-01-05 11:00:00")
comment_id = reddit_test.random_id()
thread_id = reddit_test.random_id()
comment = reddit_test.RedditObject(
body=f"{static.TRIGGER_CAKEDAY}!",
author=username,
created=created,
id=comment_id,
link_id="t3_" + thread_id,
permalink=f"/r/test/{thread_id}/_/{comment_id}/",
subreddit="test")
reddit.add_comment(comment)
utils.debug_time = utils.parse_datetime_string("2019-01-05 12:00:00")
comments.process_comment(comment.get_pushshift_dict(), reddit, database)
result = comment.get_first_child().body
assert "to remind you of your cakeday" in result
reminders = database.get_all_user_reminders(username)
assert len(reminders) == 1
assert reminders[0].user.name == username
assert reminders[0].source == utils.reddit_link(comment.permalink)
assert reminders[0].requested_date == created
assert reminders[0].target_date == utils.parse_datetime_string(
"2019-05-05 15:25:17")
assert reminders[0].id is not None
assert reminders[0].recurrence == "1 year"
assert reminders[0].message == "Happy Cakeday!"
def test_update_incorrect_comments(database, reddit):
comment_id1 = reddit_test.random_id()
thread_id1 = reddit_test.random_id()
comment1 = reddit_test.RedditObject(
body=f"{static.TRIGGER}! 1 day",
author="Watchful1",
created=utils.datetime_now(),
id=comment_id1,
link_id="t3_" + thread_id1,
permalink=f"/r/test/{thread_id1}/_/{comment_id1}/",
subreddit="test")
reddit.add_comment(comment1)
comments.process_comment(comment1.get_pushshift_dict(), reddit, database)
comment_id2 = reddit_test.random_id()
thread_id2 = reddit_test.random_id()
comment2 = reddit_test.RedditObject(
body=f"{static.TRIGGER}! 1 day",
author="Watchful1",
created=utils.datetime_now(),
id=comment_id2,
link_id="t3_" + thread_id2,
permalink=f"/r/test/{thread_id2}/_/{comment_id2}/",
subreddit="test")
reddit.add_comment(comment2)
comments.process_comment(comment2.get_pushshift_dict(), reddit, database)
comment_id3 = reddit_test.random_id()
thread_id3 = reddit_test.random_id()
comment3 = reddit_test.RedditObject(
body=f"{static.TRIGGER}! 1 day",
author="Watchful1",
created=utils.datetime_now(),
id=comment_id3,
link_id="t3_" + thread_id3,
permalink=f"/r/test/{thread_id3}/_/{comment_id3}/",
subreddit="test")
reddit.add_comment(comment3)
comments.process_comment(comment3.get_pushshift_dict(), reddit, database)
reminders = [
Reminder(
source="https://www.reddit.com/message/messages/XXXXX",
message=utils.reddit_link(comment1.permalink),
user=database.get_or_add_user("Watchful1"),
requested_date=utils.parse_datetime_string("2019-01-01 04:00:00"),
target_date=utils.parse_datetime_string("2019-01-05 05:00:00")),
Reminder(
source="https://www.reddit.com/message/messages/XXXXX",
message=utils.reddit_link(comment1.permalink),
user=database.get_or_add_user("Watchful1"),
requested_date=utils.parse_datetime_string("2019-01-01 04:00:00"),
target_date=utils.parse_datetime_string("2019-01-06 05:00:00")),
Reminder(
source="https://www.reddit.com/message/messages/XXXXX",
message=utils.reddit_link(comment1.permalink),
user=database.get_or_add_user("Watchful1"),
requested_date=utils.parse_datetime_string("2019-01-01 04:00:00"),
target_date=utils.parse_datetime_string("2019-01-07 05:00:00")),
Reminder(
source="https://www.reddit.com/message/messages/XXXXX",
message=utils.reddit_link(comment2.permalink),
user=database.get_or_add_user("Watchful1"),
requested_date=utils.parse_datetime_string("2019-01-01 04:00:00"),
target_date=utils.parse_datetime_string("2019-01-08 05:00:00")),
Reminder(
source="https://www.reddit.com/message/messages/XXXXX",
message=utils.reddit_link(comment2.permalink),
user=database.get_or_add_user("Watchful1"),
requested_date=utils.parse_datetime_string("2019-01-01 04:00:00"),
target_date=utils.parse_datetime_string("2019-01-09 05:00:00"))
]
for reminder in reminders:
database.add_reminder(reminder)
comments.update_comments(reddit, database)
assert "3 OTHERS CLICKED THIS LINK" in reddit.get_comment(
comment_id1).get_first_child().body
assert "2 OTHERS CLICKED THIS LINK" in reddit.get_comment(
comment_id2).get_first_child().body
assert "CLICK THIS LINK" in reddit.get_comment(
comment_id3).get_first_child().body
