Adjust the rake task to make id = invite code

This will simplify invite code deletion after running tests.

1 files changed, 3 insertions, 2 deletions

diff --git a/app/models/invite_code.rb b/app/models/invite_code.rb
index 30a6498..8f69582 100644
--- a/app/models/invite_code.rb
+++ b/app/models/invite_code.rb
@@ -1,4 +1,3 @@
-require 'coupon_code'
 
 class InviteCode < CouchRest::Model::Base
   use_database 'invite_codes'
@@ -14,9 +13,11 @@ class InviteCode < CouchRest::Model::Base
 
   def initialize(attributes = {}, options = {})
     super(attributes, options)
-    write_attribute('invite_code', CouponCode.generate) if new?
   end
 
+  def set_invite_code(code)
+    write_attribute(:invite_code, code)
+  end
 end