Skip to content

Commit aa4242c

Browse files
committed
pytest
Signed-off-by: rakdutta <rakhibiswas@yahoo.com>
1 parent a2c2c17 commit aa4242c

File tree

2 files changed

+16
-16
lines changed

2 files changed

+16
-16
lines changed

tests/unit/mcpgateway/services/test_resource_service.py

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def mock_resource():
7373
resource.text_content = "Test content"
7474
resource.binary_content = None
7575
resource.size = 12
76-
resource.is_active = True
76+
resource.enabled = True
7777
resource.created_by = "test_user"
7878
resource.modified_by = "test_user"
7979
resource.created_at = datetime.now(timezone.utc)
@@ -110,7 +110,7 @@ def mock_resource_template():
110110
resource.text_content = "Test content"
111111
resource.binary_content = None
112112
resource.size = 12
113-
resource.is_active = True
113+
resource.enabled = True
114114
resource.created_by = "test_user"
115115
resource.modified_by = "test_user"
116116
resource.created_at = datetime.now(timezone.utc)
@@ -147,7 +147,7 @@ def mock_inactive_resource():
147147
resource.text_content = None
148148
resource.binary_content = None
149149
resource.size = 0
150-
resource.is_active = False
150+
resource.enabled = False
151151
resource.created_by = "test_user"
152152
resource.modified_by = "test_user"
153153
resource.created_at = datetime.now(timezone.utc)
@@ -230,7 +230,7 @@ async def test_register_resource_success(self, resource_service, mock_db, sample
230230
description=sample_resource_create.description or "",
231231
mime_type="text/plain",
232232
size=len(sample_resource_create.content),
233-
is_active=True,
233+
enabled=True,
234234
created_at=datetime.now(timezone.utc),
235235
updated_at=datetime.now(timezone.utc),
236236
template=None,
@@ -346,7 +346,7 @@ async def test_register_resource_binary_content(self, resource_service, mock_db)
346346
description=binary_resource.description or "",
347347
mime_type="application/octet-stream",
348348
size=len(binary_resource.content),
349-
is_active=True,
349+
enabled=True,
350350
created_at=datetime.now(timezone.utc),
351351
updated_at=datetime.now(timezone.utc),
352352
template=None,
@@ -535,7 +535,7 @@ async def test_toggle_resource_status_activate(self, resource_service, mock_db,
535535
description=mock_inactive_resource.description or "",
536536
mime_type=mock_inactive_resource.mime_type or "text/plain",
537537
size=mock_inactive_resource.size or 0,
538-
is_active=True,
538+
enabled=True,
539539
created_at=datetime.now(timezone.utc),
540540
updated_at=datetime.now(timezone.utc),
541541
template=None,
@@ -553,7 +553,7 @@ async def test_toggle_resource_status_activate(self, resource_service, mock_db,
553553

554554
result = await resource_service.toggle_resource_status(mock_db, 2, activate=True)
555555

556-
assert mock_inactive_resource.is_active is True
556+
assert mock_inactive_resource.enabled is True
557557
mock_db.commit.assert_called_once()
558558

559559
@pytest.mark.asyncio
@@ -569,7 +569,7 @@ async def test_toggle_resource_status_deactivate(self, resource_service, mock_db
569569
description=mock_resource.description,
570570
mime_type=mock_resource.mime_type,
571571
size=mock_resource.size,
572-
is_active=False,
572+
enabled=False,
573573
created_at=datetime.now(timezone.utc),
574574
updated_at=datetime.now(timezone.utc),
575575
template=None,
@@ -587,7 +587,7 @@ async def test_toggle_resource_status_deactivate(self, resource_service, mock_db
587587

588588
result = await resource_service.toggle_resource_status(mock_db, 1, activate=False)
589589

590-
assert mock_resource.is_active is False
590+
assert mock_resource.enabled is False
591591
mock_db.commit.assert_called_once()
592592

593593
@pytest.mark.asyncio
@@ -605,7 +605,7 @@ async def test_toggle_resource_status_not_found(self, resource_service, mock_db)
605605
async def test_toggle_resource_status_no_change(self, resource_service, mock_db, mock_resource):
606606
"""Test toggling status when no change needed."""
607607
mock_db.get.return_value = mock_resource
608-
mock_resource.is_active = True
608+
mock_resource.enabled = True
609609

610610
with patch.object(resource_service, "_convert_resource_to_read") as mock_convert:
611611
mock_convert.return_value = ResourceRead(
@@ -615,7 +615,7 @@ async def test_toggle_resource_status_no_change(self, resource_service, mock_db,
615615
description=mock_resource.description,
616616
mime_type=mock_resource.mime_type,
617617
size=mock_resource.size,
618-
is_active=True,
618+
enabled=True,
619619
created_at=datetime.now(timezone.utc),
620620
updated_at=datetime.now(timezone.utc),
621621
template=None,
@@ -656,7 +656,7 @@ async def test_update_resource_success(self, resource_service, mock_db, mock_res
656656
description="Updated description",
657657
mime_type="text/plain",
658658
size=15, # length of "Updated content"
659-
is_active=True,
659+
enabled=True,
660660
created_at=datetime.now(timezone.utc),
661661
updated_at=datetime.now(timezone.utc),
662662
template=None,
@@ -729,7 +729,7 @@ async def test_update_resource_binary_content(self, resource_service, mock_db, m
729729
description=mock_resource.description,
730730
mime_type="application/octet-stream",
731731
size=len(b"new binary content"),
732-
is_active=True,
732+
enabled=True,
733733
created_at=datetime.now(timezone.utc),
734734
updated_at=datetime.now(timezone.utc),
735735
template=None,
@@ -1338,7 +1338,7 @@ async def test_notify_resource_activated(self, resource_service, mock_resource):
13381338
resource_service._event_service.publish_event.assert_called_once()
13391339
call_args = resource_service._event_service.publish_event.call_args[0][0]
13401340
assert call_args["type"] == "resource_activated"
1341-
assert call_args["data"]["is_active"] is True
1341+
assert call_args["data"]["enabled"] is True
13421342

13431343
@pytest.mark.asyncio
13441344
async def test_notify_resource_deactivated(self, resource_service, mock_resource):
@@ -1350,7 +1350,7 @@ async def test_notify_resource_deactivated(self, resource_service, mock_resource
13501350
resource_service._event_service.publish_event.assert_called_once()
13511351
call_args = resource_service._event_service.publish_event.call_args[0][0]
13521352
assert call_args["type"] == "resource_deactivated"
1353-
assert call_args["data"]["is_active"] is False
1353+
assert call_args["data"]["enabled"] is False
13541354

13551355
@pytest.mark.asyncio
13561356
async def test_notify_resource_deleted(self, resource_service):

tests/unit/mcpgateway/test_admin.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -920,7 +920,7 @@ async def test_admin_get_prompt_with_detailed_metrics(self, mock_get_prompt_deta
920920
"arguments": [{"name": "var", "type": "string"}],
921921
"created_at": datetime.now(timezone.utc),
922922
"updated_at": datetime.now(timezone.utc),
923-
"is_active": True,
923+
"enabled": True,
924924
"metrics": {
925925
"total_executions": 1000,
926926
"successful_executions": 950,

0 commit comments

Comments
 (0)