Skip to content

Commit

Permalink
fix: test case to align with the latest codes!
Browse files Browse the repository at this point in the history
  • Loading branch information
amindadgar committed Dec 5, 2024
1 parent ba4c527 commit 3a018eb
Showing 1 changed file with 8 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -367,14 +367,17 @@ def test_fetch_raw_messages_fetch_limited_characters(self):

client[guild_id]["rawinfos"].insert_many(raw_data)

# Fetch messages from a specific date (October 3, 2023)
messages = fetch_raw_messages(guild_id, selected_channels=channels)
messages = fetch_raw_messages(
guild_id,
selected_channels=channels,
from_date=datetime(2023, 9, 20),
)
# Check if the fetched messages are equal to the expected messages
self.assertCountEqual(messages, 1)

def test_fetch_raw_messages_fetch_limited_characters(self):
def test_fetch_raw_messages_fetch_limited_characters_specified(self):
"""
fetch raw messages and do filtering
fetch raw messages and do filtering with a specified value
"""
client = MongoSingleton.get_instance().client

Expand Down Expand Up @@ -456,6 +459,7 @@ def test_fetch_raw_messages_fetch_limited_characters(self):
messages = fetch_raw_messages(
guild_id,
selected_channels=channels,
from_date=datetime(2023, 9, 20),
min_word_limit=1,
)
# Check if the fetched messages are equal to the expected messages
Expand Down

0 comments on commit 3a018eb

Please sign in to comment.