1 files changed, 33 insertions, 0 deletions
diff --git a/spec/models/instance_metadata/kas_spec.rb b/spec/models/instance_metadata/kas_spec.rb
new file mode 100644
index 00000000000..f8cc34fa8d3
--- /dev/null
+++ b/spec/models/instance_metadata/kas_spec.rb
@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe ::InstanceMetadata::Kas do
+  it 'has InstanceMetadataPolicy as declarative policy' do
+    expect(described_class.declarative_policy_class).to eq("InstanceMetadataPolicy")
+  end
+
+  context 'when KAS is enabled' do
+    it 'has the correct properties' do
+      allow(Gitlab::Kas).to receive(:enabled?).and_return(true)
+
+      expect(subject).to have_attributes(
+        enabled: Gitlab::Kas.enabled?,
+        version: Gitlab::Kas.version,
+        external_url: Gitlab::Kas.external_url
+      )
+    end
+  end
+
+  context 'when KAS is disabled' do
+    it 'has the correct properties' do
+      allow(Gitlab::Kas).to receive(:enabled?).and_return(false)
+
+      expect(subject).to have_attributes(
+        enabled: Gitlab::Kas.enabled?,
+        version: nil,
+        external_url: nil
+      )
+    end
+  end
+end