Notification operations test hooks (#1897)

Co-authored-by: stas <statis@microsoft.com>

src/ApiService/ApiService/TestHooks/NodeOperationsTestHooks.cs

@@ -7,7 +7,6 @@ using Microsoft.OneFuzz.Service;
 using Microsoft.OneFuzz.Service.OneFuzzLib.Orm;
 
 
-
 #if DEBUG
 namespace ApiService.TestHooks {
 
@@ -186,9 +185,11 @@ namespace ApiService.TestHooks {
                 numResults = int.Parse(query["numResults"]);
             }
             var r = _nodeOps.SearchStates(poolId, scaleSetId, states, poolName, excludeUpdateScheduled, numResults);
-            var resp = req.CreateResponse(HttpStatusCode.OK);
 
-            await resp.WriteStringAsync(JsonSerializer.Serialize(await r.ToListAsync(), EntityConverter.GetJsonSerializerOptions()));
+
+            var json = JsonSerializer.Serialize(await r.ToListAsync(), EntityConverter.GetJsonSerializerOptions());
+            var resp = req.CreateResponse(HttpStatusCode.OK);
+            await resp.WriteStringAsync(json);
             return resp;
         }
 

src/ApiService/ApiService/TestHooks/NotificationOperationsTestHooks.cs

@@ -0,0 +1,82 @@
+using System.Net;
+using System.Text.Json;
+using System.Threading.Tasks;
+using Microsoft.Azure.Functions.Worker;
+using Microsoft.Azure.Functions.Worker.Http;
+using Microsoft.OneFuzz.Service;
+using Microsoft.OneFuzz.Service.OneFuzzLib.Orm;
+
+
+#if DEBUG
+namespace ApiService.TestHooks {
+    public class NotificationOperationsTestHooks {
+
+        private readonly ILogTracer _log;
+        private readonly IConfigOperations _configOps;
+        private readonly INotificationOperations _notificationOps;
+
+        public NotificationOperationsTestHooks(ILogTracer log, IConfigOperations configOps, INotificationOperations notificationOps) {
+            _log = log.WithTag("TestHooks", nameof(NotificationOperationsTestHooks));
+            _configOps = configOps; ;
+            _notificationOps = notificationOps;
+        }
+
+        [Function("NewFilesTestHook")]
+        public async Task<HttpResponseData> NewFiles([HttpTrigger(AuthorizationLevel.Anonymous, "put", Route = "testhooks/notificationOperations/newFiles")] HttpRequestData req) {
+            _log.Info("new files");
+            var query = UriExtension.GetQueryComponents(req.Url);
+
+            var container = query["container"];
+            var fileName = query["fileName"];
+            var failTaskOnTransientError = UriExtension.GetBoolValue("failTaskOnTransientError", query, true);
+
+            await _notificationOps.NewFiles(new Container(container), fileName, failTaskOnTransientError);
+            var resp = req.CreateResponse(HttpStatusCode.OK);
+            return resp;
+        }
+
+        [Function("GetNotificationsTestHook")]
+        public async Task<HttpResponseData> GetNotifications([HttpTrigger(AuthorizationLevel.Anonymous, "get", Route = "testhooks/notificationOperations/getNotifications")] HttpRequestData req) {
+            _log.Info("get notifications");
+
+            var s = await req.ReadAsStringAsync();
+
+            var query = UriExtension.GetQueryComponents(req.Url);
+            var container = query["container"];
+            var notifications = _notificationOps.GetNotifications(new Container(container));
+
+            var json = JsonSerializer.Serialize(await notifications.ToListAsync(), EntityConverter.GetJsonSerializerOptions());
+            var resp = req.CreateResponse(HttpStatusCode.OK);
+            await resp.WriteStringAsync(json);
+            return resp;
+        }
+
+        [Function("GetQueueTasksTestHook")]
+        public async Task<HttpResponseData> GetQueueTasksTestHook([HttpTrigger(AuthorizationLevel.Anonymous, "get", Route = "testhooks/notificationOperations/getQueueTasks")] HttpRequestData req) {
+
+            _log.Info("get queue tasks");
+            var queueuTasks = _notificationOps.GetQueueTasks();
+
+            var json = JsonSerializer.Serialize(await queueuTasks.ToListAsync(), EntityConverter.GetJsonSerializerOptions());
+            var resp = req.CreateResponse(HttpStatusCode.OK);
+            await resp.WriteStringAsync(json);
+            return resp;
+        }
+
+
+        [Function("GetRegressionReportTaskTestHook")]
+        public async Task<HttpResponseData> GetRegressionReportTask([HttpTrigger(AuthorizationLevel.Anonymous, "post", Route = "testhooks/notificationOperations/getRegressionReportTask")] HttpRequestData req) {
+            _log.Info("get regression report task");
+
+            var s = await req.ReadAsStringAsync();
+            var report = JsonSerializer.Deserialize<RegressionReport>(s!, EntityConverter.GetJsonSerializerOptions());
+            var task = (_notificationOps as NotificationOperations)!.GetRegressionReportTask(report);
+
+            var json = JsonSerializer.Serialize(task, EntityConverter.GetJsonSerializerOptions());
+            var resp = req.CreateResponse(HttpStatusCode.OK);
+            await resp.WriteStringAsync(json);
+            return resp;
+        }
+    }
+}
+#endif

src/ApiService/ApiService/onefuzzlib/NotificationOperations.cs

@@ -6,6 +6,8 @@ namespace Microsoft.OneFuzz.Service;
 
 public interface INotificationOperations : IOrm<Notification> {
     Async.Task NewFiles(Container container, string filename, bool failTaskOnTransientError);
+    IAsyncEnumerable<Notification> GetNotifications(Container container);
+    IAsyncEnumerable<(Task, IEnumerable<string>)> GetQueueTasks();
 }
 
 public class NotificationOperations : Orm<Notification>, INotificationOperations {
@@ -81,7 +83,7 @@ public class NotificationOperations : Orm<Notification>, INotificationOperations
             .Where(taskTuple => taskTuple.Item2 != null)!;
     }
 
-    private async Async.Task<Task?> GetRegressionReportTask(RegressionReport report) {
+    public async Async.Task<Task?> GetRegressionReportTask(RegressionReport report) {
         if (report.CrashTestResult.CrashReport != null) {
             return await _context.TaskOperations.GetByJobIdAndTaskId(report.CrashTestResult.CrashReport.JobId, report.CrashTestResult.CrashReport.TaskId);
         }

src/ApiService/Tests/OrmModelsTest.cs

@@ -911,6 +911,12 @@ namespace Tests {
             return Test(e);
         }
 
+        [Property]
+        public bool Container(Container c) {
+            return Test(c);
+        }
+
+
         /*
         //Sample function on how repro a failing test run, using Replay
         //functionality of FsCheck. Feel free to