add function to get all tables from Server

slackhq · Aug 17, 2020 · 99951da · 99951da
1 parent 8a1a34a
commit 99951da
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/Server.php b/src/Server.php
@@ -2,7 +2,7 @@
 
 namespace Slack\SQLFake;
 
-use namespace HH\Lib\C;
+use namespace HH\Lib\{C, Dict};
 
 final class Server {
 
@@ -15,6 +15,12 @@ public static function getAll(): dict<string, this> {
     return static::$instances;
   }
 
+  public static function getAllTables(): dict<string, dict<string, dict<string, vec<dict<string, mixed>>>>> {
+    return Dict\map(static::getAll(), ($server) ==> {
+      return $server->databases;
+    });
+  }
+
   /**
    * This will override everything in $instances
    */
@@ -42,7 +48,7 @@ public static function getOrCreate(string $name): this {
   public static function cloneByName(string $name, string $clone_name): this {
     $clone = static::get($clone_name);
     if ($clone === null) {
-        throw new SQLFakeRuntimeException("Server $clone_name not found, unable to clone databases and snapshots");
+      throw new SQLFakeRuntimeException("Server $clone_name not found, unable to clone databases and snapshots");
     }
 
     $server = static::getOrCreate($name);