test_main.py 文件源码

def test_mock_classify_text_files_to_db(mock_manager, mock_workers,
                                        mock_indices_selector,
                                        mock_db_utils):

    mock_db_utils.connected.return_value = True
    w = mock_workers.return_value = Mock()
    man = mock_manager.return_value = Mock()

    a = Mock()
    b = Mock()

    producers = w.run_read_files_worker.return_value = [a, Mock()]
    consumers = w.run_classifying_workers.return_value = \
        [b, Mock()]

    # Bugs other fixtures if imported globally.
    from testfixtures import LogCapture
    with LogCapture() as l:
        main.classify_textfiles_to_db(Mock(), True, 1)
        assert (l.__sizeof__()) > 0

        assert mock_workers.called
        assert mock_manager.called
        assert man.Queue.called
        assert w.run_read_files_worker.called
        assert w.run_classifying_workers.called
        assert w.set_file_producers_done.called
        assert w.clear_file_producers_done.called
        assert a.join.called
        assert b.join.called