firebase-admin-java/src/test/java/com/google/firebase/auth/FirebaseTokenTest.java at master · JoeCP17/firebase-admin-java · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
/*
 * Copyright  2019 Google Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.google.firebase.auth;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;

import com.google.api.client.auth.openidconnect.IdToken;
import com.google.common.collect.ImmutableMap;
import java.util.Map;
import org.junit.Test;

public class FirebaseTokenTest {

  @Test
  public void testFirebaseToken() {
    Map<String, Object> claims = ImmutableMap.<String, Object>builder()
        .put("sub", "testUser")
        .put("iss", "test-project-id")
        .put("email", "test@example.com")
        .put("email_verified", true)
        .put("name", "Test User")
        .put("picture", "https://picture.url")
        .put("custom", "claim")
        .build();

    FirebaseToken firebaseToken = new FirebaseToken(claims);

    assertEquals("testUser", firebaseToken.getUid());
    assertEquals("test-project-id", firebaseToken.getIssuer());
    assertEquals("test@example.com", firebaseToken.getEmail());
    assertTrue(firebaseToken.isEmailVerified());
    assertEquals("Test User", firebaseToken.getName());
    assertEquals("https://picture.url", firebaseToken.getPicture());
    assertEquals("claim", firebaseToken.getClaims().get("custom"));
    assertEquals(7, firebaseToken.getClaims().size());
  }

  @Test
  public void testFirebaseTokenMinimal() {
    Map<String, Object> claims = ImmutableMap.<String, Object>builder()
        .put("sub", "testUser")
        .build();

    FirebaseToken firebaseToken = new FirebaseToken(claims);

    assertEquals("testUser", firebaseToken.getUid());
    assertNull(firebaseToken.getIssuer());
    assertNull(firebaseToken.getEmail());
    assertFalse(firebaseToken.isEmailVerified());
    assertNull(firebaseToken.getName());
    assertNull(firebaseToken.getPicture());
    assertEquals(1, firebaseToken.getClaims().size());
  }

  @Test
  public void testFirebaseTokenFromIdToken() {
    IdToken.Payload payload = new IdToken.Payload()
        .setSubject("testUser")
        .setIssuer("test-project-id")
        .set("email", "test@example.com")
        .set("email_verified", true)
        .set("name", "Test User")
        .set("picture", "https://picture.url")
        .set("custom", "claim");

    FirebaseToken firebaseToken = new FirebaseToken(payload);

    assertEquals("testUser", firebaseToken.getUid());
    assertEquals("test-project-id", firebaseToken.getIssuer());
    assertEquals("test@example.com", firebaseToken.getEmail());
    assertTrue(firebaseToken.isEmailVerified());
    assertEquals("Test User", firebaseToken.getName());
    assertEquals("https://picture.url", firebaseToken.getPicture());
    assertEquals("claim", firebaseToken.getClaims().get("custom"));
    assertEquals(7, firebaseToken.getClaims().size());
  }

  @Test
  public void testFirebaseTokenFromMinimalIdToken() {
    IdToken.Payload payload = new IdToken.Payload()
        .setSubject("testUser");

    FirebaseToken firebaseToken = new FirebaseToken(payload);

    assertEquals("testUser", firebaseToken.getUid());
    assertNull(firebaseToken.getIssuer());
    assertNull(firebaseToken.getEmail());
    assertFalse(firebaseToken.isEmailVerified());
    assertNull(firebaseToken.getName());
    assertNull(firebaseToken.getPicture());
    assertEquals(1, firebaseToken.getClaims().size());
  }

  @Test(expected = IllegalArgumentException.class)
  public void testFirebaseTokenNullClaims() {
    new FirebaseToken(null);
  }

  @Test(expected = IllegalArgumentException.class)
  public void testFirebaseTokenNoUid() {
    ImmutableMap<String, Object> claimsWithoutSub = ImmutableMap.<String, Object>of(
        "custom", "claim");
    new FirebaseToken(claimsWithoutSub);
  }
}