Allow callers to specify their own org defaults

1 files changed, 12 insertions, 16 deletions

diff --git a/lib/chef_zero/data_store/v1_to_v2_adapter.rb b/lib/chef_zero/data_store/v1_to_v2_adapter.rb
index 3f2276e..56641a6 100644
--- a/lib/chef_zero/data_store/v1_to_v2_adapter.rb
+++ b/lib/chef_zero/data_store/v1_to_v2_adapter.rb
@@ -3,26 +3,22 @@ require 'chef_zero/data_store/interface_v2'
 module ChefZero
   module DataStore
     class V1ToV2Adapter < ChefZero::DataStore::InterfaceV2
-      def initialize(real_store, single_org)
+      def initialize(real_store, single_org, options = {})
         @real_store = real_store
         @single_org = single_org
-        # Handle defaults per V2 specification
-        @defaults = {
-          'organizations' => {
-            single_org => {
-              'clients' => {
-                'chef-validator' => '{ "validator": true }',
-                'chef-webui' => '{ "admin": true }'
-              },
-              'environments' => {
-                '_default' => '{ "description": "The default Chef environment" }'
-              },
-              'users' => {
-                'admin' => '{ "admin": "true" }'
-              }
-            }
+        org_defaults = options[:org_defaults] || {
+          'clients' => {
+            'chef-validator' => '{ "validator": true }',
+            'chef-webui' => '{ "admin": true }'
+          },
+          'environments' => {
+            '_default' => '{ "description": "The default Chef environment" }'
+          },
+          'users' => {
+            'admin' => '{ "admin": "true" }'
           }
         }
+        @defaults = { 'organizations' => { single_org => org_defaults }}
       end
 
       attr_reader :real_store