add model test1

Signed-off-by: Yang Wang <[email protected]>

Author: yangw-dev

.github/workflows/lambda-do-release-runners.yml

@@ -146,8 +146,8 @@ jobs:
           tag: ${{ inputs.tag }}
           updateOnlyUnreleased: true
 
-  release-oss-ci-aws-ce-tracking:
-    name:  Upload Release for oss-ci-aws-ce-tracking lambda
+  release-oss-ci-cur:
+    name:  Upload Release for oss-ci-cur lambda
     runs-on: ubuntu-latest
     permissions:
       contents: write
@@ -164,15 +164,15 @@ jobs:
           python-version: '3.10'
 
       - name: Build deployment.zip
-        working-directory: aws/lambda/oss_ci_aws_ce_tracking
+        working-directory: aws/lambda/oss_ci_cur
         run: make deployment.zip
 
       - name: Copy deployment.zip to root
-        run: cp aws/lambda/oss_ci_aws_ce_tracking/deployment.zip oss-ci-aws-ce-tracking.zip
+        run: cp aws/lambda/oss_ci_cur/deployment.zip oss-ci-cur.zip
 
       - uses: ncipollo/release-action@440c8c1cb0ed28b9f43e4d1d670870f059653174 # v1.16.0
         with:
-          artifacts: "oss-ci-aws-ce-tracking.zip"
+          artifacts: "oss-ci-cur.zip"
           allowUpdates: true
           draft: true
           name: ${{ inputs.tag }}

aws/lambda/oss_ci_aws_ce_tracking/.gitignore renamed to aws/lambda/oss_ci_cur/.gitignore

@@ -26,7 +26,7 @@
 }
 
 DB_NAME = "misc"
-DB_TABLE_NAME = "oss_ci_aws_ce_tracking"
+DB_TABLE_NAME = "oss_ci_cur"
 
 # todo(elainewy): make it a shared library for lambda
 def get_latest_time_from_table(
@@ -272,7 +272,7 @@ def get_time_range(
         # skip the job if the latest time is already covered
         if db_end >= end_time:
             return None  # skip the job if the latest time is already covered
-        start = max(db_end, start_time).strftime("%Y-%m-%d")
+        start = db_end.strftime("%Y-%m-%d")
         return [start, end]
 
     def start(self, args: Optional[argparse.Namespace] = None):

aws/lambda/oss_ci_aws_ce_tracking/Makefile renamed to aws/lambda/oss_ci_cur/Makefile

@@ -7,7 +7,7 @@
 )
 import logging
 
-from oss_ci_aws_ce_tracking.lambda_function import (
+from oss_ci_cur.lambda_function import (
     CostExplorerProcessor,
 )
 
@@ -69,20 +69,20 @@ class TestCostExplorerProcessor(unittest.TestCase):
     def setUp(self):
         # Mock environment variables
         envs_patcher = patch(
-            "oss_ci_aws_ce_tracking.lambda_function.ENVS",
+            "oss_ci_cur.lambda_function.ENVS",
             new=get_default_environment_variables(),
         )
         self.mock_envs = envs_patcher.start()
         self.addCleanup(envs_patcher.stop)
 
         # Mock boto3 client
-        boto3_patcher = patch("oss_ci_aws_ce_tracking.lambda_function.boto3.client")
+        boto3_patcher = patch("oss_ci_cur.lambda_function.boto3.client")
         self.mock_boto3_client = boto3_patcher.start()
         self.addCleanup(boto3_patcher.stop)
 
         # Mock get_clickhouse_client method
         get_clickhouse_client_patcher = patch(
-            "oss_ci_aws_ce_tracking.lambda_function.get_clickhouse_client"
+            "oss_ci_cur.lambda_function.get_clickhouse_client"
         )
         self.mock_get_cc = get_clickhouse_client_patcher.start()
         self.addCleanup(get_clickhouse_client_patcher.stop)