shushu 0.1.5

data/lib/helpers/http_helpers.rb

@@ -0,0 +1,46 @@
+module Shushu
+  module HttpHelpers
+    extend self
+
+    def build_q_params(hash)
+      "?" + hash.map do |k, v|
+        CGI.escape("#{k}=#{v}")
+      end.join("&")
+    end
+
+    def headers
+      {:content_type => :json, :accept => :json}
+    end
+
+    def enc_json(hash)
+      Yajl::Encoder.encode(hash)
+    end
+
+    def dec_json(json)
+      Yajl::Parser.parse(json)
+    end
+
+    def handle_req(&blk)
+      begin
+        resp = yield
+        dec_json(resp.body)
+      rescue RestClient::Exception => e
+        body = dec_json(e.http_body)
+        case e.http_code
+        when 404
+        when 401
+          raise(AuthorizationError, "Response: #{body.inspect}")
+        when 403
+        when 409
+        when 500
+          raise(UnexpectedError, "Response: #{body.inspect}")
+        when 503
+          raise(ServiceDownError, "#{Shushu.url} is down for maintenance.")
+        else
+          raise(UnexpectedError, "Response: #{body.inspect}")
+        end
+      end
+    end
+
+  end
+end

data/lib/models/account.rb

@@ -0,0 +1,13 @@
+module Shushu
+  module Account
+    extend self
+
+    def create
+      Shushu.handle_req {RestClient.post(accounts_url, {}, Shushu.headers)}
+    end
+
+    def accounts_url
+      [Shushu.url, "/accounts"].join
+    end
+  end
+end

data/lib/models/acct_own.rb

@@ -0,0 +1,31 @@
+module Shushu
+  module AcctOwn
+    extend self
+
+    def act(args)
+      pmid, eid = args.delete(:payment_method_id), args.delete(:entity_id)
+      Shushu.handle_req {RestClient.post(acct_own_url(pmid, eid), args, Shushu.headers)}
+    end
+
+    def xfr(args)
+      prev_pmid, prev_eid = args.delete(:prev_payment_method_id), args.delete(:prev_entity_id)
+      Shushu.handle_req {RestClient.put(acct_own_url(prev_pmid, prev_eid), args, Shushu.headers)}
+    end
+
+    def deact(args)
+      pmid = args.delete(:payment_method_id)
+      eid = args.delete(:entity_id)
+      account_id = args[:account_id]
+      time = args[:time]
+      Shushu.handle_req {RestClient.delete([acct_own_url(pmid, eid), CGI.encode("?account_id=#{account_id}&time=#{time}")].join)}
+    end
+
+    def query(args)
+      Shushu.handle_req {RestClient.get([Shushu.url, "/accounts/#{args[:account_id]}/resource_ownerships"].join)}
+    end
+
+    def acct_own_url(payment_method_id, entity_id)
+      [Shushu.url, "/payment_methods/#{payment_method_id}/account_ownerships/#{entity_id}"].join
+    end
+  end
+end

data/lib/models/b_event.rb

@@ -0,0 +1,31 @@
+module Shushu
+  module BEvent
+    extend self
+
+    OPEN = "open"
+    CLOSE = "close"
+
+    def all
+      Shushu.handle_req {RestClient.get([Shushu.url, "/billable_events"].join, Shushu.headers)}
+    end
+
+    def open(args)
+      hid = args.delete(:hid)
+      entity_id = args.delete(:entity_id)
+      args[:state] = OPEN
+      Shushu.handle_req {RestClient.put(events_url(hid, entity_id), args, Shushu.headers)}
+    end
+
+    def close(args)
+      hid = args.delete(:hid)
+      entity_id = args.delete(:entity_id)
+      args[:state] = CLOSE
+      Shushu.handle_req {RestClient.put(events_url(hid, entity_id), args, Shushu.headers)}
+    end
+
+    def events_url(hid, entity_id)
+      [Shushu.url, "/resources/#{hid}/billable_events/#{entity_id}"].join
+    end
+
+  end
+end

data/lib/models/billable_unit.rb

@@ -0,0 +1,35 @@
+module Shushu
+  class BillableUnit
+    attr_accessor(
+      :hid,
+      :from,
+      :to,
+      :qty,
+      :rate,
+      :rate_period,
+      :product_group,
+      :product_name
+    )
+
+    def initialize(args_hash)
+      @hid = args_hash["hid"]
+      @from = args_hash["from"]
+      @to = args_hash["to"]
+      @qty = args_hash["qty"].to_f
+      @rate = args_hash["rate"].to_i
+      @rate_period = args_hash["rate_period"]
+      @product_name = args_hash["product_name"]
+      @product_group = args_hash["product_group"]
+    end
+
+    def id
+      #TODO remove stub
+      Time.now.to_i
+    end
+
+    def total
+      @rate * @qty
+    end
+
+  end
+end

data/lib/models/heart_beat.rb

@@ -0,0 +1,10 @@
+module Shushu
+  module HeartBeat
+    extend self
+
+    def alive?
+      Shushu.handle_req {RestClient.get([Shushu.url, "/heartbeat"].join, Shushu.headers)}
+    end
+
+  end
+end

data/lib/models/line_item.rb

@@ -0,0 +1,19 @@
+module Shushu
+  class LineItem
+
+    attr_reader :unit_groups
+
+    def initialize(unit_groups)
+      @unit_groups = unit_groups
+    end
+
+    def app_name
+      @unit_groups.first.hid
+    end
+
+    def total
+      @unit_groups.map(&:total).reduce(:+)
+    end
+
+  end
+end

data/lib/models/r_code.rb

@@ -0,0 +1,14 @@
+module Shushu
+  module RCode
+    extend self
+
+    def create(args)
+      Shushu.handle_req {RestClient.post(rate_codes_url, args, Shushu.headers)}
+    end
+
+    def rate_codes_url
+      [Shushu.url, "/rate_codes"].join
+    end
+
+  end
+end

data/lib/models/report.rb

@@ -0,0 +1,51 @@
+module Shushu
+  class Report
+    attr_accessor(
+      :from,
+      :to,
+      :billable_units
+    )
+
+    def fetch
+      Shushu.handle_req do
+        RestClient.get([Shushu.url, resource].join, {:params => {:from => from.utc.to_s, :to => to.utc.to_s}})
+      end
+    end
+
+    def billable_units
+      @billable_units ||= report["billable_units"].map {|buh| BillableUnit.new(buh)}
+    end
+
+    def total
+      report["total"]
+    end
+
+    def report
+      @report ||= fetch
+    end
+  end
+
+  class UsageReport < Report
+    attr_accessor :account_id
+
+    def initialize(account_id, from, to)
+      @account_id, @from, @to = account_id, from, to
+    end
+
+    def resource
+      "/accounts/#{@account_id}/usage_reports"
+    end
+  end
+
+  class Invoice < Report
+    attr_accessor :payment_method_id
+
+    def initialize(payment_method_id, from, to)
+      @payment_method_id, @from, @to = payment_method_id, from, to
+    end
+
+    def resource
+      "/payment_methods/#{payment_method_id}/invoices"
+    end
+  end
+end

data/lib/models/res_own.rb

@@ -0,0 +1,31 @@
+module Shushu
+  module ResOwn
+    extend self
+
+    def act(args)
+      aid, eid = args.delete(:account_id), args.delete(:entity_id)
+      Shushu.handle_req {RestClient.post(res_own_url(aid, eid), args, Shushu.headers)}
+    end
+
+    def xfr(args)
+      prev_aid, prev_eid = args.delete(:prev_account_id), args.delete(:prev_entity_id)
+      Shushu.handle_req {RestClient.put(res_own_url(prev_aid, prev_eid), args, Shushu.headers)}
+    end
+
+    def deact(args)
+      aid = args.delete(:account_id)
+      eid = args.delete(:entity_id)
+      hid = args[:hid]
+      time = args[:time]
+      Shushu.handle_req {RestClient.delete([res_own_url(aid, eid), CGI.encode("?hid=#{hid}&time=#{time}")].join)}
+    end
+
+    def query(args)
+      Shushu.handle_req {RestClient.get([Shushu.url, "/accounts/#{args[:account_id]}/resource_ownerships"].join)}
+    end
+
+    def res_own_url(account_id, entity_id)
+      [Shushu.url, "/accounts/#{account_id}/resource_ownerships/#{entity_id}"].join
+    end
+  end
+end

data/lib/models/rev_rep.rb

@@ -0,0 +1,13 @@
+module Shushu
+  module RevRep
+    extend self
+
+    def get(from, to)
+      Shushu.handle_req do
+        RestClient.get([Shushu.url, "/rev_report"].join, {:params => {:from => from.utc.to_s, :to => to.utc.to_s}})
+      end
+    end
+
+  end
+end
+

data/lib/models/unit_group.rb

@@ -0,0 +1,59 @@
+module Shushu
+  class UnitGroup
+    InvalidUnitGroup = Class.new(ShushuException)
+
+    attr_reader :units
+
+    def initialize(units)
+      @units = units
+      check_rates
+    end
+
+    def check_rates
+      if @units.map(&:rate).uniq.length > 1
+        raise(InvalidUnitGroup, "Rates must be homogenius.")
+      end
+    end
+
+    def total
+      rate * qty
+    end
+
+    def qty
+      @units.map(&:qty).reduce(:+)
+    end
+
+    def rate
+      sample_unit.rate
+    end
+
+    def rate_period
+      sample_unit.rate_period
+    end
+
+    def name
+      sample_unit.product_name
+    end
+
+    def hid
+      sample_unit.hid
+    end
+
+    def product_name
+      sample_unit.product_name
+    end
+
+    def product_group
+      sample_unit.product_group
+    end
+
+    def description
+      sample_unit.product_name
+    end
+
+    def sample_unit
+      @sample_unit ||= @units.sample
+    end
+
+  end
+end

data/lib/presenters/base_presenter.rb

@@ -0,0 +1,25 @@
+module Shushu
+  class BasePresenter
+
+    def money(cents)
+      add_commas(pennies_to_dollar(cents.to_i))
+    end
+
+    def add_commas(str)
+      str.reverse.scan(/(?:\d*\.)?\d{1,3}-?/).join(',').reverse
+    end
+
+    def pennies_to_dollar(qty)
+      sprintf("%.2f", qty / 100.0)
+    end
+
+    def trunc_hours(hrs)
+      sprintf("%.3f", hrs)
+    end
+
+    def date_range(s, f)
+      s.strftime('%d %b %H:%M ') + " - " + f.strftime('%d %b %H:%M ')
+    end
+
+  end
+end

data/lib/presenters/line_item_presenter.rb

@@ -0,0 +1,49 @@
+module Shushu
+  class LineItemPresenter < BasePresenter
+
+    def initialize(line_item)
+      @line_item = line_item
+      @unit_groups = @line_item.unit_groups
+      @unit_group_presenters = @unit_groups.map {|ug| UnitGroupPresenter.new(ug)}
+    end
+
+    def total
+      money(@line_item.total)
+    end
+
+    def name
+      @line_item.app_name
+    end
+
+    def unit_group_rate(product_group)
+      unit_group_presenters(product_group).sample.rate
+    end
+
+    def unit_group_total(product_group)
+      money(
+        unit_group_presenters(product_group).
+        map(&:unit_group).
+        map(&:total).
+        reduce(:+)
+      )
+    end
+
+    def unit_group_qty(product_group)
+      trunc_hours(
+        unit_group_presenters(product_group).
+        map(&:unit_group).
+        map(&:qty).
+        reduce(:+)
+      )
+    end
+
+    def unit_group_presenters(product_group=nil)
+      if product_group
+        @unit_group_presenters.select {|ugp| ugp.product_group == product_group}
+      else
+        @unit_group_presenters
+      end
+    end
+
+  end
+end

data/lib/presenters/report_presenter.rb

@@ -0,0 +1,23 @@
+module Shushu
+  class ReportPresenter < BasePresenter
+
+    def initialize(report, line_item_builder=LineItemBuilder)
+      @report = report
+      @units = @report.billable_units
+      @builder = line_item_builder
+    end
+
+    def total
+      money(@report.total)
+    end
+
+    def line_item_presenters
+      @line_item_presenters ||= begin
+        @builder.build(@units).map do |li|
+          LineItemPresenter.new(li)
+        end
+      end
+    end
+
+  end
+end

data/lib/presenters/unit_group_presenter.rb

@@ -0,0 +1,63 @@
+module Shushu
+  class UnitGroupPresenter < BasePresenter
+
+    attr_reader :unit_group
+
+    def initialize(unit_group)
+      @unit_group = unit_group
+    end
+
+    def id
+      [:hid, :name].map {|m| unit_group.send(m)}.join("_").gsub("-", "_")
+    end
+
+    def unit_presenters
+      @unit_presenters ||= @unit_group.units.map {|u| UnitPresenter.new(u)}
+    end
+
+    def daily_report_url
+      #TODO remove chart stub
+      #payments_uri("/billable_units/daily")
+      "/billable_units/daily"
+    end
+
+    def compacted_reports
+      #TODO remove chart stub
+      #BillableUnitReporter.compacted(@unit_group.units)
+      []
+    end
+
+    def total
+      money(@unit_group.total)
+    end
+
+    def qty
+      trunc_hours(@unit_group.qty)
+    end
+
+    def rate
+      @unit_group.rate
+    end
+
+    def name
+      @unit_group.name
+    end
+
+    def description
+      @unit_group.description
+    end
+
+    def product_name
+      @unit_group.product_name
+    end
+
+    def product_group
+      @unit_group.product_group
+    end
+
+    def unit_of_measure
+      @unit_group.rate_period
+    end
+
+  end
+end

data/lib/presenters/unit_presenter.rb

@@ -0,0 +1,34 @@
+module Shushu
+  class UnitPresenter < BasePresenter
+
+    def initialize(unit)
+      @unit = unit
+    end
+
+    def total
+      money(@unit.total)
+    end
+
+    def rate
+      [money(@unit.rate), @unit.rate_period].join("/")
+    end
+
+    def qty
+      trunc_hours(@unit.qty)
+    end
+
+    def start
+      @unit.from
+    end
+
+    def end
+      @unit.to
+    end
+
+    def description
+      @unit.product_name
+    end
+
+  end
+end
+

data/lib/services/line_item_builder.rb

@@ -0,0 +1,17 @@
+module Shushu
+  module LineItemBuilder
+    extend self
+
+    # Give build() a collection of units and it will return a
+    # collection of line_items. Each line_item will have a collection
+    # of unit_groups wich will hold a collection of units
+
+    def build(units)
+      units.group_by(&:hid).map do |hid, units_by_hid|
+        units_by_hid.group_by(&:product_name).map do |product_name, units_by_name|
+          UnitGroup.new(units_by_name)
+        end
+      end.map {|unit_groups| LineItem.new(unit_groups)}.flatten
+    end
+  end
+end

data/lib/shushu.rb

@@ -0,0 +1,64 @@
+require "yajl"
+require "rest-client"
+require "helpers/http_helpers"
+
+require "active_support/inflector"
+Inflector = ActiveSupport::Inflector.send(:extend, ActiveSupport::Inflector)
+
+module Shushu
+  extend self
+  extend HttpHelpers
+
+  ShushuException     = Class.new(Exception)
+  AuthenticationError = Class.new(ShushuException)
+  AuthorizationError  = Class.new(ShushuException)
+  ServiceDownError    = Class.new(ShushuException)
+  UnexpectedError     = Class.new(ShushuException)
+  SymanticError       = Class.new(ShushuException)
+
+  def url
+    Client.url
+  end
+
+  def url=(url)
+    Client.url = url
+  end
+
+  class Client
+    def self.url=(url)
+      @@url = url
+    end
+
+    def self.url
+      @@url || ENV["SHUSHU_URL"]
+    end
+
+    def initialize(url)
+      self.class.url = url
+    end
+
+    def [](api)
+      Shushu.const_get(Inflector.camelize(api))
+    end
+  end
+end
+
+
+require "models/account"
+require "models/billable_unit"
+require "models/report"
+require "models/unit_group"
+require "models/line_item"
+require "models/b_event"
+require "models/res_own"
+require "models/r_code"
+require "models/rev_rep"
+require "models/heart_beat"
+
+require "services/line_item_builder"
+
+require "presenters/base_presenter"
+require "presenters/unit_group_presenter"
+require "presenters/unit_presenter"
+require "presenters/line_item_presenter"
+require "presenters/report_presenter"

data/readme.md

@@ -0,0 +1,252 @@
+# Shushu Client
+
+## Purpose
+
+This gem wraps the APIs defined [here](https://github.com/heroku/shushu/tree/master/doc).
+The shushu client also provides a set of objects that help with the presentation
+of an invoice.
+
+## Setup
+
+```bash
+$ gem install shushu
+```
+
+```bash
+# Optional. When set, shushu will ignore bad responses from Shushu's API.
+# Default: false
+$SHUSHU_CLIENT_UNSAFE=true
+
+# Required
+# Default: nil
+$SHUSHU_URL=https://shushu-stg.heroku.com
+```
+
+**Ruby configuration will take precedence over environment variables.**
+
+```ruby
+Shushu.url = "https://core:secret@shushu-stg.heroku.com"
+```
+
+## Usage
+
+### PaymentMethods
+
+This API deals primarily with credit cards. PaymentMethods can be created
+indipendintly of Accounts. You will need a payment_method to generate an
+invoice.
+
+```ruby
+  #TODO Build API
+```
+
+### Accounts
+
+This API deals with accounts which is a primitive for grouping resources. You
+will need an account to generate a usage report.
+
+```ruby
+  Shushu::Account.create
+  #=> {:id => "001"}
+```
+
+### AccountOwnerships
+
+Use this API when you want to setup associations between Vault accounts and
+Vault payment_methods.
+
+For complete details on the semantics of this API, read the [AccountOwnerships
+API docs.](https://github.com/heroku/shushu/blob/master/doc/account_ownership_api.md)
+
+```ruby
+  # To associate an account with a payment_method
+  Shushu::AcctOwn.act(
+    :account_id        => vault_account_id,
+    :payment_method_id => payment_method_id,
+    :entity_id         => entity_id,
+    :time              => time
+  )
+  #=> {"payment_method_id"=>"123", "account_id"=>"1", "entity_id"=>"entity123"}
+
+  # Now we need to change the payment_method on an account
+  Shushu::AcctOwn.xfr(
+    :prev_payment_method_id => prev_payment_method_id,
+    :payment_method_id      => new_payment_method_id,
+    :account_id             => account_id,
+    :prev_entity_id         => prev_entity_id,
+    :entity_id              => entity_id,
+    :time                   => time
+  )
+  #=> {"payment_method_id"=>"456", "account_id"=>"1", "entity_id"=>"event124"}
+```
+
+### RateCodes
+
+Before you can send BillableEvents to Shushu, you will need to provision a
+RateCode.
+
+* rate - int. units should be pennies.
+* product_group - string.
+* product_name - string. optional. you can also pass the product name in an event.
+
+[Shushu API Docs](https://github.com/heroku/shushu/blob/master/doc/rate_code_api.md)
+
+```ruby
+  Shushu::RCode.create(
+    :rate => 5,
+    :product_group => "addon",
+    :product_name => "postgres"
+  )
+  #=> {:slug => "AO01", :rate => 5, :product_group => "addon", :product_name => "postgres"}
+```
+
+### ResourceOwnerships
+
+Use this API when dealing with resources and Vault accounts. For instance:
+Heroku apps and Teams.
+
+For complete details on the semantics of this API, read the [ResourceOwnerships
+API docs.](https://github.com/heroku/shushu/blob/master/doc/resource_ownership_api.md)
+
+```ruby
+  # When a new app is created, activate a new resource_ownership record.
+  Shushu::ResOwn.act(
+    :hid              => hid,
+    :entity_id        => entity_id,
+    :account_id       => vault_account_id,
+    :time             => time
+  )
+  #=> {"hid"=>"123", "account_id"=>"1", "entity_id"=>"event123"}
+
+
+  # When an app is transfered to another vault account, transfer the resource_ownership record.
+  Shushu::ResOwn.xfr(
+    :hid                   => hid,
+    :prev_entity_id        => prev_entity_id,
+    :prev_vault_account_id => prev_vault_account_id,
+    :entity_id             => entity_id,
+    :account_id            => new_vault_account_id,
+    :time                  => time
+  )
+  #=> {"hid"=>"123", "account_id"=>"1", "entity_id"=>"event123"}
+
+  # When an app is destroyed, deactivate the resource_ownership record.
+  Shushu::ResOwn.deact(
+    :hid        => hid,
+    :entity_id  => entity_id,
+    :account_id => vault_account_id,
+    :time       => time
+  )
+  #=> {"hid"=>"123", "account_id"=>"1", "entity_id"=>"event123"}
+```
+
+### BillableEvents
+
+Use this API when you want to start billing for a resource. You can start
+emitting events prior to setting up relationships between accounts and
+payment_methods. (Although, usage reports and invoices will not be available
+until account_ownerships and resource_ownerships have been established.)
+
+For complete details on the semantics of this API, read the [BillableEvents
+API docs.](https://github.com/heroku/shushu/blob/master/doc/events_api.md)
+
+```ruby
+  # Open an event when you would like to start billing.
+  Shushu::BEvent.open(
+    :entity_id    => entity_id,
+    :hid          => hid,
+    :time         => time,
+    :rate_code    => rate_code,
+    :product_name => product_name,
+    :qty          => qty
+  )
+
+  # Don't forget to close it.
+  Shushu::BEvent.close(
+    :entity_id => entity_id,
+    :time      => time
+  )
+```
+
+### UsageReports
+
+```ruby
+  report = Shushu::UsageReport.new(account_id, from, to)
+  report.billable_units
+```
+
+
+### Invoices
+
+```ruby
+  invoice = Shushu::UsageReport.new(account_id, from, to)
+  invoice.billable_units
+```
+
+
+### Report Generation
+
+Invoices and UsageReports can be used for report generation. Basically, the
+report generation code expects a collection of BillableUnits. BillableUnits are
+returned from both the Invoice API and the UsageReport API. However, the details
+of the billable_units may be different with respect to the type of the report.
+
+#### Presenters
+
+Clients of this library will want to generate some sort of view for the reports, the
+presenter objects were created to aid with that effort. You should only need to
+use the presenters while building views. Each view wraps a simple model object.
+All of the models and presenters are derived from the billable_unit which is
+retreived from the remote API.
+
+Report --> LineItem --> UnitGroup --> Billable Unit
+
+
+#### ReportPresenter
+
+The report presenter is how you will kick off the process of generating a
+report. You hand it a report object, either a UsageReport or and Invoice, and
+using the reports billable_units, it will build the line_items and a set of
+line_item_presenters for the line_items.
+
+#### LineItemBuilder
+
+The ReportPresenter will create a set of line_items based upon the
+billable_units in the report. The default is to group things by HID. Thus there
+will be a line_item for each distinct HID in the set of billable_units. The
+builder will also give a collection of unit_groups to the line_item. By default,
+the unit_groups will be partitioned by the product_group. (i.e. dyno, addon)
+
+You can customize the LineItemBuilder by creating a new class that responds to
+build and passing it to the ReportPresenter.
+
+```ruby
+  ReportPresenter.new(report, CustomLineItemBuilder)
+```
+
+#### LineItemPresenter
+
+The LineItemPresenter is responsible for handling the total and names of the
+line_items. It also manages the set of unit_groups. Since the default
+LineItemBuilder partitioned unit_groups based upon product_group, you can ask
+the LineItemPresenter for infomation about subsets of unit_groups. For instance:
+
+```ruby
+  line_item_presenter.unit_group_presenters("dyno")
+  line_item_presenter.unit_group_total("dyno")
+  line_item_presenter.unit_group_qty("dyno")
+```
+
+
+#### UnitGroupPresenter
+
+UnitGroups are collections of billable_units that are partitioned by
+product_name. So if there exists a set of billable_units that all have
+product_group = "dyno" and product_name= "worker", then this presenter
+will give you information about that group.
+
+
+#### UnitPresenter
+
+Finally, the UnitPresenter wraps a billable_unit and exposes methods to show
+totals, quantities and other meta-data.

data/test/fixtures.rb

@@ -0,0 +1,29 @@
+module Fixtures
+  extend self
+
+  def billable_units
+    [
+      {
+        :hid => 123,
+        :from => Time.mktime(2000,1),
+        :to => Time.mktime(2000,2),
+        :qty => 744,
+        :rate => 5,
+        :rate_period => "hour",
+        :product_group => "dyno",
+        :product_name => "web"
+      },
+      {
+        :hid => 123,
+        :from => Time.mktime(2000,1),
+        :to => Time.mktime(2000,2),
+        :qty => 744,
+        :rate => 5,
+        :rate_period => "hour",
+        :product_group => "dyno",
+        :product_name => "worker"
+      }
+    ]
+  end
+
+end

data/test/models/b_event_test.rb

@@ -0,0 +1,20 @@
+require File.expand_path('../../test_helper', __FILE__)
+
+class BEventTest < ShushuTest
+  def test_open
+    our_params = {
+      :hid => "app123",
+      :entity_id => 1,
+      :rate_code => "RT01",
+      :time => Time.utc(2012,1).to_s,
+      :product_name => "web",
+      :qty => 1
+    }
+    api_resp_body = Shushu::HttpHelpers.enc_json(our_params)
+    Shushu.url = "http://provider:password@shushu.heroku.com"
+    FakeWeb.register_uri(:put, (Shushu.url + "/resources/app123/billable_events/1"), :body => api_resp_body)
+    event = Shushu::BEvent.open(our_params)
+    assert_equal(1, event["entity_id"])
+  end
+end
+

data/test/models/billable_unit_test.rb

@@ -0,0 +1,8 @@
+require File.expand_path('../../test_helper', __FILE__)
+
+class BillableUnitTest < ShushuTest
+  def test_total
+    unit1 = Shushu::BillableUnit.new("rate" => 10, "qty" => 10)
+    assert_equal(100, unit1.total)
+  end
+end

data/test/models/line_item_test.rb

@@ -0,0 +1,10 @@
+require File.expand_path('../../test_helper', __FILE__)
+
+class LineItemTest < ShushuTest
+  def test_total
+    unit1 = Shushu::BillableUnit.new("rate" => 10, "qty" => 10)
+    unit_group1 = Shushu::UnitGroup.new([unit1])
+    line_item = Shushu::LineItem.new([unit_group1])
+    assert_equal(100, line_item.total)
+  end
+end

data/test/models/r_code_test.rb

@@ -0,0 +1,18 @@
+require File.expand_path('../../test_helper', __FILE__)
+
+class RCodeTest < ShushuTest
+  def test_open
+    our_params = {
+      :rate          => 5,
+      :product_group => "addon",
+      :product_name  => "postgres",
+      :description   => "A database service."
+    }
+    api_resp_body = Shushu::HttpHelpers.enc_json(our_params.merge(:slug => "AO01"))
+    Shushu.url = "http://provider:password@shushu.heroku.com"
+    FakeWeb.register_uri(:post, (Shushu.url + "/rate_codes"), :body => api_resp_body)
+    rate_code = Shushu::RCode.create(our_params)
+    assert_equal("AO01", rate_code["slug"])
+  end
+end
+

data/test/models/rev_rep_test.rb

@@ -0,0 +1,12 @@
+require File.expand_path('../../test_helper', __FILE__)
+
+class RevRepTest < ShushuTest
+  def test_for_month
+    api_resp_body = Shushu::HttpHelpers.enc_json({:ok => true})
+    Shushu.url = "http://provider:password@shushu.heroku.com"
+    url = Shushu.url + "/rev_rep?from%3D2011-01-01+00%3A00%3A00+UTC&to%3D2011-02-01+00%3A00%3A00+UTC"
+    FakeWeb.register_uri(:get, url, :body => api_resp_body)
+    Shushu::RevRep.get(Time.utc(2011,1), Time.utc(2011,2))
+  end
+end
+

data/test/models/unit_group_test.rb

@@ -0,0 +1,17 @@
+require File.expand_path('../../test_helper', __FILE__)
+
+class UnitGroupTest < ShushuTest
+  def test_total
+    unit1 = Shushu::BillableUnit.new("rate" => 10, "qty" => 10)
+    unit_group1 = Shushu::UnitGroup.new([unit1])
+    assert_equal(100, unit_group1.total)
+  end
+
+  def test_check_rates
+    unit1 = Shushu::BillableUnit.new("rate" => 10, "qty" => 10)
+    unit2 = Shushu::BillableUnit.new("rate" => 9,  "qty" => 10)
+    unit_group1 = Shushu::UnitGroup.new([unit1])
+    unit_group2 = Shushu::UnitGroup.new([unit2])
+    assert_raises(Shushu::UnitGroup::InvalidUnitGroup) {Shushu::UnitGroup.new([unit1, unit2])}
+  end
+end

data/test/presenters/line_item_presenter_test.rb

@@ -0,0 +1,36 @@
+require File.expand_path('../../test_helper', __FILE__)
+
+class LineItemPresenterTest < ShushuTest
+  def test_unit_group_total
+    unit1 = Shushu::BillableUnit.new("rate" => 1000, "qty" => 10, "product_group" => "dyno")
+    unit_group1 = Shushu::UnitGroup.new([unit1])
+    line_item = Shushu::LineItem.new([unit_group1])
+    lip = Shushu::LineItemPresenter.new(line_item)
+    assert_equal("100.00", lip.unit_group_total("dyno"))
+  end
+
+  def test_unit_group_qty
+    unit1 = Shushu::BillableUnit.new("rate" => 1000, "qty" => 10, "product_group" => "dyno")
+    unit2 = Shushu::BillableUnit.new("rate" => 1000, "qty" => 1,  "product_group" => "dyno")
+    unit_group1 = Shushu::UnitGroup.new([unit1, unit2])
+    line_item = Shushu::LineItem.new([unit_group1])
+    lip = Shushu::LineItemPresenter.new(line_item)
+    assert_equal("11.000", lip.unit_group_qty("dyno"))
+  end
+
+  def test_unit_group_presenters
+    unit1 = Shushu::BillableUnit.new("product_group" => "dyno")
+    unit2 = Shushu::BillableUnit.new("product_group" => "dyno")
+    unit3 = Shushu::BillableUnit.new("product_group" => "addon")
+    unit_group1 = Shushu::UnitGroup.new([unit1, unit2])
+    unit_group2 = Shushu::UnitGroup.new([unit3])
+    line_item = Shushu::LineItem.new([unit_group1, unit_group2])
+    lip = Shushu::LineItemPresenter.new(line_item)
+
+    assert_equal(2, lip.unit_group_presenters.length)
+    assert_equal(1, lip.unit_group_presenters("dyno").map(&:product_group).uniq.length)
+    assert_equal("addon", lip.unit_group_presenters("addon").map(&:product_group).uniq.first)
+  end
+end
+
+

data/test/presenters/report_presenter_test.rb

@@ -0,0 +1,22 @@
+require File.expand_path('../../test_helper', __FILE__)
+
+class ReportPresenterTest < ShushuTest
+
+  def report
+    Shushu::Report.new.tap do |r|
+      r.billable_units = Fixtures.billable_units.map {|h| Shushu::BillableUnit.new(h)}
+    end
+  end
+
+  def test_init
+    presenter = Shushu::ReportPresenter.new(report)
+    assert(!presenter.nil?)
+  end
+
+  def test_returns_line_item_presenters
+    presenter = Shushu::ReportPresenter.new(report)
+    lip = presenter.line_item_presenters
+    assert_equal(Shushu::LineItemPresenter, lip.first.class)
+  end
+
+end

data/test/services/line_item_builder_test.rb

@@ -0,0 +1,30 @@
+require File.expand_path('../../test_helper', __FILE__)
+
+class LineItemBuilderTest < ShushuTest
+
+  def units
+    Fixtures.billable_units.map {|h| Shushu::BillableUnit.new(h)}
+  end
+
+  def test_build
+    line_items = Shushu::LineItemBuilder.build(units)
+    assert(!line_items.nil?)
+  end
+
+  def test_build_creates_line_item_by_hid
+    line_items = Shushu::LineItemBuilder.build(units)
+    assert_equal(1, line_items.length)
+  end
+
+  def test_build_creates_one_unit_group_for_line_item
+    line_item = Shushu::LineItemBuilder.build(units).pop
+    assert_equal(1, line_item.unit_groups.length)
+  end
+
+  def test_build_puts_both_units_on_unit_group
+    line_item = Shushu::LineItemBuilder.build(units).pop
+    unit_group = line_item.unit_groups.pop
+    assert_equal(2, unit_group.units.length)
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,19 @@
+$:.unshift("lib")
+$:.unshift("test")
+
+ENV["RACK_ENV"] = "test"
+
+require "rubygems"
+require "bundler"
+Bundler.require(:default, :test)
+
+require "minitest/autorun"
+require "shushu"
+require "fakeweb"
+require "fixtures"
+
+puts "Blocking network connectivity"
+FakeWeb.allow_net_connect = false
+
+class ShushuTest < MiniTest::Unit::TestCase
+end

metadata

@@ -0,0 +1,117 @@
+--- !ruby/object:Gem::Specification
+name: shushu
+version: !ruby/object:Gem::Version
+  version: 0.1.5
+  prerelease: 
+platform: ruby
+authors:
+- Ryan Smith
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-12-29 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rest-client
+  requirement: &26931360 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.6.7
+  type: :runtime
+  prerelease: false
+  version_requirements: *26931360
+- !ruby/object:Gem::Dependency
+  name: yajl-ruby
+  requirement: &26930880 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *26930880
+- !ruby/object:Gem::Dependency
+  name: active_support
+  requirement: &26930500 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *26930500
+description: A ruby wrapper around Shushu's HTTP API.
+email: ryan@heroku.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- readme.md
+- lib/presenters/base_presenter.rb
+- lib/presenters/unit_presenter.rb
+- lib/presenters/unit_group_presenter.rb
+- lib/presenters/report_presenter.rb
+- lib/presenters/line_item_presenter.rb
+- lib/shushu.rb
+- lib/models/account.rb
+- lib/models/acct_own.rb
+- lib/models/report.rb
+- lib/models/unit_group.rb
+- lib/models/rev_rep.rb
+- lib/models/line_item.rb
+- lib/models/heart_beat.rb
+- lib/models/r_code.rb
+- lib/models/res_own.rb
+- lib/models/b_event.rb
+- lib/models/billable_unit.rb
+- lib/helpers/http_helpers.rb
+- lib/services/line_item_builder.rb
+- test/presenters/line_item_presenter_test.rb
+- test/presenters/report_presenter_test.rb
+- test/fixtures.rb
+- test/models/r_code_test.rb
+- test/models/billable_unit_test.rb
+- test/models/b_event_test.rb
+- test/models/unit_group_test.rb
+- test/models/rev_rep_test.rb
+- test/models/line_item_test.rb
+- test/services/line_item_builder_test.rb
+- test/test_helper.rb
+homepage: http://github.com/heroku/shushu
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Shushu client lib
+test_files:
+- test/presenters/line_item_presenter_test.rb
+- test/presenters/report_presenter_test.rb
+- test/models/r_code_test.rb
+- test/models/billable_unit_test.rb
+- test/models/b_event_test.rb
+- test/models/unit_group_test.rb
+- test/models/rev_rep_test.rb
+- test/models/line_item_test.rb
+- test/services/line_item_builder_test.rb