Make clear restore org defaults

1 files changed, 8 insertions, 3 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 29dbd19..3b2fc8d 100644
--- a/lib/chef_zero/data_store/v1_to_v2_adapter.rb
+++ b/lib/chef_zero/data_store/v1_to_v2_adapter.rb
@@ -6,8 +6,8 @@ module ChefZero
       def initialize(real_store, single_org, options = {})
         @real_store = real_store
         @single_org = single_org
-        org_defaults = options[:org_defaults] || {}
-        @defaults = { 'organizations' => { single_org => org_defaults }}
+        @options = options
+        clear
       end
 
       ORG_DEFAULTS = {
@@ -27,7 +27,12 @@ module ChefZero
       attr_reader :single_org
 
       def clear
-        real_store.clear
+        if @options[:org_defaults]
+          @defaults = { 'organizations' => { @single_org => @options[:org_defaults] }}
+        else
+          @defaults = {}
+        end
+        real_store.clear if real_store.respond_to?(:clear)
       end
 
       def create_dir(path, name, *options)