From 56105916271ff9a7c0bfa237b45b9b3fab9c570b Mon Sep 17 00:00:00 2001
From: Sean Hall <r.sean.hall@gmail.com>
Date: Wed, 9 Jun 2021 11:30:46 -0500
Subject: Fix building and perf for bundles with 1000s of package and BA
 payloads

Fixes other half of #5300
---
 src/test/burn/TestExe/Task.cs | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'src/test/burn/TestExe/Task.cs')

diff --git a/src/test/burn/TestExe/Task.cs b/src/test/burn/TestExe/Task.cs
index 7d39bfd9..59f774fb 100644
--- a/src/test/burn/TestExe/Task.cs
+++ b/src/test/burn/TestExe/Task.cs
@@ -59,6 +59,24 @@ namespace TestExe
         }
     }
 
+    public class GenerateFilesTask : Task
+    {
+        public GenerateFilesTask(string Data) : base(Data) { }
+
+        public override void RunTask()
+        {
+            string[] tokens = this.data.Split(new char[] { '|' }, 2);
+            string folderPath = System.Environment.ExpandEnvironmentVariables(tokens[0]);
+            long size = long.Parse(tokens[1]);
+            Directory.CreateDirectory(folderPath);
+            var bytes = new byte[0];
+            for (long i = 1; i <= size; i++)
+            {
+                File.WriteAllBytes(Path.Combine(folderPath, $"{i}.txt"), bytes);
+            }
+        }
+    }
+
     public class LargeFileTask : Task
     {
         public LargeFileTask(string Data) : base(Data) { }
@@ -163,6 +181,10 @@ namespace TestExe
                                 t = new SleepRandomTask(args[i + 1]);
                                 tasks.Add(t);
                                 break;
+                            case "/gf":
+                                t = new GenerateFilesTask(args[i + 1]);
+                                tasks.Add(t);
+                                break;
                             case "/lf":
                                 t = new LargeFileTask(args[i + 1]);
                                 tasks.Add(t);
-- 
cgit v1.2.3-55-g6feb