Add period unit support to domain price method #2741

app/models/domain.rb

@@ -374,7 +374,12 @@ class Domain < ActiveRecord::Base
 
   def price(operation)
     zone = name.split('.').drop(1).join('.')
-    p = "#{self.period}year"
+
+    p = period / 365 if period_unit == 'd'
+    p = period / 12 if period_unit == 'm'
+    p = period if period_unit == 'y'
+
+    p = "#{p}year"
     Pricelist.price_for(zone, operation, p)
   end
 

spec/models/domain_spec.rb

@@ -182,6 +182,26 @@ describe Domain do
       @domain.force_delete_at.should be_nil
     end
 
+    it 'should know its price' do
+      Fabricate(:pricelist, {
+        category: 'ee',
+        operation_category: 'create',
+        duration: '1year',
+        price: 1.50,
+        valid_from: Time.zone.parse('2015-01-01'),
+        valid_to: nil
+      })
+
+      domain = Fabricate(:domain)
+      domain.price('create').should == 1.50
+
+      domain = Fabricate(:domain, period: 12, period_unit: 'm')
+      domain.price('create').should == 1.50
+
+      domain = Fabricate(:domain, period: 365, period_unit: 'd')
+      domain.price('create').should == 1.50
+    end
+
     context 'about registrant update confirm' do
       before :all do
         @domain.registrant_verification_token = 123