add spec for patched belongs_to

This commit is contained in:
Konstantin Haase 2015-02-23 17:18:07 +01:00
parent 0a485d2137
commit 135b4b4a99
2 changed files with 18 additions and 1 deletions

View File

@ -41,7 +41,7 @@ module Travis::API::V3
end
def []=(key, value)
value &&= value.sub("#{self.class.parent}::") if self.class.polymorfic_foreign_types.include?(key)
value &&= value.sub("#{self.class.parent}::", ''.freeze) if self.class.polymorfic_foreign_types.include?(key)
super(key, value)
end
end

View File

@ -0,0 +1,17 @@
require 'spec_helper'
describe Travis::API::V3::Extensions::BelongsTo do
describe 'reading polymorphic relation' do
subject(:repo) { Travis::API::V3::Models::Repository.first }
example { expect(repo.owner).to be_a(Travis::API::V3::Models::User) }
end
describe 'writing polymorphic relation' do
let(:repo) { Travis::API::V3::Models::Repository.create(owner: user) }
let(:user) { Travis::API::V3::Models::User.create }
after { repo.destroy; user.destroy }
example { expect(repo.owner).to be_a(Travis::API::V3::Models::User) }
example { expect(::Repository.find(repo.id).owner).to be_a(::User) }
end
end