From 268640d3ce7c57ea69500d2c0ac829d5f5c90483 Mon Sep 17 00:00:00 2001
From: s5u13b <sunbiao.sun@alibaba-inc.com>
Date: Thu, 17 Oct 2024 08:38:07 +0000
Subject: [PATCH] Fix unit test

---
 Makefile                                                  | 4 ++--
 tests/unit_test/llumlet/test_local_migration_scheduler.py | 8 ++++----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/Makefile b/Makefile
index 945991b2..c3180797 100644
--- a/Makefile
+++ b/Makefile
@@ -29,13 +29,13 @@ lint: check_pylint_installed check_pytest_installed
 
 .PHONY: test
 test: check_pytest_installed
-	@pytest -x -v --ignore=third_party/ --ignore=tests/e2e_test --disable-warnings
+	@pytest -v --ignore=third_party/ --ignore=tests/e2e_test --disable-warnings
 	@python examlpes/offline_inference.py
 	@pytest -v tests/e2e_test/test_e2e.py
 
 .PHONY: unit_test
 unit_test: check_pytest_installed
-	@pytest -x -v --ignore=third_party/ --ignore=tests/e2e_test --disable-warnings
+	@pytest -v --ignore=third_party/ --ignore=tests/e2e_test --disable-warnings
 	
 .PHONY: offline_test
 offline_test:
diff --git a/tests/unit_test/llumlet/test_local_migration_scheduler.py b/tests/unit_test/llumlet/test_local_migration_scheduler.py
index c79b42d7..04dee326 100644
--- a/tests/unit_test/llumlet/test_local_migration_scheduler.py
+++ b/tests/unit_test/llumlet/test_local_migration_scheduler.py
@@ -13,7 +13,7 @@
 
 import math
 from llumnix.llumlet.local_migration_scheduler import LocalMigrationScheduler
-from llumnix.llumlet.request import LlumnixRequest, RequestInferenceType
+from llumnix.llumlet.request import LlumnixRequest, RequestInferenceType, RequestStatus
 
 class MockRequest(LlumnixRequest):
     def __init__(self, request_id, length, expected_steps) -> None:
@@ -32,11 +32,11 @@ def inference_type(self) -> RequestInferenceType:
 
     @property
     def request_len(self) -> int:
-        pass
+        return self.length
 
     @property
     def prompt_len(self) -> int:
-        pass
+        return self.length
 
     @property
     def output_len(self) -> int:
@@ -48,7 +48,7 @@ def arrival_time(self) -> float:
 
     @property
     def status(self) -> RequestStatus:
-        pass
+        return self._status
 
     @property
     def prefill_num_blocks(self) -> int: