omega-tariffs-base 0.1.5

data/lib/settings.rb

@@ -0,0 +1,42 @@
+require 'rubygems'
+gem 'activesupport', '=2.3.8'
+require 'active_support'
+
+module UniqSysOmega
+  module Tariffs
+
+    class Settings
+      def initialize(hash)
+        @proxy = Proxy.new(hash)
+      end
+
+      def method_missing(meth, *args)
+        @proxy.send(meth, *args)
+      end
+
+    protected
+      class Proxy
+        def initialize(hash)
+          raise ArgumentError, "Invalid argument for settings. Should be hash." unless hash.kind_of?(Hash)
+          @hash = HashWithIndifferentAccess.new(hash)
+        end
+  
+        def method_missing(meth, *args)
+          entry = @hash[meth]
+          return Proxy.new(entry) if entry.kind_of?(Hash)
+          entry
+        end
+
+        def to_hash
+          @hash.dup
+        end
+
+        def ==(other)
+          raise "Invalid comparison" unless other.kind_of?(Hash) || other.kind_of?(Settings::Proxy) || other.kind_of?(Settings)
+          @hash == HashWithIndifferentAccess.new(other.to_hash)
+        end    
+      end
+    end
+
+  end
+end

data/lib/single_time_filter.rb

@@ -0,0 +1,67 @@
+module UniqSysOmega
+  module Tariffs
+    
+    class SingleTimeFilterRepository
+      class <<self
+        def had_single_payment?(tarif_id, login_id, period_id)
+          return false unless @payments
+          @payments.member? [ tarif_id, login_id, period_id ]
+        end
+
+        def register_single_payment(tarif_id, login_id, period_id)
+          @payments ||= [ ]
+          @payments << [ tarif_id, login_id, period_id ] unless had_single_payment?(tarif_id, login_id, period_id)
+        end
+      end
+    end
+
+    class SingleTimeFilter < Filter
+      def start_permitted?(computer_id, login_id, options={})
+        single_payment_setup? login_id, options[:period].begin
+      end
+
+      def permitted?(computer_id, login_id, options={})
+        single_payment_setup? login_id, options[:period].begin
+      end
+
+      def calculate_cost(computer_id, login_id, started_at, amount, options={})
+        SingleTimeFilterRepository.had_single_payment?(tarif_id, login_id, options[:period].begin) ?
+                0 : @price
+      end
+
+      def process_activity(unprocessed_activity, ft, options={})
+        if SingleTimeFilterRepository.had_single_payment?(tarif_id, unprocessed_activity.login_id, options[:period].begin)
+          true
+        else
+          make_single_payment(ft, unprocessed_activity.login_id, options)
+        end
+      end
+
+    protected
+      def single_payment_setup?(login_id, period_id)
+        SingleTimeFilterRepository.had_single_payment?(tarif_id, login_id, period_id) ||
+                can_make_single_payment?(login_id)
+      end
+
+      def can_make_single_payment?(login_id)
+        Login.find(login_id).balance - @price >= 0
+      end
+
+      def make_single_payment(ft, login_id, options)
+        return false unless can_make_single_payment?(login_id)
+
+        # TODO: Make it transactional!
+        SingleTimeFilterRepository.register_single_payment(tarif_id, login_id, options[:period].begin)
+    
+        ft = ft.dup
+        ft.volume = @price
+        ft.save
+      end
+
+      def after_initialize
+        @price = settings.price
+      end
+    end
+    
+  end
+end

data/lib/tarif_builder.rb

@@ -0,0 +1,16 @@
+module UniqSysOmega
+  module Tariffs
+
+    class TarifBuilder
+      def initialize
+        raise "Cannot instantinate singleton class"
+      end
+  
+      def self.build(settings, tarif_id)
+        class_name = eval(settings.klass)
+        class_name.new(settings.settings, tarif_id)
+      end
+    end
+  
+  end
+end

data/lib/weekday_filter.rb

@@ -0,0 +1,53 @@
+module UniqSysOmega
+  module Tariffs
+
+    class WeekdayFilter < Filter
+      def start_permitted?(computer_id, login_id, options={})
+        @weekdays[Time.now.wday].start_permitted?(computer_id, login_id, options)
+      end
+
+      def permitted?(computer_id, login_id, options={})
+        @weekdays[Time.now.wday].permitted?(computer_id, login_id, options)
+      end
+
+      def calculate_cost(computer_id, login_id, started_at, amount, options={})
+        cost = 0
+        for_each_weekday(started_at, amount) do |period_start, period_length|
+          cost += @weekdays[period_start.wday].calculate_cost(computer_id, login_id, period_start, period_length, options)
+        end
+        cost
+      end
+
+      def process_activity(unprocessed_activity, ft, options={})
+        for_each_weekday(unprocessed_activity.created_at, unprocessed_activity.amount) do |period_start, period_length|
+          @weekdays[period_start.wday].process_activity(alter_activity_period(unprocessed_activity, period_start, period_length), ft, options) or return false
+        end
+        true
+      end
+
+    protected
+      def for_each_weekday(started_at, amount, *args)
+        day_threshold = started_at.beginning_of_day
+        while true
+          period = (day_threshold .. day_threshold + 1.day) & (started_at .. started_at+amount)
+          break if period.nil?
+          yield period.begin, period.end-period.begin, *args unless period.end-period.begin == 0
+          day_threshold += 1.day
+        end
+      end
+
+      def after_initialize
+        @weekdays = [nil]*7
+        settings.to_hash.each do |period, settings|
+          filter = TarifBuilder.build Settings.new(settings), tarif_id
+          period.to_a.each do |day|
+            raise "Day already defined. Wrong tarif settings." unless @weekdays[day].nil?
+            @weekdays[day] = filter
+          end
+        end
+        raise "Some days left undefined." unless @weekdays.compact.size == 7
+      end
+    end
+    
+  end
+end

data/omega-tariffs-base.gemspec

@@ -0,0 +1,36 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{omega-tariffs-base}
+  s.version = "0.1.5"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Uniq Systems"]
+  s.date = %q{2010-09-23}
+  s.description = %q{Omega Sector base tariffs}
+  s.email = %q{ivan@uniqsystems.ru}
+  s.extra_rdoc_files = ["CHANGELOG", "README.rdoc", "lib/computer_class_filter.rb", "lib/deny_filter.rb", "lib/filter.rb", "lib/omega-tariffs-base.rb", "lib/packet_filter.rb", "lib/rate_filter.rb", "lib/regioned_calculation.rb", "lib/repeating_filter.rb", "lib/settings.rb", "lib/single_time_filter.rb", "lib/tarif_builder.rb", "lib/weekday_filter.rb"]
+  s.files = ["CHANGELOG", "Manifest", "README.rdoc", "Rakefile", "lib/computer_class_filter.rb", "lib/deny_filter.rb", "lib/filter.rb", "lib/omega-tariffs-base.rb", "lib/packet_filter.rb", "lib/rate_filter.rb", "lib/regioned_calculation.rb", "lib/repeating_filter.rb", "lib/settings.rb", "lib/single_time_filter.rb", "lib/tarif_builder.rb", "lib/weekday_filter.rb", "spec/complex_tarif_spec.rb", "spec/computer_class_filter_spec.rb", "spec/deny_filter_spec.rb", "spec/filter_spec.rb", "spec/lib/active_record_classes_stub.rb", "spec/lib/init.rb", "spec/lib/spec_helper.rb", "spec/packet_filter_spec.rb", "spec/rate_filter_spec.rb", "spec/repeating_filter_spec.rb", "spec/settings_spec.rb", "spec/single_time_filter_spec.rb", "spec/tarif_builder_spec.rb", "spec/weekday_filter_spec.rb", "omega-tariffs-base.gemspec"]
+  s.homepage = %q{http://uniqsys.ru/}
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Omega-tariffs-base", "--main", "README.rdoc"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{omega-tariffs-base}
+  s.rubygems_version = %q{1.3.7}
+  s.summary = %q{Omega Sector base tariffs}
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<activesupport>, ["= 2.3.8"])
+      s.add_development_dependency(%q<activesupport>, ["= 2.3.8"])
+    else
+      s.add_dependency(%q<activesupport>, ["= 2.3.8"])
+      s.add_dependency(%q<activesupport>, ["= 2.3.8"])
+    end
+  else
+    s.add_dependency(%q<activesupport>, ["= 2.3.8"])
+    s.add_dependency(%q<activesupport>, ["= 2.3.8"])
+  end
+end

data/spec/complex_tarif_spec.rb

@@ -0,0 +1,54 @@
+require File.expand_path(File.join(File.dirname(__FILE__), 'lib', 'init'))
+require 'mathn'
+
+describe "Flat daily tarif" do
+  before do
+    @day_start = Time.utc(2010, 1, 1, 0, 0, 0)
+    @settings = Settings.new({
+      :klass => "RepeatingFilter",
+      :settings => {
+        { :starts_at => @day_start + 8.hours, :length => 2.hours.to_i, :period => 1.day.to_i } =>
+                { :klass => "RateFilter", :settings => { :rate => 95 } },
+
+        { :starts_at => @day_start + 10.hours, :length => 4.hours.to_i, :period => 1.day.to_i } =>
+                { :klass => "RateFilter", :settings => { :rate => 195 } },
+
+        { :starts_at => @day_start + 14.hours, :length => 18.hours.to_i, :period => 1.day.to_i } =>
+                { :klass => "RateFilter", :settings => { :rate => 395 } }
+      }
+    })
+
+    @filter = TarifBuilder.build(@settings, 0)
+  end
+
+  it "should calculate daily cost correctly" do
+    @filter.calculate_cost(1, 2, @day_start, 24*3600).should == 8080
+  end
+
+  it "should calculate sliding fragment cost correctly" do
+    day_start = Time.now.utc.beginning_of_day
+    fragment = (Time.now.utc.beginning_of_day ... Time.now.utc.beginning_of_day + 4123)
+
+    zones = {
+              (day_start ... day_start + 8.hours) => 395,
+              (day_start + 8.hours ... day_start + 10.hours) => 95,
+              (day_start + 10.hours ... day_start + 14.hours) => 195,
+              (day_start + 14.hours ... day_start + 24.hours) => 395,
+
+              (day_start + 24.hours ... day_start + 24.hours + 8.hours) => 395,
+              (day_start + 24.hours + 8.hours ... day_start + 24.hours + 10.hours) => 95,
+              (day_start + 24.hours + 10.hours ... day_start + 24.hours + 14.hours) => 195,
+              (day_start + 24.hours + 14.hours ... day_start + 24.hours + 24.hours) => 395
+            }
+
+    while fragment.begin.wday == Time.now.utc.wday
+      real_summ = zones.map { |zone, price|
+        intersection = (zone & fragment) || (0...0)
+        Rational((intersection.end.to_i - intersection.begin.to_i) * price, 3600)
+      }.inject(&:+)
+
+      (@filter.calculate_cost(1, 2, fragment.begin, (fragment.end-fragment.begin).to_i)*100).to_i.should == (real_summ*100).to_i
+      fragment = Range.new(fragment.begin + 117.seconds, fragment.end + 127.seconds, true)
+    end
+  end
+end

data/spec/computer_class_filter_spec.rb

@@ -0,0 +1,82 @@
+require File.expand_path(File.join(File.dirname(__FILE__), 'lib', 'init'))
+
+class ComputerClassFilterMockFilter1 < Filter
+  def calculate_cost(computer_id, login_id, started_at, amount, options={})
+    Rational(amount.to_i*100, 3600)
+  end
+
+  def process_activity(unprocessed_activity, ft, options={})
+    ft = ft.dup
+    ft.volume = Rational(unprocessed_activity.amount.to_i*100, 3600)
+    ft.save!
+    true
+  end
+end
+
+class ComputerClassFilterMockFilter2 < Filter
+  def calculate_cost(computer_id, login_id, started_at, amount, options={})
+    Rational(amount.to_i*200, 3600)
+  end
+
+  def process_activity(unprocessed_activity, ft, options={})
+    ft = ft.dup
+    ft.volume = Rational(unprocessed_activity.amount.to_i*200, 3600)
+    ft.save!
+    true
+  end
+end
+
+describe "ComputerClassFilter" do
+  before do
+    settings = {
+      :klass => "ComputerClassFilter",
+      :settings => {
+        1 => { :klass => "ComputerClassFilterMockFilter1", :settings => { :vasya => "pupkin" } },
+        2 => { :klass => "ComputerClassFilterMockFilter2", :settings => { :vasya => "pupkin" } }
+      }
+    }
+    @filter = TarifBuilder.build Settings.new(settings), 0
+  end
+
+  it "should fail to instantinate with incorrect config" do
+    settings = [ {
+      :klass => "ComputerClassFilter",
+      :settings => {
+        1 => { :klass => "ComputerClassFilterMockFilter1", :settings => { :vasya => "pupkin" } },
+      }
+    }, {
+      :klass => "ComputerClassFilter",
+      :settings => {
+        2 => { :klass => "ComputerClassFilterMockFilter1", :settings => { :vasya => "pupkin" } },
+      }
+    } ]
+
+    lambda {
+      TarifBuilder.build Settings.new(settings[0]), 0
+    }.should raise_error(Exception, /Some classes left undefined/)
+
+    lambda {
+      TarifBuilder.build Settings.new(settings[1]), 0
+    }.should raise_error(Exception, /Some classes left undefined/)
+  end
+
+  it "should direct calls to certain mock filters" do
+    { 1 => ComputerClassFilterMockFilter1, 2 => ComputerClassFilterMockFilter2 }.each do |class_id, klass|
+      klass.any_instance.expects(:start_permitted?).with(class_id, anything, anything)
+      @filter.start_permitted?(class_id, DOESNT_MATTER)
+
+      klass.any_instance.expects(:permitted?).with(class_id, anything, anything)
+      @filter.permitted?(class_id, DOESNT_MATTER)
+
+      klass.any_instance.expects(:calculate_cost).with(class_id, anything, anything, anything, anything)
+      @filter.calculate_cost(class_id, DOESNT_MATTER, DOESNT_MATTER, DOESNT_MATTER, DOESNT_MATTER)
+
+      activity_doesnt_matter = Activity.new
+      activity_doesnt_matter.stubs(:created_at).returns(Time.now + 30.seconds)
+      activity_doesnt_matter.stubs(:amount).returns(3600)
+      activity_doesnt_matter.stubs(:computer_id).returns(class_id)
+      klass.any_instance.expects(:process_activity).with(anything, anything, anything)
+      @filter.process_activity(activity_doesnt_matter, DOESNT_MATTER, DOESNT_MATTER)
+    end
+  end
+end

data/spec/deny_filter_spec.rb

@@ -0,0 +1,23 @@
+require File.expand_path(File.join(File.dirname(__FILE__), 'lib', 'init'))
+
+describe "DenyFilter" do
+  before do
+    @filter = DenyFilter.new(stub(), 0)
+  end
+
+  it "should deny any service start" do
+    @filter.start_permitted?(1,2,3).should == false
+  end
+
+  it "should deny any service conducting its work" do
+    @filter.permitted?(1,2,3).should == false
+  end
+
+  it "should return maximum price for any activity" do
+    @filter.calculate_cost(1, 2, 3, 4).should == Integer::MAX
+  end
+
+  it "should not deny activity processing" do
+    @filter.process_activity(stub(), stub()).should == true
+  end
+end

data/spec/filter_spec.rb

@@ -0,0 +1,36 @@
+require File.expand_path(File.join(File.dirname(__FILE__), 'lib', 'init'))
+
+describe "Filter" do
+  it "should quack like a Filter :)" do
+    f = Filter.new(stub(), 0)
+    f.should respond_to :start_permitted?
+    f.should respond_to :permitted?
+    f.should respond_to :calculate_cost
+    f.should respond_to :process_activity
+  end
+
+  it "should raise exceptions when called" do
+    f = Filter.new(stub(), 0)
+
+    lambda {
+      f.start_permitted?
+    }.should raise_error
+
+    lambda {
+      f.permitted?
+    }.should raise_error
+
+    lambda {
+      f.calculate_cost
+    }.should raise_error
+
+    lambda {
+      f.process_activity
+    }.should raise_error
+  end
+  
+  it "should call after_initialize" do
+    Filter.any_instance.expects(:after_initialize)
+    Filter.new(stub(), 0)
+  end
+end

data/spec/lib/active_record_classes_stub.rb

@@ -0,0 +1,151 @@
+require 'rubygems'
+gem 'activerecord', '=2.3.8' 
+require 'active_record'
+
+class MockBase < ActiveRecord::Base
+end
+
+MockBase.class_eval do
+  alias_method :save, :valid?
+
+  def new_save!
+    raise "Invalid record: #{errors.full_messages.join("\n")}" unless valid? 
+  end
+
+  alias_method :save!, :new_save!
+
+  def self.columns()
+    @columns ||= [];
+  end
+
+  def self.column(name, sql_type = nil, default = nil, null = true)
+    columns << ActiveRecord::ConnectionAdapters::Column.new(name.to_s, default, sql_type, null)
+  end
+
+  def self.template
+    instance = new
+    instance.instance_eval %Q{
+      def save
+        raise "Cannot save a template transaction"
+      end
+
+      def save!
+        raise "Cannot save a template transaction"
+      end
+    }
+
+    instance.expects(:save).never
+    instance.expects(:save!).never
+
+    instance
+  end
+end
+
+class FinTransaction < MockBase
+  column :club_id, :integer
+  column :activity_id, :integer
+  column :author_id, :integer
+  column :src_account_id, :integer
+  column :dst_account_id, :integer
+  column :created_at, :timestamp
+  column :volume, :decimal
+
+  validates_presence_of :club_id
+  validates_presence_of :author_id
+  validates_presence_of :volume
+  
+  class << self
+    attr_accessor :transactions
+    
+    def reset_transactions
+      self.transactions = [ ]
+    end
+
+    def charge_login(login_id, volume)
+      FinTransaction.new(:author_id => 1, :club_id => 2,
+                         :src_account_id => login_id, :dst_account_id => -1,
+                         :volume => volume)
+    end
+  end
+  
+  def save
+    return false unless valid?
+    do_save
+    true
+  end
+  
+  def save!
+    raise "Validation error" unless valid?
+    do_save
+  end
+  
+protected
+  def do_save
+    self.class.transactions ||= []
+    self.class.transactions << volume
+  end
+end
+
+class Activity < MockBase
+  column :created_at
+  column :login_id, :decimal
+  column :amount
+  column :computer_id
+end
+
+class Login
+  attr_accessor :id
+
+  def balance
+    return 0 unless self.class.balance
+    self.class.balance[id] || 0
+  end
+
+  class <<self
+    def find(id)
+      Login.new.tap do |login|
+        login.id = id
+      end
+    end
+
+    attr_accessor :balance    
+
+    def set_login_balance(id, balance)
+      self.balance ||= { }
+      self.balance[id] = balance
+    end
+  end
+end
+
+class Computer
+  attr_accessor :id
+  attr_accessor :computer_class_id
+
+  class <<self
+    def find(computer_id)
+      Computer.new.tap do |computer|
+        computer.id = computer_id
+        computer.computer_class_id = computer_id
+      end
+    end
+  end
+end
+
+class ComputerClass
+  attr_accessor :id
+
+  class <<self
+    def count
+      all.size
+    end
+
+    def all
+      [ 1, 2 ].map { |id| with_id(id) }
+    end
+
+  private
+    def with_id(id)
+      ComputerClass.new.tap { |me| me.id = id}
+    end
+  end
+end