package org.apache.continuum.web.test.parent;

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */

import org.testng.annotations.BeforeClass;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Optional;
import org.testng.annotations.Parameters;

import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;

public abstract class AbstractAdminTest
    extends AbstractContinuumTest
{
    protected String buildAgentUrl;

    @BeforeMethod( alwaysRun = true )
    public void loginAsAdmin()
    {
        loginAs( getProperty( "ADMIN_USERNAME" ), getProperty( "ADMIN_PASSWORD" ) );
    }

    protected void loginAs( String username, String password )
    {
        if ( !getSelenium().isElementPresent( "//span[@class='username' and text()='" + username + "']" ) )
        {
            login( username, password );

            assertElementPresent( "//span[@class='username' and text()='" + username + "']" );
        }
    }

    public void goToConfigurationPage()
    {
        clickLinkWithText( "Configuration" );
        assertEditConfigurationPage();
    }

    protected void enableDisableBuildAgent( String agentName, boolean enable )
    {
        assertFieldValue( agentName, "saveBuildAgent_buildAgent_url" );

        if ( enable )
        {
            checkField( "saveBuildAgent_buildAgent_enabled" );
        }
        else
        {
            uncheckField( "saveBuildAgent_buildAgent_enabled" );
        }
        submit();
        assertBuildAgentPage();
        assertTextPresent( Boolean.toString( enable ) );
    }

    protected void goToAddBuildAgentGroup()
    {
        goToBuildAgentPage();
        clickAndWait( "editBuildAgentGroup_0" ); //add button
        String[] options = new String[]{"--- Available Build Agents ---"};
        assertAddEditBuildAgentGroupPage( options, null );
    }

    protected void addEditBuildAgentGroup( String name, String[] addBuildAgents, String[] removeBuildAgents,
                                           boolean success )
    {
        setFieldValue( "saveBuildAgentGroup_buildAgentGroup_name", name );
        if ( addBuildAgents != null && addBuildAgents.length > 0 )
        {
            for ( String ba : addBuildAgents )
            {
                selectValue( "buildAgentIds", ba );
                clickButtonWithValue( "->", false );
            }
        }
        if ( removeBuildAgents != null && removeBuildAgents.length > 0 )
        {
            for ( String ba : removeBuildAgents )
            {
                selectValue( "selectedBuildAgentIds", ba );
                clickButtonWithValue( "<-", false );
            }
        }
        submit();
        if ( success )
        {
            assertBuildAgentPage();
        }
        else
        {
            assertAddEditBuildAgentGroupPage( null, null );
        }
    }

    void assertAddEditBuildAgentGroupPage( String[] availableBuildAgents, String[] usedBuildAgents )
    {
        assertPage( "Continuum - Add/Edit Build Agent Group" );
        assertTextPresent( "Add/Edit Build Agent Group" );
        assertTextPresent( "Name*:" );
        assertTextPresent( "Configure the used Build Agents:" );
        assertElementPresent( "buildAgentGroup.name" );
        if ( availableBuildAgents != null && availableBuildAgents.length > 0 )
        {
            assertOptionPresent( "buildAgentIds", availableBuildAgents );
        }
        if ( usedBuildAgents != null && usedBuildAgents.length > 0 )
        {
            assertOptionPresent( "selectedBuildAgentIds", usedBuildAgents );
        }

        assertButtonWithValuePresent( "Save" );
        assertButtonWithValuePresent( "Cancel" );
    }

    protected void goToEditBuildAgentGroup( String name, String[] buildAgents )
    {
        goToBuildAgentPage();
        String xPath = "//preceding::td[text()='" + name + "']//following::img[@alt='Edit']";
        clickLinkWithXPath( xPath );
        assertAddEditBuildAgentGroupPage( null, buildAgents );
        assertFieldValue( name, "buildAgentGroup.name" );
    }

    protected void removeBuildAgentGroup( String name )
        throws UnsupportedEncodingException
    {
        removeBuildAgentGroup( name, true );
    }

    protected void removeBuildAgentGroup( String name, boolean failIfMissing )
        throws UnsupportedEncodingException
    {
        goToBuildAgentPage();
        if ( isTextPresent( name ) || failIfMissing )
        {
            clickLinkWithXPath(
                "(//a[contains(@href,'deleteBuildAgentGroup.action') and contains(@href, '" + URLEncoder.encode( name,
                                                                                                                 "UTF-8" ) +
                    "')])//img" );
            assertPage( "Continuum - Delete Build Agent Group" );
            assertTextPresent( "Delete Build Agent" );
            assertTextPresent( "Are you sure you want to delete build agent group " + name + " ?" );
            assertButtonWithValuePresent( "Delete" );
            assertButtonWithValuePresent( "Cancel" );
            clickButtonWithValue( "Delete" );
            assertBuildAgentPage();
            assertTextNotPresent( name );
        }
    }

    protected void addBuildAgent( String buildAgentUrl )
    {
        addBuildAgent( buildAgentUrl, "Default description" );
    }

    protected void addBuildAgent( String buildAgentUrl, String description )
    {
        goToBuildAgentPage();
        assertBuildAgentPage();

        if ( !isElementPresent( "link=" + buildAgentUrl ) )
        {

            clickAndWait( "editBuildAgent_0" ); //add button
            assertAddEditBuildAgentPage( true );

            setFieldValue( "saveBuildAgent_buildAgent_url", buildAgentUrl );
            setFieldValue( "saveBuildAgent_buildAgent_description", description );
            checkField( "saveBuildAgent_buildAgent_enabled" );

            submit();

            assertBuildAgentPage();
            assertElementPresent( "link=" + buildAgentUrl );
        }
    }

    protected void goToAddBuildAgent()
    {
        goToBuildAgentPage();
        assertBuildAgentPage();
        clickAndWait( "editBuildAgent_0" ); //add button
        assertAddEditBuildAgentPage( true );
    }

    void assertAddEditBuildAgentPage( boolean isChecked )
    {
        assertPage( "Continuum - Add/Edit Build Agent" );
        assertTextPresent( "Add/Edit Build Agent" );
        assertTextPresent( "Build Agent URL*:" );
        assertTextPresent( "Description:" );
        assertTextPresent( "Enabled" );
        assertElementPresent( "saveBuildAgent_buildAgent_url" );
        assertElementPresent( "saveBuildAgent_buildAgent_description" );

        if ( isChecked )
        {
            assertIsChecked( "saveBuildAgent_buildAgent_enabled" );
        }

        assertButtonWithValuePresent( "Save" );
        assertButtonWithValuePresent( "Cancel" );
    }

    @BeforeClass( alwaysRun = true )
    @Parameters( { "buildAgentUrl" } )
    public void initializeBuildAgent(
        @Optional( "http://localhost:9595/continuum-buildagent/xmlrpc" ) String buildAgentUrl )
    {
        this.buildAgentUrl = buildAgentUrl;
    }

    protected void enableDistributedBuilds()
    {
        goToConfigurationPage();
        setFieldValue( "numberOfAllowedBuildsinParallel", "2" );
        if ( !isChecked( "distributedBuildEnabled" ) )
        {
            // must use click here so the JavaScript enabling the shared secret gets triggered
            click( "distributedBuildEnabled" );
        }
        setFieldValue( "sharedSecretPassword", SHARED_SECRET );
        clickAndWait( "css=input[value='Save']" );
        assertTextPresent( "true" );
        assertTextPresent( "Distributed Builds" );
        assertElementPresent( "link=Build Agents" );
    }

    protected void disableDistributedBuilds()
    {
        goToConfigurationPage();
        setFieldValue( "numberOfAllowedBuildsinParallel", "2" );
        if ( isChecked( "distributedBuildEnabled" ) )
        {
            uncheckField( "distributedBuildEnabled" );
        }
        submit();
        assertTextPresent( "false" );
        assertElementNotPresent( "link=Build Agents" );
    }

    protected void goToBuildAgentPage()
    {
        clickAndWait( "link=Build Agents" );
        assertPage( "Continuum - Build Agents" );
    }

    void assertBuildAgentPage()
    {
        assertPage( "Continuum - Build Agents" );
        assertTextPresent( "Build Agents" );
        assertTextPresent( "Build Agent Groups" );
        assertButtonWithValuePresent( "Add" );
    }

    protected void removeBuildAgent( String agentName )
        throws Exception
    {
        removeBuildAgent( agentName, true );
    }

    protected void removeBuildAgent( String agentName, boolean failIfMissing )
        throws Exception
    {
        String deleteButton = "//a[contains(@href,'deleteBuildAgent.action') and contains(@href, '" + URLEncoder.encode(
            agentName, "UTF-8" ) + "')]/img";
        if ( failIfMissing || isElementPresent( deleteButton ) )
        {
            clickLinkWithXPath( deleteButton );
            assertPage( "Continuum - Delete Build Agent" );
            assertTextPresent( "Delete Build Agent" );
            assertTextPresent( "Are you sure you want to delete build agent " + agentName + " ?" );
            assertButtonWithValuePresent( "Delete" );
            assertButtonWithValuePresent( "Cancel" );
            clickButtonWithValue( "Delete" );
            assertBuildAgentPage();
            assertElementNotPresent( deleteButton );
        }
    }

    protected void addBuildAgent( String agentURL, String description, boolean success, boolean enabled,
                                  boolean pingOk )
    {
        setFieldValue( "saveBuildAgent_buildAgent_url", agentURL );
        setFieldValue( "saveBuildAgent_buildAgent_description", description );

        if ( enabled )
        {
            checkField( "saveBuildAgent_buildAgent_enabled" );
        }
        else
        {
            uncheckField( "saveBuildAgent_buildAgent_enabled" );
        }

        submit();

        if ( success )
        {
            if ( pingOk )
            {
                assertBuildAgentPage();
                assertElementPresent( "link=" + agentURL );
                clickLinkWithText( agentURL );

                if ( enabled )
                {
                    assertTextPresent( "true" );
                }
            }
            else
            {
                assertTextPresent( "Unable to ping" );
                assertAddEditBuildAgentPage( true );
            }
        }
        else
        {
            assertAddEditBuildAgentPage( enabled );
        }
    }

    protected void goToEditBuildAgent( String name, String description )
    {
        goToBuildAgentPage();
        clickImgWithAlt( "Edit" );
        assertAddEditBuildAgentPage( false );
        assertFieldValue( name, "saveBuildAgent_buildAgent_url" );
        assertFieldValue( description, "saveBuildAgent_buildAgent_description" );
    }

    protected void addEditBuildAgent( String agentName, String newDesc )
    {
        assertFieldValue( agentName, "saveBuildAgent_buildAgent_url" );
        setFieldValue( "saveBuildAgent_buildAgent_description", newDesc );
        submit();
        assertBuildAgentPage();
        assertTextPresent( newDesc );
    }

    protected void removeSchedule( String name )
    {
        goToSchedulePage();
        clickLinkWithXPath( "(//a[contains(@href,'removeSchedule.action') and contains(@href, '" + name + "')])//img" );
        assertPage( "Continuum - Delete Schedule" );
        assertTextPresent( "Delete Schedule" );
        assertTextPresent( "Are you sure you want to delete the schedule \"" + name + "\"?" );
        assertButtonWithValuePresent( "Delete" );
        assertButtonWithValuePresent( "Cancel" );
        clickButtonWithValue( "Delete" );
        assertSchedulePage();
    }
}