dsl keyword base class added

1 files changed, 10 insertions, 4 deletions

diff --git a/lib/dslkeywords/only_when.rb b/lib/dslkeywords/only_when.rb
index 485fd42..f23c33e 100644
--- a/lib/dslkeywords/only_when.rb
+++ b/lib/dslkeywords/only_when.rb
@@ -1,9 +1,15 @@
+require 'socket'
+
+require_relative 'keyword'
+
 module RCM
   # OnlyWhen (e.g. run on host foo)
-  class OnlyWhen
-    require 'socket'
+  class OnlyWhen < Keyword
+    def initialize(dsl_id)
+      super(dsl_id)
+      @conds = {}
+    end
 
-    def initialize = @conds = {}
     def is(arg) = arg
     def method_missing(method_name, *args) = @conds[method_name] = args.first
     def respond_to_missing? = true
@@ -18,7 +24,7 @@ module RCM
   # Add 'only_when' to DSL
   class DSL
     def only_when(&block)
-      conds = OnlyWhen.new
+      conds = OnlyWhen.new(id)
       conds.instance_eval(&block)
       @conds_met = conds.met?
     end