def test_handle_global_vendor_point(self, global_vendor_point_handler, get_logger):
        nightly_process_config = MagicMock()
        nightly_process = NightlyProcess(nightly_process_config)

        nightly_process.handle_global_vendor_points()

        global_vendor_point_handler.return_value.process_all_global_vendor_points.assert_called_with()
    def test_handle_component_data(self, handle_numeric_points, handle_calculated_points, get_logger):
        nightly_process_config = MagicMock()

        nightly_process = NightlyProcess(nightly_process_config)
        nightly_process.handle_component_data()

        handle_calculated_points.assert_called_with()
        handle_numeric_points.assert_called_with()
    def test_handle_numeric_points(self, numeric_point_handler, get_logger):
        nightly_process_config = MagicMock()

        nightly_process = NightlyProcess(nightly_process_config)

        nightly_process.handle_numeric_points()

        numeric_point_handler.return_value.run.assert_called_with()
    def test_group_vendor_data(self, group_siemens_files, group_invensys_files, group_fieldserver_files, group_johnson_files, get_logger):
        nightly_process_config = MagicMock()

        nightly_process = NightlyProcess(nightly_process_config)
        nightly_process.group_vendor_data()
        group_johnson_files.assert_called_with()
        group_fieldserver_files.assert_called_with()
        group_invensys_files.assert_called_with()
        group_siemens_files.assert_called_with()
    def test_process_syrx_files(self, syrx_file_handler, get_logger):
        nightly_process_config = MagicMock()

        nightly_process = NightlyProcess(nightly_process_config)

        nightly_process.process_syrx_files()

        assert syrx_file_handler.return_value.syrx_upload_folder == nightly_process_config.SYRX_UPLOAD_FOLDER
        syrx_file_handler.return_value.process_all_syrx_files.assert_called_with()
    def test_process_siemens_files(self, siemens_file_handler, get_logger):
        nightly_process_config = MagicMock()

        nightly_process = NightlyProcess(nightly_process_config)

        nightly_process.process_siemens_files()

        assert siemens_file_handler.return_value.siemens_grouped_folder == nightly_process_config.SIEMENS_GROUPED_FOLDER
        assert siemens_file_handler.return_value.syrx_upload_folder == nightly_process_config.SYRX_UPLOAD_FOLDER
        siemens_file_handler.return_value.run.assert_called_with()
    def test_process_fieldserver_files(self, fieldserver_file_handler, get_logger):
        nightly_process_config = MagicMock()

        nightly_process = NightlyProcess(nightly_process_config)

        nightly_process.process_fieldserver_files()

        assert fieldserver_file_handler.return_value.fieldserver_grouped_folder == nightly_process_config.FIELDSERVER_GROUPED_FOLDER
        assert fieldserver_file_handler.return_value.syrx_upload_folder == nightly_process_config.SYRX_UPLOAD_FOLDER
        fieldserver_file_handler.return_value.run.assert_called_with()
    def test_process_johnson_files(self, johnson_file_handler, get_logger):
        nightly_process_config = MagicMock()

        nightly_process = NightlyProcess(nightly_process_config)

        nightly_process.process_johnson_files()

        assert johnson_file_handler.return_value.johnson_grouped_folder == nightly_process_config.JOHNSON_GROUPED_FOLDER
        assert johnson_file_handler.return_value.syrx_upload_folder == nightly_process_config.SYRX_UPLOAD_FOLDER
        johnson_file_handler.return_value.run.assert_called_with()
    def test_run(self, handle_global_vendor_points, handle_unmapped_syrx_nums, import_weather_data, group_vendor_data, process_vendor_data, handle_component_data, get_logger):
        nightly_process_config = MagicMock()

        nightly_process = NightlyProcess(nightly_process_config)
        nightly_process.run()

        handle_unmapped_syrx_nums.assert_called_with()
        import_weather_data.assert_called_with()
        group_vendor_data.assert_called_with()
        process_vendor_data.assert_called_with()
        handle_component_data.assert_called_with()
        handle_global_vendor_points.assert_called_with()
Пример #10
0
from config import BaseConfig
from nightly_process.main import NightlyProcess

nightly_process = NightlyProcess(BaseConfig)
nightly_process.run()