import pytest
from lightrag.base import BaseVectorStorage
from lightrag.utils import EmbeddingFunc

def test_base_vector_storage_integrity():
    # Just checking if we can import and inspect the class
    assert hasattr(BaseVectorStorage, '_generate_collection_suffix')
    assert hasattr(BaseVectorStorage, '_get_legacy_collection_name')
    assert hasattr(BaseVectorStorage, '_get_new_collection_name')

    # Verify methods raise NotImplementedError
    class ConcreteStorage(BaseVectorStorage):
        async def query(self, *args, **kwargs): pass
        async def upsert(self, *args, **kwargs): pass
        async def delete_entity(self, *args, **kwargs): pass
        async def delete_entity_relation(self, *args, **kwargs): pass
        async def get_by_id(self, *args, **kwargs): pass
        async def get_by_ids(self, *args, **kwargs): pass
        async def delete(self, *args, **kwargs): pass
        async def get_vectors_by_ids(self, *args, **kwargs): pass
        async def index_done_callback(self): pass
        async def drop(self): pass

    func = EmbeddingFunc(embedding_dim=128, func=lambda x: x)
    storage = ConcreteStorage(namespace="test", workspace="test", global_config={}, embedding_func=func)
    
    assert storage._generate_collection_suffix() == "unknown_128d"
    
    with pytest.raises(NotImplementedError):
        storage._get_legacy_collection_name()
        
    with pytest.raises(NotImplementedError):
        storage._get_new_collection_name()