mirror of
https://github.com/kennethreitz/simplemind.git
synced 2026-06-05 22:50:18 +00:00
27 lines
878 B
Python
27 lines
878 B
Python
import unittest
|
|
from unittest.mock import patch, MagicMock
|
|
from simplemind.integrations.openai import OpenAI
|
|
|
|
|
|
class TestOpenAIProvider(unittest.TestCase):
|
|
@patch("simplemind.integrations.openai.BaseOpenAI")
|
|
def setUp(self, mock_openai):
|
|
self.mock_openai = mock_openai.return_value
|
|
self.mock_openai.models.list.return_value = [MagicMock(id="gpt-4")]
|
|
self.provider = OpenAI(api_key="test_api_key", model="gpt-4")
|
|
|
|
def test_available_models(self):
|
|
models = self.provider.available_models
|
|
self.assertIn("gpt-4", models)
|
|
|
|
def test_test_connection_success(self):
|
|
self.assertTrue(self.provider.test_connection())
|
|
|
|
def test_generate_response_not_implemented(self):
|
|
with self.assertRaises(NotImplementedError):
|
|
self.provider.generate_response(None)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|