From 24f36f27a73a7ec2bcd687220e1f9688fc293783 Mon Sep 17 00:00:00 2001 From: Carlos Alexandro Becker Date: Wed, 20 Mar 2019 13:52:30 -0300 Subject: [PATCH] fix: check if it is both primary and verified --- providers/github.go | 3 ++- providers/github_test.go | 19 ++++++++++++++++--- 2 files changed, 18 insertions(+), 4 deletions(-) diff --git a/providers/github.go b/providers/github.go index 26d8c3f..798b140 100644 --- a/providers/github.go +++ b/providers/github.go @@ -203,6 +203,7 @@ func (p *GitHubProvider) GetEmailAddress(s *SessionState) (string, error) { var emails []struct { Email string `json:"email"` + Primary bool `json:"primary"` Verified bool `json:"verified"` } @@ -248,7 +249,7 @@ func (p *GitHubProvider) GetEmailAddress(s *SessionState) (string, error) { } for _, email := range emails { - if email.Verified { + if email.Primary && email.Verified { return email.Email, nil } } diff --git a/providers/github_test.go b/providers/github_test.go index 2da2a32..4b093ca 100644 --- a/providers/github_test.go +++ b/providers/github_test.go @@ -97,7 +97,7 @@ func TestGitHubProviderOverrides(t *testing.T) { } func TestGitHubProviderGetEmailAddress(t *testing.T) { - b := testGitHubBackend([]string{`[ {"email": "michael.bland@gsa.gov", "verified": true} ]`}) + b := testGitHubBackend([]string{`[ {"email": "michael.bland@gsa.gov", "verified": true, "primary": true} ]`}) defer b.Close() bURL, _ := url.Parse(b.URL) @@ -109,10 +109,23 @@ func TestGitHubProviderGetEmailAddress(t *testing.T) { assert.Equal(t, "michael.bland@gsa.gov", email) } +func TestGitHubProviderGetEmailAddressNotVerified(t *testing.T) { + b := testGitHubBackend([]string{`[ {"email": "michael.bland@gsa.gov", "verified": false, "primary": true} ]`}) + defer b.Close() + + bURL, _ := url.Parse(b.URL) + p := testGitHubProvider(bURL.Host) + + session := &SessionState{AccessToken: "imaginary_access_token"} + email, err := p.GetEmailAddress(session) + assert.Equal(t, nil, err) + assert.Empty(t, "", email) +} + func TestGitHubProviderGetEmailAddressWithOrg(t *testing.T) { b := testGitHubBackend([]string{ - `[ {"email": "michael.bland@gsa.gov", "verified": true, "login":"testorg"} ]`, - `[ {"email": "michael.bland1@gsa.gov", "verified": true, "login":"testorg1"} ]`, + `[ {"email": "michael.bland@gsa.gov", "primary": true, "verified": true, "login":"testorg"} ]`, + `[ {"email": "michael.bland1@gsa.gov", "primary": true, "verified": true, "login":"testorg1"} ]`, `[ ]`, }) defer b.Close()