multitenancy 0.0.4

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,17 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in multitenancy.gemspec
+gemspec
+
+group :test do
+  gem 'test-unit', '2.4.8'
+  gem 'rack-test', :require => "rack/test"
+  gem 'mocha'
+  gem 'shoulda'
+  gem 'shoulda-matchers'
+  gem 'shoulda-context'
+  gem 'activerecord'
+  gem 'database_cleaner'
+  gem 'sqlite3'
+  gem 'rest-client'
+end

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 ganeshs
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,43 @@
+# Multitenancy
+
+Multitenancy gem nicely plugs in to activerecord to provide multitenant support within a single schema. It allows multitenancy at two levels, tenant and sub-tenant. For instance you can have SAAS application where the primary tenant could be an organization and sub-tenant will be users in that organization. 
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'multitenancy'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install multitenancy
+
+## Usage
+
+This gem expects the tenant and sub-tenant values passed in the request header. But you can enhace this to support other logic as well.
+
+You use it in your padrino/sinatra application, add the below lines to the config.ru
+    
+    Multitenancy.init(:tenant_header => 'X_COMPANY_ID', :sub_tenant_header => 'X_USER_ID')
+    Padrino.use Multitenancy::Filter
+    
+You can also outside of a filter or in a standalone application,
+
+    tenant = Multitenancy::Tenant.new('flipkart', 'ganeshs')
+    Multitenancy.with_tenant(tenant) do
+        # Your code here
+    end
+    
+Any active record query executed within the tenant block, will be tenant/sub-tenant scoped. New records will persist the tenant and sub-tenant ids, find queries will be scoped to teanant and sub-tenant ids. If the sub-tenant id is not specified, it will be de-scoped.
+    
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/lib/multitenancy/model_extensions.rb

@@ -0,0 +1,75 @@
+module Multitenancy
+  
+  class << self
+    attr_accessor :tenant_field, :sub_tenant_field
+  end
+  
+  module ModelExtensions
+    extend ActiveSupport::Concern
+    
+    module ClassMethods
+      
+      def acts_as_tenant(tenant_id, sub_tenant_id=nil)
+        raise "tenant_id can't be nil! [Multitenancy]" unless tenant_id
+        
+        def self.is_scoped_by_tenant?
+          true
+        end
+        
+        Multitenancy.tenant_field = tenant_id
+        Multitenancy.sub_tenant_field = sub_tenant_id
+        
+        # set the current_tenant on newly created objects
+        before_validation Proc.new {|m|
+          tenant = Multitenancy.current_tenant
+          return unless tenant && tenant.tenant_id
+          m.send "#{tenant_id}=".to_sym, tenant.tenant_id
+          m.send "#{sub_tenant_id}=".to_sym, tenant.sub_tenant_id
+        }, :on => :create
+        
+        # set the default_scope to scope to current tenant
+        default_scope lambda {
+          tenant = Multitenancy.current_tenant
+          if tenant && tenant.tenant_id
+            conditions = {}
+            conditions[tenant_id] = tenant.tenant_id
+            conditions[sub_tenant_id] = tenant.sub_tenant_id if sub_tenant_id && tenant.sub_tenant_id
+            where(conditions)
+          end
+        }
+        
+        # Rewrite accessors to make tenantimmutable
+        define_method "#{tenant_id}=" do |value|
+          if new_record?
+            write_attribute(tenant_id, value)
+          else
+            raise "#{tenant_id} is immutable! [Multitenancy]"
+          end
+        end
+        
+        # Rewrite accessors to make sub_tenant immutable
+        define_method "#{sub_tenant_id}=" do |value|
+          if new_record?
+            write_attribute(sub_tenant_id, value)
+          else
+            raise "#{sub_tenant_id} is immutable! [Multitenancy]"
+          end
+        end
+      end
+      
+      def validates_uniqueness_to_tenant(fields, args ={})
+        raise "[Multitenancy] validates_uniqueness_to_tenant: no current tenant" unless respond_to?(:is_scoped_by_tenant?)
+        tenant_id = lambda {Multitenancy.tenant_field.downcase}.call
+        sub_tenant_id = Multitenancy.sub_tenant_field ? lambda {Multitenancy.sub_tenant_field.downcase}.call : nil 
+        
+        if args[:scope].nil?
+          args[:scope] = [tenant_id]
+        else
+          args[:scope] << tenant_id
+        end
+        args[:scope] = sub_tenant_id if sub_tenant_id
+        validates_uniqueness_of(fields, args)
+      end
+    end
+  end
+end

data/lib/multitenancy/rack/filter.rb

@@ -0,0 +1,27 @@
+module Multitenancy
+  
+  class Filter
+    
+    def initialize(app)
+      @app = app
+    end
+    
+    def call(env)
+      fix_headers!(env)
+      tenant = Tenant.new env[Multitenancy.tenant_header], env[Multitenancy.sub_tenant_header]
+      Multitenancy.with_tenant tenant do
+        @app.call env
+      end
+    end
+    
+    private
+    # rack converts X_FOO to HTTP_X_FOO, so strip "HTTP_"
+    def fix_headers!(env)
+      env.keys.select { |k| k =~ /^HTTP_X_/ }.each do |k|
+        env[k.gsub("HTTP_", "")] = env[k]
+        env.delete(k)
+      end
+      env
+    end
+  end
+end

data/lib/multitenancy/rest_client/rest_client.rb

@@ -0,0 +1,30 @@
+module RestClient
+  class Request
+    
+    class << self
+      
+      def execute_with_tenant_headers(args, &block)
+        args[:headers] = append_tenant_headers(args[:headers]) if Multitenancy.current_tenant && Multitenancy.append_headers_to_rest_calls?
+        return execute_without_tenant_headers(args, &block)
+      end
+
+      alias_method :execute_without_tenant_headers, :execute      
+      alias_method :execute, :execute_with_tenant_headers
+      
+      private
+      def append_tenant_headers(headers)
+        headers ||= {}
+        if !headers[Multitenancy.tenant_header] && Multitenancy.current_tenant
+          headers[Multitenancy.tenant_header] = Multitenancy.current_tenant.tenant_id
+        end
+        
+        if !headers[Multitenancy.sub_tenant_header] && Multitenancy.current_tenant
+          headers[Multitenancy.sub_tenant_header] = Multitenancy.current_tenant.sub_tenant_id
+        end
+        
+        headers
+      end
+    end
+    
+  end
+end

data/lib/multitenancy/tenant.rb

@@ -0,0 +1,15 @@
+module Multitenancy
+  
+  class Tenant
+    attr_reader :tenant_id, :sub_tenant_id
+    
+    def initialize(tenant_id, sub_tenant_id=nil)
+      @tenant_id = tenant_id
+      @sub_tenant_id = sub_tenant_id
+    end
+    
+    def headers
+      {Multitenancy.tenant_header => tenant_id, Multitenancy.sub_tenant_header => sub_tenant_id}
+    end
+  end
+end

data/lib/multitenancy/version.rb

@@ -0,0 +1,3 @@
+module Multitenancy
+  VERSION = "0.0.4"
+end

data/lib/multitenancy.rb

@@ -0,0 +1,67 @@
+require "active_record"
+require "active_model"
+require "multitenancy/version"
+require "multitenancy/tenant"
+require "multitenancy/rack/filter"
+require "multitenancy/model_extensions"
+require "multitenancy/rest_client/rest_client.rb"
+
+module Multitenancy
+  
+  @@tenant_header = 'X_TENANT_ID'
+  @@sub_tenant_header = 'X_SUB_TENANT_ID'
+  @@append_headers_to_rest_calls = true
+  @@logger = (logger rescue nil) || Logger.new(STDOUT)
+
+  class << self
+    def init(config)
+      @@tenant_header = config[:tenant_header]
+      @@sub_tenant_header = config[:sub_tenant_header]
+      @@logger = config[:logger] if config[:logger]
+      @@append_headers_to_rest_calls = config[:append_headers_to_rest_calls] unless config[:append_headers_to_rest_calls].nil?
+    end
+    
+    def logger
+      @@logger
+    end
+  
+    def tenant_header
+      @@tenant_header
+    end
+    
+    def sub_tenant_header
+      @@sub_tenant_header
+    end
+    
+    def append_headers_to_rest_calls?
+      @@append_headers_to_rest_calls
+    end
+    
+    def with_tenant(tenant, &block)
+      self.logger.debug "Executing the block with the tenant - #{tenant}"
+      if block.nil?
+        raise ArgumentError, "block required"
+      end
+      old_tenant = self.current_tenant
+      self.current_tenant = tenant
+      begin
+        return block.call
+      ensure
+        self.current_tenant = old_tenant
+      end
+    end
+    
+    def current_tenant=(tenant)
+      self.logger.debug "Setting the current tenant to - #{tenant}"
+      Thread.current[:tenant] = tenant
+    end
+    
+    def current_tenant
+      Thread.current[:tenant]
+    end
+  end
+end
+
+if defined?(ActiveRecord::Base)
+  ActiveRecord::Base.send(:include, Multitenancy::ModelExtensions)
+end

data/multitenancy.gemspec

@@ -0,0 +1,17 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/multitenancy/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["ganeshs"]
+  gem.email         = ["ganeshs@flipkart.com"]
+  gem.description   = 'Support multitennacy with active record'
+  gem.summary       = 'Support multitennacy with active record at tenant and sub-tenant level'
+  gem.homepage      = "https://github.com/Flipkart/multitenancy"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "multitenancy"
+  gem.require_paths = ["lib"]
+  gem.version       = Multitenancy::VERSION
+end

data/test/database.yml

@@ -0,0 +1,3 @@
+sqlite:
+  adapter: sqlite3
+  database: test/multitenancy.sqlite3

data/test/test_config.rb

@@ -0,0 +1,31 @@
+require 'bundler/setup'
+require 'test/unit'
+require 'pp'
+require 'logger'
+require 'mocha'
+require 'shoulda'
+require 'active_record'
+require 'active_model'
+require 'restclient'
+require 'multitenancy'
+require 'database_cleaner'
+
+ROOT = File.expand_path("../..", __FILE__)
+$:.unshift(ROOT + "/lib")
+
+config = YAML::load(IO.read(File.join(File.dirname(__FILE__), 'database.yml')))
+ActiveRecord::Base.logger = Logger.new(File.join(File.dirname(__FILE__), "debug.log"))
+ActiveRecord::Base.establish_connection(config[ENV['DB'] || 'sqlite'])
+
+class ActiveSupport::TestCase
+  
+  setup do
+    DatabaseCleaner.strategy = :transaction
+    DatabaseCleaner.clean_with(:truncation)
+    DatabaseCleaner.start
+  end
+  
+  teardown do
+    DatabaseCleaner.clean
+  end
+end

data/test/unit/filter_test.rb

@@ -0,0 +1,20 @@
+require File.expand_path(File.dirname(__FILE__) + '/../test_config.rb')
+
+module Multitenancy
+  class FilterTest < Test::Unit::TestCase
+    
+    setup do
+      @env = {
+        'X_TENANT_ID' => 'tenant_id',
+        'X_SUB_TENANT_ID' => 'seller_id'
+      }
+      @app = mock
+      @app.stubs(:call).with(@env).returns([:status, :headers, :body])
+    end
+    
+    should "call app" do
+      filter = Multitenancy::Filter.new(@app)
+      assert_equal [:status, :headers, :body], filter.call(@env)
+    end
+  end
+end

data/test/unit/model_extensions_test.rb

@@ -0,0 +1,223 @@
+require File.expand_path(File.dirname(__FILE__) + '/../test_config.rb')
+
+# Setup the db
+ActiveRecord::Schema.define(:version => 1) do
+  create_table :model_with_tenant_ids, :force => true do |t|
+    t.column :org_id, :string 
+    t.column :name, :string
+  end
+  
+  create_table :model_with_sub_tenant_ids, :force => true do |t|
+    t.column :seller_id, :string 
+    t.column :name, :string
+  end 
+
+  create_table :model_with_tenant_id_and_sub_tenant_ids, :force => true do |t|
+    t.column :org_id, :string 
+    t.column :seller_id, :string 
+    t.column :name, :string
+  end
+  
+  create_table :model_without_tenant_id_and_sub_tenant_ids, :force => true do |t|
+    t.column :name, :string
+  end
+end
+
+class ModelWithTenantId < ActiveRecord::Base
+  acts_as_tenant :org_id
+end
+
+class ModelWithoutTenantIdAndSubTenantId < ActiveRecord::Base
+end
+
+class ModelWithTenantIdAndSubTenantId < ActiveRecord::Base
+  acts_as_tenant :org_id, :seller_id
+end
+
+class ModelWithSubTenantId < ActiveRecord::Base
+  # acts_as_tenant nil, :seller_id
+end
+
+class ModelExtensionsTest < ActiveSupport::TestCase
+    
+    context "create model with tenant id and sub tenant id" do
+      should "populate tenant and sub tenant id when set in context" do
+        Multitenancy.with_tenant Multitenancy::Tenant.new('Flipkart', 'Poorvika') do
+          model = ModelWithTenantIdAndSubTenantId.new({:name => "test name"})
+          model.save!
+          assert_equal 'Flipkart', model.org_id
+          assert_equal 'Poorvika', model.seller_id
+        end
+      end
+      
+      should "populate only tenant when sub tenant is not set in context" do
+        Multitenancy.with_tenant Multitenancy::Tenant.new('Flipkart') do
+          model = ModelWithTenantIdAndSubTenantId.new({:name => "test name"})
+          model.save!
+          assert_equal 'Flipkart', model.org_id
+          assert_nil model.seller_id
+        end
+      end
+      
+      should "not populate tenant and sub tenant when tenant id is not set in context" do
+        Multitenancy.with_tenant Multitenancy::Tenant.new(nil, 'Poorvika') do
+          model = ModelWithTenantIdAndSubTenantId.new({:name => "test name"})
+          model.save!
+          assert_nil model.org_id
+          assert_nil model.seller_id
+        end
+      end
+      
+      should "not populate tenant and sub tenant when context is not set" do
+        model = ModelWithTenantIdAndSubTenantId.new({:name => "test name"})
+        model.save!
+        assert_nil model.org_id
+        assert_nil model.seller_id
+      end
+    end
+    
+    context "create model without tenant id and sub tenant id" do
+      should "not populate tenant and sub tenant id when set in context" do
+        Multitenancy.with_tenant Multitenancy::Tenant.new('Flipkart', 'Poorvika') do
+          model = ModelWithoutTenantIdAndSubTenantId.new({:name => "test name"})
+          model.save!
+          assert_raises NoMethodError do 
+            model.org_id
+          end
+          assert_raises NoMethodError do 
+            model.seller_id
+          end
+        end
+      end
+      
+      should "not populate tenant and sub tenant when context is not set" do
+        model = ModelWithoutTenantIdAndSubTenantId.new({:name => "test name"})
+        model.save!
+        assert_raises NoMethodError do 
+          model.org_id
+        end
+        assert_raises NoMethodError do 
+          model.seller_id
+        end
+      end
+    end
+    
+    context "create model only with tenant id" do
+      should "populate tenant when set in context" do
+        Multitenancy.with_tenant Multitenancy::Tenant.new('Flipkart') do
+          model = ModelWithTenantId.new({:name => "test name"})
+          model.save!
+          assert_equal 'Flipkart', model.org_id
+        end
+      end
+      
+      should "not populate tenant when tenant id is not set in context" do
+        Multitenancy.with_tenant Multitenancy::Tenant.new(nil) do
+          model = ModelWithTenantId.new({:name => "test name"})
+          model.save!
+          assert_nil model.org_id
+        end
+      end
+      
+      should "not populate tenant when context is not set" do
+        model = ModelWithTenantId.new({:name => "test name"})
+        model.save!
+        assert_nil model.org_id
+      end
+    end
+    
+    context "create model only with sub tenant id" do
+      should "raise error when tenant id is nil" do
+        model = ModelWithSubTenantId.new
+        assert_raises RuntimeError do 
+          ModelWithSubTenantId.acts_as_tenant nil
+        end
+      end
+    end
+    
+    context "find by scope for model with tenant and sub tenant" do
+      setup do
+        Multitenancy.with_tenant Multitenancy::Tenant.new('Flipkart', 'Poorvika') do
+          model = ModelWithTenantIdAndSubTenantId.new({:name => "test name"})
+          model.save!
+        end
+        
+        Multitenancy.with_tenant Multitenancy::Tenant.new('ebay', 'Poorvika') do
+          model = ModelWithTenantIdAndSubTenantId.new({:name => "test name"})
+          model.save!
+        end
+        
+        Multitenancy.with_tenant Multitenancy::Tenant.new('ebay', 'Univercell') do
+          model = ModelWithTenantIdAndSubTenantId.new({:name => "test name"})
+          model.save!
+        end
+        
+        Multitenancy.with_tenant Multitenancy::Tenant.new('Flipkart', nil) do
+          model = ModelWithTenantIdAndSubTenantId.new({:name => "test name"})
+          model.save!
+        end
+        
+        model = ModelWithTenantIdAndSubTenantId.new({:name => "test name1"})
+        model.save!
+      end
+      
+      should "return by tenant id and sub tenant id when set in context" do
+        Multitenancy.with_tenant Multitenancy::Tenant.new('Flipkart', 'Poorvika') do
+          assert_equal 1, ModelWithTenantIdAndSubTenantId.all.count
+          assert_equal 'Flipkart', ModelWithTenantIdAndSubTenantId.find_by_name('test name').org_id  
+        end
+        
+        Multitenancy.with_tenant Multitenancy::Tenant.new('ebay', 'Univercell') do
+          assert_equal 1, ModelWithTenantIdAndSubTenantId.all.count
+        end
+      end
+      
+      should "return by tenant id when sub tenant is not set in context" do
+        Multitenancy.with_tenant Multitenancy::Tenant.new('Flipkart') do
+          assert_equal 2, ModelWithTenantIdAndSubTenantId.all.count 
+        end
+        
+        Multitenancy.with_tenant Multitenancy::Tenant.new('ebay') do
+          assert_equal 2, ModelWithTenantIdAndSubTenantId.all.count
+        end
+      end
+      
+      should "return all when tenant id is not set in context" do
+        Multitenancy.with_tenant Multitenancy::Tenant.new(nil, 'Poorvika') do
+          assert_equal 5, ModelWithTenantIdAndSubTenantId.all.count 
+          assert_equal 4, ModelWithTenantIdAndSubTenantId.where(:name => 'test name').count
+        end
+      end
+    end
+    
+    context "find by scope for model with tenant" do
+      setup do
+        Multitenancy.with_tenant Multitenancy::Tenant.new('Flipkart') do
+          model = ModelWithTenantIdAndSubTenantId.new({:name => "test name"})
+          model.save!
+        end
+        
+        Multitenancy.with_tenant Multitenancy::Tenant.new('ebay') do
+          model = ModelWithTenantIdAndSubTenantId.new({:name => "test name1"})
+          model.save!
+        end
+      end
+      
+      should "return by tenant id when set in context" do
+        Multitenancy.with_tenant Multitenancy::Tenant.new('Flipkart') do
+          assert_equal 1, ModelWithTenantIdAndSubTenantId.all.count 
+        end
+        
+        Multitenancy.with_tenant Multitenancy::Tenant.new('ebay') do
+          assert_equal 1, ModelWithTenantIdAndSubTenantId.all.count
+        end
+      end
+      
+      should "return all when tenant id is not set in context" do
+        Multitenancy.with_tenant Multitenancy::Tenant.new(nil) do
+          assert_equal 2, ModelWithTenantIdAndSubTenantId.all.count 
+          assert_equal 1, ModelWithTenantIdAndSubTenantId.where(:name => 'test name').count
+        end
+      end
+    end
+end

data/test/unit/multitenancy_test.rb

@@ -0,0 +1,31 @@
+require File.expand_path(File.dirname(__FILE__) + '/../test_config.rb')
+
+module Multitenancy
+  class MultiTenancyTest < Test::Unit::TestCase
+    
+    should "configure headers" do
+      Multitenancy.init(:tenant_header => 'tenant_id', :sub_tenant_header => 'seller_id')
+      assert_equal 'tenant_id', Multitenancy.tenant_header
+    end
+    
+    should "return current tenant" do
+      tenant = Tenant.new('tenant_id', 'seller_id')
+      Multitenancy.current_tenant = tenant
+      assert_equal tenant, Multitenancy.current_tenant
+    end
+    
+    context "block within context" do
+      setup do
+        tenant = Tenant.new('tenant_id', 'seller_id')
+        Multitenancy.current_tenant = tenant
+      end
+      
+      should "call block with right tenant and subtenant ids" do
+        Multitenancy.with_tenant(Tenant.new('dummy_tenant_id', 'dummy_seller_id')) do 
+          assert_equal Multitenancy.current_tenant.tenant_id, 'dummy_tenant_id'
+          assert_equal Multitenancy.current_tenant.sub_tenant_id, 'dummy_seller_id'
+        end
+      end
+    end
+  end
+end

data/test/unit/rest_client_test.rb

@@ -0,0 +1,37 @@
+require File.expand_path(File.dirname(__FILE__) + '/../test_config.rb')
+
+class RestClientTest < Test::Unit::TestCase
+
+  setup do
+    Multitenancy.init({:tenant_header => 'X_COMPANY_ID', :sub_tenant_header => 'X_USER_ID'})
+  end
+  
+  should "append tenant headers to rest calls within a tenant scope" do
+    RestClient::Request.any_instance.expects(:make_headers).with() {|headers| headers[Multitenancy.tenant_header] == 'Flipkart' && headers[Multitenancy.sub_tenant_header] == 'ganeshs'}
+    RestClient::Request.any_instance.expects(:execute)
+    Multitenancy.with_tenant(Multitenancy::Tenant.new('Flipkart', 'ganeshs')) do
+      RestClient.get('http://some/resource')
+    end
+  end
+  
+  should "not append tenant headers to rest calls outside of a tenant scope" do
+    RestClient::Request.any_instance.expects(:make_headers).with() {|headers| headers[Multitenancy.tenant_header].nil? && headers[Multitenancy.sub_tenant_header].nil? }
+    RestClient::Request.any_instance.expects(:execute)
+    RestClient.get('http://some/resource')
+  end
+  
+  context "no tenant headers in rest calls" do
+    setup do
+      Multitenancy.init({:tenant_header => 'X_COMPANY_ID', :sub_tenant_header => 'X_USER_ID', :append_headers_to_rest_calls => false})
+    end
+    
+    should "not append tenant headers to rest calls within tenant scope" do
+      RestClient::Request.any_instance.expects(:make_headers).with() {|headers| headers[Multitenancy.tenant_header].nil? && headers[Multitenancy.sub_tenant_header].nil? }
+      RestClient::Request.any_instance.expects(:execute)
+      Multitenancy.with_tenant(Multitenancy::Tenant.new('Flipkart', 'ganeshs')) do
+        RestClient.get('http://some/resource')
+      end
+    end
+  end
+    
+end

data/test/unit/tenant_test.rb

@@ -0,0 +1,27 @@
+require File.expand_path(File.dirname(__FILE__) + '/../test_config.rb')
+
+module Multitenancy
+  class TenantTest < Test::Unit::TestCase
+  
+    should "create tenant" do
+      tenant = Tenant.new('tenant_id', 'seller_id')
+      assert_equal tenant.tenant_id, 'tenant_id'
+      assert_equal tenant.sub_tenant_id, 'seller_id'
+    end
+    
+    should "get headers" do
+      tenant = Tenant.new('tenant_id', 'seller_id')
+      assert_equal tenant.headers, {'X_TENANT_ID' => 'tenant_id', 'X_SUB_TENANT_ID' => 'seller_id'}
+    end
+    
+    should "not allow setting tenant and sub tenant id" do
+      tenant = Tenant.new('tenant_id', 'seller_id')
+      assert_raises NoMethodError do
+        tenant.tenant_id = 'test'
+      end
+      assert_raises NoMethodError do
+        tenant.sub_tenant_id = 'test'
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,74 @@
+--- !ruby/object:Gem::Specification
+name: multitenancy
+version: !ruby/object:Gem::Version
+  version: 0.0.4
+  prerelease: 
+platform: ruby
+authors:
+- ganeshs
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-12-11 00:00:00.000000000Z
+dependencies: []
+description: Support multitennacy with active record
+email:
+- ganeshs@flipkart.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- lib/multitenancy.rb
+- lib/multitenancy/model_extensions.rb
+- lib/multitenancy/rack/filter.rb
+- lib/multitenancy/rest_client/rest_client.rb
+- lib/multitenancy/tenant.rb
+- lib/multitenancy/version.rb
+- multitenancy.gemspec
+- test/database.yml
+- test/multitenancy.sqlite3
+- test/test_config.rb
+- test/unit/filter_test.rb
+- test/unit/model_extensions_test.rb
+- test/unit/multitenancy_test.rb
+- test/unit/rest_client_test.rb
+- test/unit/tenant_test.rb
+homepage: https://github.com/Flipkart/multitenancy
+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: Support multitennacy with active record at tenant and sub-tenant level
+test_files:
+- test/database.yml
+- test/multitenancy.sqlite3
+- test/test_config.rb
+- test/unit/filter_test.rb
+- test/unit/model_extensions_test.rb
+- test/unit/multitenancy_test.rb
+- test/unit/rest_client_test.rb
+- test/unit/tenant_test.rb
+has_rdoc: