package lib.form;

import com.gargoylesoftware.htmlunit.html.HtmlForm;
import com.gargoylesoftware.htmlunit.html.HtmlPage;
import com.gargoylesoftware.htmlunit.html.HtmlSelect;
import hudson.Extension;
import hudson.model.BallColor;
import hudson.model.Describable;
import hudson.model.Descriptor;
import hudson.model.InvisibleAction;
import hudson.model.RootAction;
import jenkins.model.Jenkins;
import net.sf.json.JSONObject;
import org.junit.Rule;
import org.junit.Test;
import org.jvnet.hudson.test.JenkinsRule;
import org.jvnet.hudson.test.TestExtension;
import org.kohsuke.stapler.StaplerRequest;

import static org.junit.Assert.assertEquals;

public class EnumTest {
    @Rule
    public JenkinsRule rule = new JenkinsRule();

    @Test
    public void testSelectionNoDefault() throws Exception {
        HtmlForm form = getForm("noDefault");
        HtmlSelect select;

        select = form.getSelectByName("enum1");
        assertEquals(BallColor.values().length, select.getOptionSize());
        assertEquals(BallColor.YELLOW.name(), select.getDefaultValue());

        select = form.getSelectByName("enum2");
        assertEquals(BallColor.values().length, select.getOptionSize());
        assertEquals(BallColor.values()[0].name(), select.getDefaultValue());
    }

    @Test
    public void testSelectionWithDefault() throws Exception {
        HtmlForm form = getForm("withDefault");
        HtmlSelect select;

        select = form.getSelectByName("enum1");
        assertEquals(BallColor.YELLOW.name(), select.getDefaultValue());

        select = form.getSelectByName("enum2");
        assertEquals(BallColor.BLUE.name(), select.getDefaultValue());
    }

    private HtmlForm getForm(String viewName) throws Exception {
        rule.jenkins.setCrumbIssuer(null);
        HtmlPage page = rule.createWebClient().goTo("test/" + viewName);
        return page.getFormByName("config");
    }

    @TestExtension
    public static class Form extends InvisibleAction implements RootAction, Describable<EnumTest.Form> {

        public BallColor enum1 = BallColor.YELLOW;
        public BallColor enum2 = null;

        public void doSubmitForm(StaplerRequest req) throws Exception {
            JSONObject json = req.getSubmittedForm();
            System.out.println(json);
        }

        @Override
        public Form.DescriptorImpl getDescriptor() {
            return Jenkins.get().getDescriptorByType(Form.DescriptorImpl.class);
        }

        @Override
        public String getUrlName() {
            return "test";
        }

        @Extension
        public static final class DescriptorImpl extends Descriptor<EnumTest.Form> {}
    }
}
