Tuning Up The TreeView - Part 2

using System;

using DirectoryTree.VM;

namespace DirectoryTree.Test
{
   // Test Data Generator
   public static class DataFactory
   {
      private static int driveCount = 1;

      // Populate a Drive hierarchy
      public static Drive CreateDrive()
      {
         Drive drive = new Drive(null, "Drive" + driveCount++);

         for (int i = 0; i < 8; i++)
         {
            Directory dir = CreateDir(drive, false);
            for (int j = 0; j < i; j++)
               CreateDir(dir, false);
         }
         return drive;
      }

      // partial = true  - construct a partial directory without size
      //         = false - fully construct
      public static Directory CreateDir(TreeNode parent, bool partial)
      {
         Directory dir = new Directory(parent, RandomDirName() );
         if (!partial)
            dir.Size = RandomDirSize();
         return dir;
      }

      private static Random gen = new Random(32767);
      public static int RandomInt(int max) { return gen.Next(max); }

      public static int RandomDirSize()
      {
         int size = RandomInt(10000);
         if (size % 11 == 0) size = 0; // create more empties
         return size;
      }

      private const string letters = "AaBbCcDdEeFfGgHhIiJjKkLlMm";
      private static int seq;
      private static string RandomDirName()
      {
         return "Dir" +
              letters.Substring(RandomInt(letters.Length / 2) * 2, 2) +
                  seq++.ToString("D3");

      }
   }
}