Added general user stamping on contact

app/controllers/concerns/shared/user_stamper.rb

@@ -0,0 +1,15 @@
+module Shared::UserStamper
+  extend ActiveSupport::Concern
+
+ def stamp obj
+    return false if obj.nil? || !obj.has_attribute?( :created_by_id && :updated_by_id )
+
+    if obj.new_record?
+      obj.created_by_id = current_epp_user.id
+    else
+      obj.updated_by_id = current_epp_user.id
+    end
+
+    return true
+  end
+end

app/controllers/epp/commands_controller.rb

@@ -2,6 +2,7 @@ class Epp::CommandsController < ApplicationController
   include Epp::Common
   include Epp::DomainsHelper
   include Epp::ContactsHelper
+  include Shared::UserStamper
 
   private
   def create

app/helpers/epp/contacts_helper.rb

@@ -21,6 +21,8 @@ module Epp::ContactsHelper
       zip: ph[:postalInfo][:pc]
     )
 
+    stamp @contact
+
     @contact.save
     render '/epp/contacts/create'
   end

db/migrate/20140804095654_add_user_stamps_to_contact.rb

@@ -0,0 +1,6 @@
+class AddUserStampsToContact < ActiveRecord::Migration
+  def change
+    add_column :contacts, :created_by_id, :integer
+    add_column :contacts, :updated_by_id, :integer
+  end
+end

db/schema.rb

@@ -11,7 +11,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema.define(version: 20140801140249) do
+ActiveRecord::Schema.define(version: 20140804095654) do
 
   # These are extensions that must be enabled in order to support this database
   enable_extension "plpgsql"
@@ -39,6 +39,8 @@ ActiveRecord::Schema.define(version: 20140801140249) do
     t.string   "ident"
     t.string   "ident_type"
     t.string   "org_name"
+    t.integer  "created_by_id"
+    t.integer  "updated_by_id"
   end
 
   create_table "countries", force: true do |t|

spec/epp/contact_spec.rb

@@ -12,6 +12,8 @@ describe 'EPP Contact', epp: true do
       expect(response[:result_code]).to eq('1000')
       expect(response[:msg]).to eq('Command completed successfully')
       expect(response[:clTRID]).to eq('ABC-12345')
+      expect(Contact.first.created_by_id).to be 1
+      expect(Contact.first.updated_by_id).to be nil
 
       expect(Contact.count).to eq(1)
       expect(Contact.first.org_name).to eq('Example Inc.')
@@ -27,6 +29,10 @@ describe 'EPP Contact', epp: true do
       expect(Contact.first.name).to eq("John Doe")
       expect(Contact.first.ident_type).to eq("op")
 
+      expect(Contact.first.updated_by_id).to be 1
+      #nil because we fabricate, hence stamping in controller won't happen
+      expect(Contact.first.created_by_id).to be nil
+
       expect(Contact.count).to eq(1)
     end
     #TODO tests for missing/invalid/etc ident