Add spec for admin mapping

opf · Nov 2, 2023 · 46fc0f9 · 46fc0f9
1 parent 6f6464c
commit 46fc0f9
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 2 deletions.
diff --git a/test/lib/omniauth/strategies/openid_connect_test.rb b/test/lib/omniauth/strategies/openid_connect_test.rb
@@ -363,6 +363,30 @@ def test_info_custom_mapping
         assert_equal 'bar', info[:login]
       end
 
+      def test_info_boolean_mapping_false
+        strategy.options.attribute_map = { admin: 'isAdmin' }
+
+        info = strategy.info
+
+        assert_equal user_info.name, info[:name]
+        assert_equal user_info.email, info[:email]
+        assert_equal false, info[:admin]
+      end
+
+      def test_info_boolean_mapping_true
+        azure_userinfo = ::OpenIDConnect::ResponseObject::UserInfo.new(
+          sub: SecureRandom.hex(16),
+          name: Faker::Name.name,
+          isAdmin: true
+        )
+        strategy.stubs(:user_info).returns(azure_userinfo)
+        strategy.options.attribute_map = { admin: 'isAdmin' }
+
+        info = strategy.info
+
+        assert_equal true, info[:admin]
+      end
+
       def test_info_mail_unique_mapping
         azure_userinfo = ::OpenIDConnect::ResponseObject::UserInfo.new(
           sub: SecureRandom.hex(16),

diff --git a/test/strategy_test_case.rb b/test/strategy_test_case.rb
@@ -1,4 +1,4 @@
-class StrategyTestCase < MiniTest::Test
+class StrategyTestCase < Minitest::Test
   class DummyApp
     def call(env); end
   end
@@ -28,7 +28,8 @@ def user_info
       phone_number: Faker::PhoneNumber.phone_number,
       website: Faker::Internet.url,
       # custom claim
-      foobar: 'bar'
+      foobar: 'bar',
+      isAdmin: false
     )
   end