2 files changed, 68 insertions, 0 deletions
diff --git a/lib/puppet/util/user_attr.rb b/lib/puppet/util/user_attr.rb
new file mode 100644
index 000000000..db8fb81b9
--- /dev/null
+++ b/lib/puppet/util/user_attr.rb
@@ -0,0 +1,21 @@
+class UserAttr
+    def self.get_attributes_by_name(name)
+        attributes = nil
+
+        File.readlines('/etc/user_attr').each do |line|
+            next if line =~ /^#/
+
+            token = line.split(':')
+
+            if token[0] == name
+                attributes = {:name => name}
+                token[4].split(';').each do |attr|
+                    key_value = attr.split('=')
+                    attributes[key_value[0].intern] = key_value[1].strip
+                end
+                break
+            end
+        end
+        return attributes
+    end
+end
diff --git a/spec/unit/util/user_attr.rb b/spec/unit/util/user_attr.rb
new file mode 100644
index 000000000..57787b1a6
--- /dev/null
+++ b/spec/unit/util/user_attr.rb
@@ -0,0 +1,47 @@
+#!/usr/bin/env ruby
+
+Dir.chdir(File.dirname(__FILE__)) { (s = lambda { |f| File.exist?(f) ? require(f) : Dir.chdir("..") { s.call(f) } }).call("spec/spec_helper.rb") }
+
+require 'puppet/util/user_attr'
+
+describe UserAttr do
+    before do
+        user_attr = ["foo::::type=role", "bar::::type=normal;profile=foobar"]
+        File.stubs(:readlines).returns(user_attr)
+    end
+
+    describe "when getting attributes by name" do
+        it "should return nil if there is no entry for that name" do
+            UserAttr.get_attributes_by_name('baz').should == nil
+        end
+
+        it "should return a hash if there is an entry in /etc/user_attr" do
+            UserAttr.get_attributes_by_name('foo').class.should == Hash
+        end
+
+        it "should return a hash with the name value from /etc/user_attr" do
+             UserAttr.get_attributes_by_name('foo')[:name].should == 'foo'
+        end
+
+        #this test is contrived
+        #there are a bunch of possible parameters that could be in the hash
+        #the role/normal is just a the convention of the file
+        describe "when the name is a role" do
+            it "should contain :type = role" do
+                UserAttr.get_attributes_by_name('foo')[:type].should == 'role'
+            end
+        end
+
+        describe "when the name is not a role" do
+            it "should contain :type = normal" do
+                UserAttr.get_attributes_by_name('bar')[:type].should == 'normal'
+            end
+        end
+
+        describe "when the name has more attributes" do
+            it "should contain all the attributes" do
+                UserAttr.get_attributes_by_name('bar')[:profile].should == 'foobar'
+            end
+        end
+    end
+end