lib/taketo/constructs/environment.rb in taketo-0.0.2 vs lib/taketo/constructs/environment.rb in taketo-0.0.3

@@ -1,19 +1,24 @@
+require 'taketo/constructs/base_construct'
 require 'taketo/support'
 
 module Taketo
   module Constructs
-    class Environment
-      attr_reader :name, :servers
+    class Environment < BaseConstruct
+      attr_reader :servers
 
       def initialize(name)
-        @name = name
+        super
         @servers = Taketo::Support::NamedNodesCollection.new
       end
 
       def append_server(server)
         server.environment = self
         @servers << server
+      end
+
+      def find_server(name)
+        @servers.find_by_name(name)
       end
     end
   end
 end