Move data store init to MemoryStore, make it possible to use other data stores

1 files changed, 21 insertions, 0 deletions

diff --git a/lib/chef_zero/data_store/memory_store.rb b/lib/chef_zero/data_store/memory_store.rb
index 25678e0..5c94241 100644
--- a/lib/chef_zero/data_store/memory_store.rb
+++ b/lib/chef_zero/data_store/memory_store.rb
@@ -23,7 +23,28 @@ module ChefZero
   module DataStore
     class MemoryStore
       def initialize
+        clear
+      end
+
+      def clear
         @data = {}
+
+        # Create containers
+        create_dir([], 'clients')
+        create_dir([], 'cookbooks')
+        create_dir([], 'data')
+        create_dir([], 'environments')
+        create_dir([], 'file_store')
+        create_dir([], 'nodes')
+        create_dir([], 'roles')
+        create_dir([], 'sandboxes')
+        create_dir([], 'users')
+
+        # Set defaults
+        create(['clients'], 'chef-validator', '{ "validator": true }')
+        create(['clients'], 'chef-webui', '{ "admin": true }')
+        create(['environments'], '_default', '{ "description": "The default Chef environment" }')
+        create(['users'], 'admin', '{ "admin": true }')
       end
 
       def create_dir(path, name, *options)