From 76cc852527a117e1cbff7f2ea80afa9033abb422 Mon Sep 17 00:00:00 2001 From: ankonym Date: Thu, 8 Oct 2015 15:37:27 +0200 Subject: Adjust the rake task to make id = invite code This will simplify invite code deletion after running tests. --- app/models/invite_code.rb | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'app/models') 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 -- cgit v1.2.3