Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix tests
  • Loading branch information
simeongelovski committed Apr 8, 2026
commit 3c89b52e0a60cad3f4143225c6c003cd535bb716
6 changes: 5 additions & 1 deletion tests/core/unit/telemetry/test_module.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,20 +12,23 @@ def test_module_values(self):
"""Test that Module enum has expected values."""
assert Module.AICORE.value == "aicore"
assert Module.AUDITLOG.value == "auditlog"
assert Module.AUDITLOG_NG.value == "auditlog_ng"
assert Module.DESTINATION.value == "destination"
assert Module.OBJECTSTORE.value == "objectstore"

def test_module_str_representation(self):
"""Test that Module enum converts to string correctly."""
assert str(Module.AICORE) == "aicore"
assert str(Module.AUDITLOG) == "auditlog"
assert str(Module.AUDITLOG_NG) == "auditlog_ng"
assert str(Module.DESTINATION) == "destination"
assert str(Module.OBJECTSTORE) == "objectstore"

def test_module_is_string_enum(self):
"""Test that Module enum inherits from str."""
assert isinstance(Module.AICORE, str)
assert isinstance(Module.AUDITLOG, str)
assert isinstance(Module.AUDITLOG_NG, str)
assert isinstance(Module.DESTINATION, str)

def test_module_equality(self):
Expand All @@ -44,7 +47,7 @@ def test_module_in_collection(self):
def test_all_modules_present(self):
"""Test that all expected modules are present."""
all_modules = list(Module)
assert len(all_modules) == 4
assert len(all_modules) == 5
assert Module.AICORE in all_modules
assert Module.AUDITLOG in all_modules
assert Module.DESTINATION in all_modules
Expand All @@ -55,5 +58,6 @@ def test_module_iteration(self):
module_values = [str(m) for m in Module]
assert "aicore" in module_values
assert "auditlog" in module_values
assert "auditlog_ng" in module_values
assert "destination" in module_values
assert "objectstore" in module_values
5 changes: 3 additions & 2 deletions tests/core/unit/telemetry/test_operation.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@ def test_auditlog_operations(self):
"""Test Audit Log operation values."""
assert Operation.AUDITLOG_LOG.value == "log"
assert Operation.AUDITLOG_LOG_BATCH.value == "log_batch"
assert Operation.AUDITLOG_CREATE_CLIENT.value == "create_client"

def test_destination_operations(self):
"""Test Destination operation values."""
Expand Down Expand Up @@ -105,5 +106,5 @@ def test_operation_iteration(self):
def test_operation_count(self):
"""Test that we have the expected number of operations."""
all_operations = list(Operation)
# 2 auditlog + 8 destination + 7 certificate + 7 fragment + 8 objectstore + 2 aicore = 34
assert len(all_operations) == 34
# 3 auditlog + 8 destination + 7 certificate + 7 fragment + 8 objectstore + 2 aicore = 34
assert len(all_operations) == 35
Loading