kuality-coeus 0.0.3 → 0.0.4

data/lib/kuality-coeus/data_objects/budget/budget_periods.rb

@@ -5,9 +5,10 @@ class BudgetPeriodObject
   include StringFactory
   include Navigation
 
-  attr_accessor :document_id, :budget_name, :number, :start_date, :end_date,
-                :total_sponsor_cost, :direct_cost, :f_and_a_cost, :unrecovered_f_and_a,
-                :cost_sharing, :cost_limit, :direct_cost_limit, :datified
+  attr_accessor :number, :start_date, :end_date, :total_sponsor_cost,
+                :direct_cost, :f_and_a_cost, :unrecovered_f_and_a,
+                :cost_sharing, :cost_limit, :direct_cost_limit, :datified,
+                :budget_name
 
   def initialize(browser, opts={})
     @browser = browser
@@ -17,7 +18,7 @@ class BudgetPeriodObject
     }
 
     set_options(defaults.merge(opts))
-    requires :document_id, :budget_name, :start_date
+    requires :start_date, :budget_name
     datify
   end
 

data/lib/kuality-coeus/data_objects/budget/budget_versions.rb

@@ -14,7 +14,7 @@ class BudgetVersionsObject
                 :project_start_date, :project_end_date, :total_direct_cost_limit,
                 :budget_periods, :unrecovered_fa_rate_type, :f_and_a_rate_type,
                 :submit_cost_sharing, :residual_funds, :total_cost_limit,
-                :subaward_budgets
+                :subaward_budgets, :personnel
 
 
   def initialize(browser, opts={})
@@ -25,7 +25,8 @@ class BudgetVersionsObject
       cost_sharing:      '0.00',
       f_and_a:           '0.00',
       budget_periods:    BudgetPeriodsCollection.new,
-      subaward_budgets:  SubawardBudgetCollection.new
+      subaward_budgets:  SubawardBudgetCollection.new,
+      personnel:         BudgetPersonnelCollection.new
     }
 
     set_options(defaults.merge(opts))
@@ -67,7 +68,6 @@ class BudgetVersionsObject
 
   def add_period opts={}
     defaults={
-        document_id: @document_id,
         budget_name: @name,
         doc_type: @doc_header
     }
@@ -94,10 +94,13 @@ class BudgetVersionsObject
   # Use it for editing the Budget Version while on the Proposal, but not the Periods
   def edit opts={}
     navigate
-    on BudgetVersions do |edit|
-      edit.final(@name).fit opts[:final]
-      edit.budget_status(@name).fit opts[:budget_status]
-      # TODO: More here as needed...
+    on(BudgetVersions).open @name
+    confirmation
+    on Parameters do |edit|
+      edit.final.fit opts[:final]
+      edit.budget_status.fit opts[:budget_status]
+      edit.total_direct_cost_limit.fit opts[:total_direct_cost_limit]
+      # TODO: More to add here...
       edit.save
     end
     set_options(opts)
@@ -153,6 +156,14 @@ class BudgetVersionsObject
     @subaward_budgets << sab
   end
 
+  def add_project_personnel(opts={})
+    open_budget
+    on(Parameters).personnel
+    person = make BudgetPersonnelObject, opts
+    person.create
+    @personnel << person
+  end
+
   # =======
   private
   # =======

data/lib/kuality-coeus/data_objects/budget/personnel.rb

@@ -0,0 +1,135 @@
+class BudgetPersonnelObject
+
+  include Foundry
+  include DataFactory
+  include StringFactory
+  include Navigation
+  include Utilities
+
+  attr_accessor :type, :name, :job_code, :appointment_type, :base_salary,
+                :salary_effective_date, :salary_anniversary_date,
+                # TODO: Some day we are going to have to allow for multiple codes and periods, here...
+                :object_code_name, :start_date, :end_date, :percent_effort, :percent_charged,
+                :period_type, :requested_salary, :calculated_fringe
+                #TODO: Add more variables here - "apply inflation", "submit cost sharing", etc.
+
+  def initialize(browser, opts={})
+    @browser = browser
+
+    defaults = {
+        # Note: 'type' must be one of:
+        # 'employee', 'non_employee', or 'to_be_named'
+        type:             'employee',
+        base_salary:      random_dollar_value(1000000),
+        appointment_type: '12M DURATION',
+        object_code_name: '::random::',
+        percent_effort:   random_percentage,
+        period_type:      '::random::'
+    }
+
+    set_options(defaults.merge(opts))
+    @percent_charged ||= (@percent_effort.to_f/2).round(2)
+  end
+
+  def create
+    # Navigation is handled by the BudgetVersionsObject method
+    if on(Personnel).job_code(@name).present?
+      # The person is already listed so do nothing
+    else
+      get_person
+    end
+    set_job_code
+    on Personnel do |page|
+      @salary_effective_date ||= page.salary_effective_date(@name).value
+      fill_out_item @name, page, :appointment_type, :base_salary, :salary_effective_date,
+                    :salary_anniversary_date
+      page.save
+      page.expand_all
+      page.person.select "#{@name} - #{@job_code}"
+      sleep 1 # this is required because the select list contents get updated when the person is selected.
+      page.object_code_name.pick! @object_code_name
+      page.add_details
+      page.expand_all
+      set_dates page
+      fill_out_item @name, page, :percent_effort, :percent_charged, :period_type
+      page.calculate @name
+      @requested_salary=page.requested_salary @name
+      @calculated_fringe=page.calculated_fringe @name
+      page.save
+    end
+  end
+
+  # ========
+  private
+  # ========
+
+  def set_job_code
+    unless @job_code.nil?
+      on(Personnel).job_code(@name).set @job_code
+    else
+      on(Personnel).lookup_job_code(@name)
+      on JobCodeLookup do |page|
+        page.search
+        page.return_random
+      end
+      @job_code=on(Personnel).job_code(@name).value
+    end
+  end
+
+  def get_person
+    on(Personnel).send("#{@type}_search")
+    if @name.nil?
+      on lookup_page do |page|
+        page.search
+        @name=page.returned_full_names.sample
+      end
+    end
+    on lookup_page do |page|
+      case(@type)
+        when 'employee'
+          page.select_person @name
+        when 'non_employee'
+          page.first_name.set @name[/^\S+/]
+          page.last_name.set @name[/\S+$/]
+          page.search
+          page.select_person @name[/\S+$/]
+        when 'to_be_named'
+          page.person_name.set @name
+          page.search
+          page.select_person @name
+      end
+      page.return_selected
+    end
+  end
+
+  # TODO: WOW! This desperately needs to be dryed up!
+  # It might be a good idea to make a new method in
+  # TestFactory for this.
+  def set_dates(page)
+    if @start_date.nil?
+      @start_date=page.start_date(@name).value
+    else
+      page.start_date(@name).set @start_date
+    end
+    if @end_date.nil?
+      @end_date=page.end_date(@name).value
+    else
+      page.end_date(@name).set @end_date
+    end
+  end
+
+  def lookup_page
+    Kernel.const_get({
+        employee:     'PersonLookup',
+        non_employee: 'NonOrgAddressBookLookup',
+        to_be_named:  'ToBeNamedPersonsLookup'
+    }[@type.to_sym])
+  end
+
+end
+
+class BudgetPersonnelCollection < Array
+
+
+
+end

data/lib/kuality-coeus/data_objects/budget/subaward_budget.rb

@@ -48,9 +48,7 @@ class SubawardBudgetObject
       add.add_file_name.set($file_folder+@file_name)
       add.add_subaward_budget
       add.show_details @organization_name
-      add.direct_cost(@organization_name).fit @direct_cost
-      add.f_and_a_cost(@organization_name).fit @f_and_a_cost
-      add.cost_sharing(@organization_name).fit @cost_sharing
+      fill_out_item @organization_name, add, :direct_cost, :f_and_a_cost, :cost_sharing
       @total_cost=add.total_cost(@organization_name)
       add.save
     end

data/lib/kuality-coeus/data_objects/identity/group.rb

@@ -0,0 +1,19 @@
+class GroupObject
+
+  include Foundry
+  include DataFactory
+  include Navigation
+
+  attr_accessor :id, :namespace, :name, :description, :type,
+                :principal_name, :assignees
+
+  def initialize(browser, opts={})
+    @browser = browser
+
+  end
+
+  def create
+
+  end
+
+end

data/lib/kuality-coeus/data_objects/navigation.rb

@@ -61,4 +61,27 @@ module Navigation
   end
   alias_method :fill_in, :fill_out
 
+  # Same as the above method, but used with methods that take a
+  # parameter to identify the target element...
+  def fill_out_item(name, page, *fields)
+    methods={
+        'Watir::TextField' => lambda{|n, p, f| p.send(f, n).fit(get f)},
+        'Watir::Select'    => lambda{|n, p, f| p.send(f, n).pick!(get f)},
+        'Watir::Radio'     => lambda{|n, p, f| p.send(f, n, get(f)) unless get(f)==nil },
+        'Watir::CheckBox'  => lambda{|n, p, f| p.send(f, n).fit(get f) }
+    }
+    fields.shuffle.each do |field|
+      # TODO: Someday see if there's a way to fix things so this rescue isn't necessary...
+      # It's here because the radio button "element" definitions are *actions* that
+      # require a parameter, so just sending the method to the page
+      # is not going to work.
+      begin
+        key = page.send(field, name).class.to_s
+      rescue NoMethodError
+        key = 'Watir::Radio'
+      end
+      methods[key].call(name, page, field)
+    end
+  end
+
 end

data/lib/kuality-coeus/data_objects/proposal_development/degrees.rb

@@ -25,8 +25,7 @@ class DegreeObject
       degrees.expand_all
       degrees.degree_type(@person).pick! @type
       degrees.degree_description(@person).fit @description
-      degrees.graduation_year(@person).fit @graduation_year
-      degrees.school(@person).fit @school
+      fill_out_item @person, degrees, :graduation_year, :school
       degrees.add_degree(@person)
     end
   end

data/lib/kuality-coeus/data_objects/proposal_development/key_personnel.rb

@@ -44,8 +44,8 @@ class KeyPersonObject
       end
       on KeyPersonnel do |person|
         @full_name=person.person_name
-        @first_name=@full_name[/^\w+/]
         @last_name=@full_name[/\w+$/]
+        @first_name=$~.pre_match.strip
       end
     else
       on PersonLookup do |look|
@@ -109,10 +109,7 @@ class KeyPersonObject
       # If it's a key person without units then they won't have credit splits,
       # otherwise, the person will, so fill them out...
       if @key_person_role==nil || !@units.empty?
-        person.space(@full_name).fit @space
-        person.responsibility(@full_name).fit @responsibility
-        person.financial(@full_name).fit @financial
-        person.recognition(@full_name).fit @recognition
+        fill_out_item @full_name, person, :space, :responsibility, :financial, :recognition
       end
 
       # Proposal Person Certification
@@ -124,7 +121,7 @@ class KeyPersonObject
       end
 
       # Add gathering/setting of more attributes here as needed
-      person.era_commons_name(@full_name).fit @era_commons_name
+      fill_out_item @full_name, person, :era_commons_name
       person.save
     end
   end

data/lib/kuality-coeus/data_objects/proposal_development/permissions.rb

@@ -26,23 +26,35 @@ class PermissionsObject
   # users to roles, but does not check who is already
   # assigned. You need to make sure that the values
   # used in the instantiation of the class are
-  # an accurate reflection of what exists in the site.
+  # an accurate reflection of what exists in the site,
+  # if that's important to the test.
   def assign
     navigate
     on Permissions do |add|
       # See the roles method defined below...
       roles.each do |inst_var, role|
         get(inst_var).each do |username|
-          unless add.user_row(username).present? && add.assigned_role(username).include?(role)
+          if add.assigned_users.include?(username)
+            unless add.assigned_role(username).include?(role)
+              add.edit_role username
+              on Roles do |roles|
+                roles.use_new_tab
+                roles.send(StringFactory.damballa(role)).set
+                roles.save
+                sleep 1.2 # Need to wait for the window to close
+                roles.use_new_tab
+              end
+              add.save
+            end
+          else
             add.user_name.set username
             add.role.select role
             add.add
             add.user_row(username).wait_until_present
+            add.save
           end
         end
       end
-      add.save
-      # TODO: Add some logic here to use in case the user is already added to the list (so use add_roles)
     end
   end
 

data/lib/kuality-coeus/data_objects/proposal_development/proposal_development.rb

@@ -73,6 +73,7 @@ class ProposalDevelopmentObject
     on Proposal do |edit|
       edit.proposal
       edit.expand_all
+      edit.project_title.fit opts[:project_title]
       edit.project_start_date.fit opts[:project_start_date]
       edit.opportunity_id.fit opts[:opportunity_id]
       edit.proposal_type.fit opts[:proposal_type]
@@ -185,15 +186,16 @@ class ProposalDevelopmentObject
     end
   end
 
-  def recall(type, reason=random_alphanums)
-    types={:revision=>:recall_to_action_list, :cancel=>:recall_and_cancel}
+  def recall(reason=random_alphanums)
     @recall_reason=reason
     open_proposal
     on(Proposal).recall
     on Confirmation do |conf|
       conf.reason.set @recall_reason
-      conf.send(types[type])
+      conf.yes
     end
+    open_proposal
+    @status=on(Proposal).document_status
   end
 
   def close
@@ -203,8 +205,8 @@ class ProposalDevelopmentObject
 
   def view(tab)
     open_proposal
-    unless @status=='CANCELED' || on(Proposal).send((tab.to_s+'_button').to_sym).parent.class_name=~/tabcurrent$/
-      on(Proposal).send(tab)
+    unless @status=='CANCELED' || on(Proposal).send(StringFactory.damballa("#{tab}_button")).parent.class_name=~/tabcurrent$/
+      on(Proposal).send(StringFactory.damballa(tab.to_s))
     end
   end
 

data/lib/kuality-coeus/data_objects/user.rb

@@ -93,31 +93,34 @@ class UserObject
   def initialize(browser, opts={})
     @browser = browser
     @user_name=case
-      when opts.empty?
-        'admin'
-      when opts.key?(:user)
-        opts[:user]
-      when opts.key?(:role)
-        USERS.have_role(ROLES[opts[:role]])[0][0]
-    end
+               when opts.empty?
+                 'admin'
+               when opts.key?(:user)
+                 opts[:user]
+               when opts.key?(:role)
+                 USERS.have_role(ROLES[opts[:role]])[0][0]
+               end
     defaults = USERS[@user_name]
-    defaults.nil? ? options=opts : options=defaults.merge(opts)
+    options = defaults.nil? ? opts : defaults.merge(opts)
     set_options options
   end
 
   def create
     # First we have to make sure we're logged in with
     # a user that has permissions to create other users...
-    if login_info_div.text =~ /admin$/
-      visit(SystemAdmin).person unless PersonLookup.new(@browser).principal_id.present?
-    else
-      @logged_in_user_name = login_info_div.text[/\w+$/]
-      visit Login do |log_in|
-        log_in.close_parents
-        log_in.username.set 'admin'
-        log_in.login
+    @browser.windows[0].use
+    visit SystemAdmin do |page|
+      page.close_children
+      if @logged_in_user_name=='admin'
+        page.person
+      else
+        s_o.click
+        visit Login do |log_in|
+          log_in.username.set 'admin'
+          log_in.login
+        end
+        visit(SystemAdmin).person
       end
-      visit(SystemAdmin).person
     end
     # Now we're certain the create button will be there, so...
     on(PersonLookup).create
@@ -208,6 +211,7 @@ class UserObject
         log_in.username.set @logged_in_user_name
         log_in.login
       end
+      @logged_in_user_name=nil
     end
   end
 
@@ -265,6 +269,8 @@ class UserObject
     visit SystemAdmin do |page|
       if username_field.present?
         UserObject.new(@browser).log_in
+      else
+        @logged_in_user_name=login_info_div.text[/\w+$/]
       end
       page.person
     end