app/models/jobshop/registration.rb in jobshop-0.0.4.2p4 vs app/models/jobshop/registration.rb in jobshop-0.0.5

@@ -1,20 +1,22 @@
 module Jobshop
   class Registration < VirtualRecord
-    memo_attr(:user) { site.users.build(user_params) }
-    memo_attr(:site) {
-      record = Jobshop::Site.find(params[:site_id])
-      record.assign_attributes(site_params)
+    memo_attr(:user) { team.users.build(user_params) }
+    memo_attr(:team) {
+      record = Jobshop::Team.find(params[:team_id])
+      record.assign_attributes(team_params)
       record
     }
 
     def save
       return false unless valid?
 
       result = ActiveRecord::Base.transaction do
         user.save!
-        site.update!(owner: user)
+        team.update!(owner: user,
+                     registration_token: nil,
+                     registration_token_sent_at: nil)
       end
 
       result != false
     end
 
@@ -22,12 +24,12 @@
 
     def registration_params
       params.fetch(:registration, ActionController::Parameters.new)
     end
 
-    def site_params
+    def team_params
       registration_params.fetch(
-        :site, ActionController::Parameters.new).permit(:name)
+        :team, ActionController::Parameters.new).permit(:name)
     end
 
     def user_params
       registration_params.fetch(:user, ActionController::Parameters.new).
         permit(:email, :password, :password_confirmation)