"""Unit tests for the create_label_top_new_releases_table tasks.""" from unittest.mock import patch from activity_detector.flows.analytics_digest.tasks.\ create_label_top_new_releases_table \ import create_label_top_new_releases_table def test_create_label_top_new_releases_table(mock_snowflake, date_params): """Test creating summary table in Snowflake.""" with patch('activity_detector.flows.analytics_digest.tasks' '.create_label_top_new_releases_table.sql_loader') \ as mock_loader: mock_loader.load_query.return_value = \ 'create_label_top_new_releases_table' create_label_top_new_releases_table( None, date_params['start_date'], date_params['end_date']) mock_snowflake.session.execute.assert_called_with( 'create_label_top_new_releases_table', { 'start_date': date_params['start_date'], 'end_date': date_params['end_date']})