| from . import include | |
| import unittest | |
| from unittest.mock import MagicMock | |
| from .mocks import ProviderMock | |
| import g4f | |
| from g4f.gui.server.backend import Backend_Api, get_error_message | |
| class TestBackendApi(unittest.TestCase): | |
| def setUp(self): | |
| self.app = MagicMock() | |
| self.api = Backend_Api(self.app) | |
| def test_version(self): | |
| response = self.api.get_version() | |
| self.assertIn("version", response) | |
| self.assertIn("latest_version", response) | |
| def test_get_models(self): | |
| response = self.api.get_models() | |
| self.assertIsInstance(response, list) | |
| self.assertTrue(len(response) > 0) | |
| def test_get_providers(self): | |
| response = self.api.get_providers() | |
| self.assertIsInstance(response, list) | |
| self.assertTrue(len(response) > 0) | |
| class TestUtilityFunctions(unittest.TestCase): | |
| def test_get_error_message(self): | |
| g4f.debug.last_provider = ProviderMock | |
| exception = Exception("Message") | |
| result = get_error_message(exception) | |
| self.assertEqual("ProviderMock: Exception: Message", result) | |
| if __name__ == '__main__': | |
| unittest.main() |