From affa92f62fd8f4484f407316ced7fad7c44a26a5 Mon Sep 17 00:00:00 2001
From: Vitalii Savitskii <visa@ecco.com>
Date: Fri, 12 Apr 2024 14:45:02 +0200
Subject: [PATCH] Fix tests

---
 src/Models/StreamClass/V1Beta1StreamClass.cs |  2 +-
 test/Services/StreamOperatorServiceTests.cs  | 18 ++++++++++++++----
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/src/Models/StreamClass/V1Beta1StreamClass.cs b/src/Models/StreamClass/V1Beta1StreamClass.cs
index 203f088..1775440 100644
--- a/src/Models/StreamClass/V1Beta1StreamClass.cs
+++ b/src/Models/StreamClass/V1Beta1StreamClass.cs
@@ -60,6 +60,6 @@ public StreamOperatorServiceConfiguration ToStreamOperatorServiceConfiguration()
     /// <inheritdoc cref="IStreamClass.KindRef"/>
     public string KindRef => this.Spec.KindRef;
     
-    /// <inheritdoc cref="IStreamClass.KindRef"/>
+    /// <inheritdoc cref="IStreamClass.MaxBufferCapacity"/>
     public int MaxBufferCapacity => this.Spec.MaxBufferCapacity;
 }
diff --git a/test/Services/StreamOperatorServiceTests.cs b/test/Services/StreamOperatorServiceTests.cs
index 31400e7..5a54033 100644
--- a/test/Services/StreamOperatorServiceTests.cs
+++ b/test/Services/StreamOperatorServiceTests.cs
@@ -7,8 +7,9 @@
 using Akka.Streams.Dsl;
 using Akka.Util;
 using Akka.Util.Extensions;
-using Arcane.Operator.Configurations;
 using Arcane.Operator.Configurations.Common;
+using Arcane.Operator.Models.StreamClass;
+using Arcane.Operator.Models.StreamClass.Base;
 using Arcane.Operator.Models.StreamDefinitions;
 using Arcane.Operator.Models.StreamDefinitions.Base;
 using Arcane.Operator.Models.StreamStatuses.StreamStatus.V1Beta1;
@@ -376,10 +377,19 @@ private ServiceProvider CreateServiceProvider()
             .AddSingleton(this.loggerFixture.Factory.CreateLogger<StreamOperatorService<FailedStreamDefinition>>())
             .AddSingleton(this.serviceFixture.MockStreamingJobOperatorService.Object)
             .AddSingleton(optionsMock.Object)
-            .AddSingleton(Options.Create(new StreamOperatorServiceConfiguration
+             // In read code StreamClass is not registered as a service, but it is used in StreamOperatorService
+            .AddSingleton<IStreamClass>(new V1Beta1StreamClass
             {
-                MaxBufferCapacity = 100
-            }))
+                Spec = new V1Beta1StreamClassSpec
+                {
+                    MaxBufferCapacity = 1000,
+                    KindRef = "StreamKind",
+                    ApiVersion = "v1",
+                    PluralName = "streams",
+                    ApiGroupRef = "example.com",
+                    StreamClassResourceKind = "StreamClass"
+                }
+            })
             .AddSingleton<IStreamOperatorService<StreamDefinition>, StreamOperatorService<StreamDefinition>>()
             .AddSingleton<IStreamOperatorService<FailedStreamDefinition>, StreamOperatorService<FailedStreamDefinition>>()
             .BuildServiceProvider();