From 663e1a9ce8e1be708d625d63c99cd58ead80833a Mon Sep 17 00:00:00 2001
From: Daniel Copley <djcopley@users.noreply.github.com>
Date: Thu, 28 Nov 2024 23:05:23 -0500
Subject: [PATCH] add tests

---
 tests/providers/test_xai.py | 41 +++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 tests/providers/test_xai.py

diff --git a/tests/providers/test_xai.py b/tests/providers/test_xai.py
new file mode 100644
index 0000000..2101988
--- /dev/null
+++ b/tests/providers/test_xai.py
@@ -0,0 +1,41 @@
+import pytest
+
+from shelloracle.providers.xai import XAI
+
+
+class TestOpenAI:
+    @pytest.fixture
+    def xai_config(self, set_config):
+        config = {
+            "shelloracle": {"provider": "XAI"},
+            "provider": {
+                "XAI": {
+                    "api_key": "xai-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx",
+                    "model": "grok-beta",
+                }
+            },
+        }
+        set_config(config)
+
+    @pytest.fixture
+    def xai_instance(self, xai_config):
+        return XAI()
+
+    def test_name(self):
+        assert XAI.name == "XAI"
+
+    def test_api_key(self, xai_instance):
+        assert (
+            xai_instance.api_key
+            == "xai-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
+        )
+
+    def test_model(self, xai_instance):
+        assert xai_instance.model == "grok-beta"
+
+    @pytest.mark.asyncio
+    async def test_generate(self, mock_asyncopenai, xai_instance):
+        result = ""
+        async for response in xai_instance.generate(""):
+            result += response
+        assert result == "head -c 100 /dev/urandom | hexdump -C"