30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
import unittest
|
|
|
|
from cli.message_buffer import MessageBuffer
|
|
|
|
|
|
class MessageBufferTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.buffer = MessageBuffer()
|
|
self.buffer.init_for_analysis(["market", "news"])
|
|
|
|
def test_current_report_tracks_most_recent_updated_section(self):
|
|
self.buffer.update_report_section("market_report", "Market content")
|
|
self.assertIn("Market Analysis", self.buffer.current_report)
|
|
|
|
self.buffer.update_report_section("news_report", "News content")
|
|
self.assertIn("News Analysis", self.buffer.current_report)
|
|
self.assertNotIn("Market Analysis", self.buffer.current_report)
|
|
|
|
def test_init_resets_last_updated_section(self):
|
|
self.buffer.update_report_section("market_report", "Market content")
|
|
self.assertEqual(self.buffer._last_updated_section, "market_report")
|
|
|
|
self.buffer.init_for_analysis(["fundamentals"])
|
|
self.assertIsNone(self.buffer._last_updated_section)
|
|
self.assertIsNone(self.buffer.current_report)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|