Added visibility level specs

1 files changed, 21 insertions, 0 deletions

diff --git a/spec/lib/gitlab/visibility_level_spec.rb b/spec/lib/gitlab/visibility_level_spec.rb
index 2c1146ceff5..fb1151d7e72 100644
--- a/spec/lib/gitlab/visibility_level_spec.rb
+++ b/spec/lib/gitlab/visibility_level_spec.rb
@@ -85,4 +85,25 @@ describe Gitlab::VisibilityLevel do
         .to eq(described_class::PRIVATE)
     end
   end
+
+  describe '.valid_level?' do
+    it 'returns true when visibility is a valid level' do
+      expect(described_class.valid_level?(described_class::PRIVATE)).to eq(true)
+      expect(described_class.valid_level?(described_class::INTERNAL)).to eq(true)
+      expect(described_class.valid_level?(described_class::PUBLIC)).to eq(true)
+    end
+
+    it 'returns true when argument is visibility' do
+      expect(described_class.options.value?(described_class.level_value("internal"))).to eq(true)
+    end
+
+    it 'return true when argument is visibility level' do
+      expect(described_class.options.value?(described_class.level_value(10))).to eq(true)
+    end
+
+    it 'returns false when visibility is invalid' do
+      expect(described_class.options.value?(" ")).to eq(false)
+      expect(described_class.options.value?("string")).to eq(false)
+    end
+  end
 end