SIENTIAPDE-1110

Enhance Gates and Redis activities by adding metadata parameter to apply_aggregation and notification methods. Refactor notification handling to use send_notification for improved consistency. Update tests to reflect changes in notification method calls and ensure proper functionality with new metadata integration.

tests/activities/test_redis.py

@@ -51,6 +51,90 @@ metadata = {
 }
 
 
+@pytest.mark.asyncio
+async def test_get_last_data_timestamp_none(redis_activity):
+    """Test get_last_data_timestamp"""
+    test_data = {
+        **metadata,
+        'workflow_name': 'test_pipeline',
+        'schedule_name': 'test_schedule'
+    }
+
+    redis_activity.get = MagicMock(return_value=None)
+
+    result = await redis_activity.get_last_data_timestamp(test_data)
+
+    assert result is None
+
+
+@pytest.mark.asyncio
+async def test_get_last_data_timestamp_not_none(redis_activity):
+    """Test get_last_data_timestamp"""
+    test_data = {
+        **metadata,
+        'workflow_name': 'test_pipeline',
+        'schedule_name': 'test_schedule'
+    }
+
+    redis_activity.get = MagicMock(return_value='2023-01-01 12:00:00')
+
+    result = await redis_activity.get_last_data_timestamp(test_data)
+
+    redis_activity.get.assert_called_once_with(
+        'last_data_timestamp_test_pipeline_test_schedule'
+    )
+
+    assert result == '2023-01-01 12:00:00'
+
+
+@pytest.mark.asyncio
+async def test_put_last_data_timestamp_empty_dataframe(redis_activity):
+    """Test put_last_data_timestamp with empty dataframe"""
+    test_data = {
+        **metadata,
+        'workflow_name': 'test_pipeline',
+        'schedule_name': 'test_schedule',
+        'data': DataFrame(columns=['name', 'value', 'timestamp']).to_dict('records')
+    }
+
+    redis_activity.set = MagicMock()
+
+    result = await redis_activity.put_last_data_timestamp(test_data)
+
+    assert result is None
+
+    redis_activity.set.assert_not_called()
+
+
+@pytest.mark.asyncio
+async def test_put_last_data_timestamp_not_empty_dataframe(redis_activity):
+    """Test put_last_data_timestamp with not empty dataframe"""
+
+    data = DataFrame({
+        'name': ['sensor1', 'sensor2'],
+        'value': [25.5, 30.0],
+        'inserted_at': ['2023-01-01 12:00:00', '2023-01-01 12:00:01']
+    })
+    test_data = {
+        **metadata,
+        'workflow_name': 'test_pipeline',
+        'schedule_name': 'test_schedule',
+        'data': data.to_dict('records')
+    }
+
+    redis_activity.set = MagicMock()
+
+    result = await redis_activity.put_last_data_timestamp(test_data)
+
+    assert result == '2023-01-01 12:00:01'
+
+    redis_activity.set.assert_called_once_with(
+        'last_data_timestamp_test_pipeline_test_schedule',
+        '2023-01-01 12:00:01',
+        ttl=None
+    )
+
+
 @pytest.mark.asyncio
 async def test_group_and_hold_data_new_key(redis_activity):
     """Test group_and_hold_data with a new key"""