41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
import os
|
|
import sys
|
|
import unittest
|
|
|
|
# Add the src directory to the Python path
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../src')))
|
|
|
|
from mcp_obsidian import obsidian
|
|
|
|
class TestObsidianIntegration(unittest.TestCase):
|
|
def setUp(self):
|
|
"""Set up the environment variables for the test."""
|
|
os.environ['OBSIDIAN_API_KEY'] = 'REDACTED_API_KEY'
|
|
os.environ['OBSIDIAN_HOST'] = 'http://obsidian.obsidian.svc.cluster.local:27123'
|
|
|
|
def test_connection(self):
|
|
"""Test the connection to the Obsidian API."""
|
|
try:
|
|
protocol, host, port, path = obsidian.Obsidian.parse_host_config(os.environ['OBSIDIAN_HOST'])
|
|
|
|
api = obsidian.Obsidian(
|
|
api_key=os.environ['OBSIDIAN_API_KEY'],
|
|
protocol=protocol,
|
|
host=host,
|
|
port=port,
|
|
path=path,
|
|
verify_ssl=False
|
|
)
|
|
|
|
# Use a basic API call to verify the connection
|
|
files = api.list_files_in_vault()
|
|
|
|
self.assertIsNotNone(files)
|
|
print("Successfully connected to Obsidian API and listed files.")
|
|
|
|
except Exception as e:
|
|
self.fail(f"Failed to connect to Obsidian API: {e}")
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|