1 files changed, 17 insertions, 0 deletions
diff --git a/spec/policies/base_policy_spec.rb b/spec/policies/base_policy_spec.rb
new file mode 100644
index 00000000000..63acc0b68cd
--- /dev/null
+++ b/spec/policies/base_policy_spec.rb
@@ -0,0 +1,17 @@
+require 'spec_helper'
+
+describe BasePolicy, models: true do
+  let(:build) { Ci::Build.new }
+
+  describe '.class_for' do
+    it 'detects policy class based on the subject ancestors' do
+      expect(described_class.class_for(build)).to eq(Ci::BuildPolicy)
+    end
+
+    it 'detects policy class for a presented subject' do
+      presentee = Ci::BuildPresenter.new(build)
+
+      expect(described_class.class_for(presentee)).to eq(Ci::BuildPolicy)
+    end
+  end
+end