expanded TokenFlowPromptProviderTests

Author: AndrewMorgan1

tests/PromptStream.AI.Tests/Integration/TokenFlowPromptProviderTests.cs

@@ -0,0 +1,38 @@
+using System;
+using PromptStream.AI.Integration.TokenFlow.Services;
+using TokenFlow.Core.Models;
+using Xunit;
+
+namespace PromptStream.AI.Tests.Integration
+{
+    public class TokenFlowPromptProviderTests
+    {
+        [Fact]
+        public void GetModel_ShouldReturnModel_WhenModelExists()
+        {
+            // Act
+            var model = TokenFlowPromptProvider.GetModel("gpt-4o");
+
+            // Assert
+            Assert.NotNull(model);
+            Assert.Equal("gpt-4o", model.Id);
+            Assert.Equal("OpenAI", model.Family);
+            Assert.True(model.InputPricePer1K > 0);
+            Assert.True(model.OutputPricePer1K > 0);
+        }
+
+        [Fact]
+        public void GetModel_ShouldThrow_WhenModelDoesNotExist()
+        {
+            // Arrange
+            var invalidModelId = "nonexistent-model";
+
+            // Act & Assert
+            var ex = Assert.Throws<InvalidOperationException>(() =>
+                TokenFlowPromptProvider.GetModel(invalidModelId));
+
+            Assert.Contains("Model not found", ex.Message);
+        }
+    }
+}
+