
self = <tests.test_daily_summary_llm.TestGatherDailyData object at 0x11ea45d60>
mock_accounting_cls = <MagicMock name='DBAccounting' id='4830876768'>
mock_journal_cls = <MagicMock name='DBJournal' id='4830746704'>
_mock_equity = <MagicMock name='_gather_account_equity' id='4830880848'>

    @patch("rtrader.utils.daily_summary_llm._gather_account_equity", return_value={})
    @patch("rtrader.trading_journal.DBJournal")
    @patch("rtrader.database.DBAccounting")
    def test_gathers_all_collections(self, mock_accounting_cls, mock_journal_cls, _mock_equity):
        from rtrader.utils.daily_summary_llm import gather_daily_data
    
        mock_accounting_cls.return_value = _build_accounting_mock(
            trade_journal_find_results=[[CLOSED_TRADE], [OPEN_TRADE], []],
            locks_results=[POSITION_LOCK],
            picks_result=CYCLE_PICKS_DOC,
        )
        mock_journal_cls.return_value = _build_journal_mock(JOURNAL_ENTRY)
    
        result = gather_daily_data(TRADING_DAY)
    
        assert result["trading_day"] == TRADING_DAY
        assert len(result["closed_trades"]) == 1
        assert result["closed_trades"][0]["symbol"] == "AAPL"
        assert len(result["open_trades"]) == 1
        assert result["open_trades"][0]["symbol"] == "MSFT"
        assert len(result["position_locks"]) == 1
>       assert result["cycle_picks_summary"]["total_picks"] == 3
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
E       KeyError: 'total_picks'

tests/test_daily_summary_llm.py:150: KeyError
