aptible-auth 0.5.4 → 0.5.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4cfd8c31974a234d4545587c92c9fe21bf9f1f98
-  data.tar.gz: 5b2fe6153c8124af9b8b65eb2d13f9aa69b66156
+  metadata.gz: 19f89d52a24659f1989ace1a8a504f0f5afe20ed
+  data.tar.gz: 07edd32d5ace4f65005eaabfe79b3fdba456d658
 SHA512:
-  metadata.gz: cecda29c55a88eae8fc329e811ba6eca4c7d625e4dafb3bbb847532acfb7525944496d3582edadb8c722aca4c5b06de231848aa797d0cc149341166d0e4db722
-  data.tar.gz: b70a7aafca2524677db22776879b449dc6c739a644794eb213843507f1689810cb6f87ecd889f40bb935e51af8379033661a8ce89f066fe37579cf83b6e92b36
+  metadata.gz: 553557d661afe5da3f45ffed18752d15a397141e2c608a4634139c5e9a07db52e546fcda8561311f1a0d683c08c7bd3a759c5ae349cb51687fabfea9527251b5
+  data.tar.gz: bf0dc5e0292f27c43ce5483145a24dd9d739d0ace1b24b9813f5a7478cbd2d2418adacfc7bb2b3a97f4fbe1634084a8ce9f15ffbdc20f782034c7919dcbc3369

data/lib/aptible/auth/user.rb

@@ -12,6 +12,23 @@ module Aptible
       field :created_at, type: Time
       field :updated_at, type: Time
 
+      # rubocop:disable MethodLength
+      def set_organization_roles(organization, roles)
+        self.roles.each do |role|
+          next unless role.organization.id == organization.id
+          next if roles.map(&:id).include? role.id
+
+          role_membership = role.memberships.find do |membership|
+            membership.user.id == id
+          end
+
+          role_membership.destroy
+        end
+
+        add_to_roles(roles)
+      end
+      # rubocop:enable MethodLength
+
       def organizations
         roles.map(&:organization).uniq(&:id)
       end
@@ -34,6 +51,18 @@ module Aptible
         roles.select { |user_role| role.id == user_role.id }.count > 0
       end
       # rubocop:enable PredicateName
+
+      def can_manage?(organization)
+        privileged_organizations.map(&:id).include? organization.id
+      end
+
+      def add_to_roles(roles)
+        roles.each { |role| add_to_role(role) }
+      end
+
+      def add_to_role(role)
+        role.create_membership(user: self, token: token) unless has_role? role
+      end
     end
   end
 end

data/lib/aptible/auth/version.rb

@@ -1,5 +1,5 @@
 module Aptible
   module Auth
-    VERSION = '0.5.4'
+    VERSION = '0.5.5'
   end
 end

data/spec/aptible/auth/user_spec.rb

@@ -1,6 +1,35 @@
 require 'spec_helper'
 
 describe Aptible::Auth::User do
+  describe '#can_manage?' do
+    let(:developer) { double 'Aptible::Auth::Role' }
+    let(:owner) { double 'Aptible::Auth::Role' }
+    let(:org) { double 'Aptible::Auth::Organization' }
+
+    before do
+      org.stub(:id) { 1 }
+      developer.stub(:organization) { org }
+      allow(developer).to receive(:privileged?).and_return(false)
+      owner.stub(:organization) { org }
+      allow(owner).to receive(:privileged?).and_return(true)
+    end
+
+    it 'should return false if not member of org privileged role' do
+      subject.stub(:roles) { [developer] }
+      expect(subject.can_manage?(org)).to eq false
+    end
+
+    it 'should return true if member of org privileged role' do
+      subject.stub(:roles) { [developer, owner] }
+      expect(subject.can_manage?(org)).to eq true
+    end
+
+    it 'should return false if member of no roles' do
+      subject.stub(:roles) { [] }
+      expect(subject.can_manage?(org)).to eq false
+    end
+  end
+
   describe '#organizations' do
     let(:so) { double 'Aptible::Auth::Role' }
     let(:owner) { double 'Aptible::Auth::Role' }
@@ -39,4 +68,75 @@ describe Aptible::Auth::User do
       expect(subject.has_role?(owner)).to eq false
     end
   end
+
+  describe '#set_organization_roles' do
+
+    let(:so) { double 'Aptible::Auth::Role' }
+    let(:owner) { double 'Aptible::Auth::Role' }
+    let(:org) { double 'Aptible::Auth::Organization' }
+    let(:owner_membership) { double 'Aptible::Auth::Membership' }
+    let(:so_membership) { double 'Aptible::Auth::Membership' }
+
+    before do
+      org.stub(:id) { 1 }
+
+      so.stub(:organization) { org }
+      so.stub(:id) { 1 }
+
+      owner.stub(:organization) { org }
+      owner.stub(:id) { 2 }
+
+      allow(Aptible::Auth::Role).to receive(:find)
+        .with(1, token: 'token').and_return(so)
+      allow(Aptible::Auth::Role).to receive(:find)
+        .with(2, token: 'token').and_return(owner)
+    end
+
+    it 'should overwrite existing memberships' do
+      subject.stub(:roles) { [so] }
+      subject.stub(:token) { 'token' }
+      subject.stub(:headers) { {} }
+      so_membership.stub(:user) { subject }
+      so_membership.stub(:role) { so }
+      so.stub(:memberships) { [so_membership] }
+      owner.stub(:memberships) { [] }
+
+      expect(so_membership).to receive(:destroy)
+      expect(owner).to receive(:create_membership)
+        .with(user: subject, token: 'token')
+
+      subject.set_organization_roles(org, [owner])
+    end
+
+    it 'should create new memberships' do
+      subject.stub(:roles) { [] }
+      subject.stub(:token) { 'token' }
+      subject.stub(:headers) { {} }
+      so.stub(:memberships) { [] }
+      owner.stub(:memberships) { [] }
+
+      expect(so).to receive(:create_membership)
+        .with(user: subject, token: 'token')
+      expect(owner).to receive(:create_membership)
+        .with(user: subject, token: 'token')
+
+      subject.set_organization_roles(org, [so, owner])
+    end
+
+    it 'should delete all existing memberships' do
+      subject.stub(:roles) { [so, owner] }
+      so.stub(:memberships) { [so_membership] }
+      owner.stub(:memberships) { [owner_membership] }
+      so_membership.stub(:user) { subject }
+      so_membership.stub(:role) { so }
+      owner_membership.stub(:user) { subject }
+      owner_membership.stub(:role) { owner }
+
+      expect(so_membership).to receive(:destroy)
+      expect(owner_membership).to receive(:destroy)
+
+      subject.set_organization_roles(org, [])
+    end
+
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aptible-auth
 version: !ruby/object:Gem::Version
-  version: 0.5.4
+  version: 0.5.5
 platform: ruby
 authors:
 - Frank Macreery
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-14 00:00:00.000000000 Z
+date: 2014-05-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aptible-resource