Login

package selectprovider import ( "io/ioutil" "net/http" "net/http/httptest" "strings" "testing" "github.com/openshift/origin/pkg/auth/oauth/handlers" ) func TestSelectAuthentication(t *testing.T) { testCases := map[string]struct { ForceInterstitial bool Providers []handlers.ProviderInfo ExpectSelectedProvider bool ExpectHandled bool ExpectContains []string }{ "should select single provider": { ForceInterstitial: false, Providers: []handlers.ProviderInfo{ { Name: "provider_1", URL: "http://example.com/redirect_1/", }, }, ExpectSelectedProvider: true, ExpectHandled: false, }, "should return empty provider info when no providers": { ForceInterstitial: false, Providers: []handlers.ProviderInfo{}, ExpectSelectedProvider: false, ExpectHandled: false, }, "should render select provider when forced": { ForceInterstitial: true, Providers: []handlers.ProviderInfo{ { Name: "provider_1", URL: "http://example.com/redirect_1/", }, }, ExpectSelectedProvider: false, ExpectHandled: true, ExpectContains: []string{ `http://example.com/redirect_1/`, }, }, "should render select provider when multiple providers": { ForceInterstitial: false, Providers: []handlers.ProviderInfo{ { Name: "provider_1", URL: "http://example.com/redirect_1/", }, { Name: "provider_2", URL: "http://example.com/redirect_2/", }, }, ExpectSelectedProvider: false, ExpectHandled: true, ExpectContains: []string{ `http://example.com/redirect_1/`, `http://example.com/redirect_2/`, }, }, } for k, testCase := range testCases { selectProviderRenderer, err := NewSelectProviderRenderer("") if err != nil { t.Errorf("%s: unexpected error: %v", k, err) continue } selectProvider := NewSelectProvider(selectProviderRenderer, testCase.ForceInterstitial) resp := httptest.NewRecorder() provider, handled, err := selectProvider.SelectAuthentication(testCase.Providers, resp, &http.Request{}) if err != nil { t.Errorf("%s: unexpected error: %#v", k, err) continue } if testCase.ExpectHandled != handled { t.Errorf("%s: unexpected value for 'handled': %#v", k, handled) continue } if testCase.ExpectSelectedProvider && provider == nil { t.Errorf("%s: expected a provider to be selected", k) continue } if len(testCase.ExpectContains) > 0 { data, _ := ioutil.ReadAll(resp.Body) body := string(data) for i := range testCase.ExpectContains { if !strings.Contains(body, testCase.ExpectContains[i]) { t.Errorf("%s: did not find expected value %s: %s", k, testCase.ExpectContains[i], body) continue } } } } } func TestValidateSelectProviderTemplate(t *testing.T) { testCases := map[string]struct { Template string TemplateValid bool }{ "default provider selection template": { Template: defaultSelectProviderTemplateString, TemplateValid: true, }, "provider selection template example": { Template: SelectProviderTemplateExample, TemplateValid: true, }, "original provider selection template example": { Template: originalSelectProviderTemplateExample, TemplateValid: true, }, "template only prints first provider URL": { Template: invalidSelectProviderTemplate, TemplateValid: false, }, } for k, testCase := range testCases { allErrs := ValidateSelectProviderTemplate([]byte(testCase.Template)) if testCase.TemplateValid { for _, err := range allErrs { t.Errorf("%s: template validation failed when it should have succeeded: %v", k, err) } } else if len(allErrs) == 0 { t.Errorf("%s: template validation succeeded when it should have failed", k) } } } // Make sure the original version of the default template always validates // this is to avoid breaking existing customized templates. const originalSelectProviderTemplateExample = ` Login {{ range $provider := .Providers }}

{{ if eq $provider.Name "anypassword" }} Log in with any username and password {{ else }} {{$provider.Name}} {{ end }}

{{ end }} ` // This template only prints the first provider URL and should fail validation. const invalidSelectProviderTemplate = ` Login