package com.github.johrstrom.test;

import com.github.johrstrom.collector.BaseCollectorConfig;
import com.github.johrstrom.collector.BaseCollectorConfig.JMeterCollectorType;
import com.github.johrstrom.listener.ListenerCollectorConfig;
import com.github.johrstrom.listener.ListenerCollectorConfig.Measurable;
import org.apache.jmeter.control.LoopController;
import org.apache.jmeter.engine.StandardJMeterEngine;
import org.apache.jmeter.gui.tree.JMeterTreeListener;
import org.apache.jmeter.gui.tree.JMeterTreeModel;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.save.SaveService;
import org.apache.jmeter.threads.ThreadGroup;
import org.apache.jmeter.threads.*;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jorphan.collections.HashTree;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;

public class TestUtilities {
	
	public static final String TEST_VAR_NAME = "arbitrary_var";
	public static final String TEST_SAMPLER_NAME = "super_cool_sampler";
	public static final String TEST_ASSERTION_NAME = "super_cool_assertion";
	public static final String TEST_ASSERTION_NAME_ALT = "other_super_cool_assertion";
	public static final String TEST_SAMPLER_CODE = "909";
	public static final String TEST_VAR_VALUE = "bar_value";
	
	public static final String[] TEST_LABELS = new String[] {TEST_VAR_NAME,"label","code"};
	public static final String[] EXPECTED_LABELS = new String[] {TEST_VAR_VALUE, TEST_SAMPLER_NAME, TEST_SAMPLER_CODE};
	
	public static final String[] TEST_ASSERTION_LABELS = new String[] {TEST_VAR_NAME,"label"};
	public static final String[] EXPECTED_ASSERTION_LABELS = new String[] {TEST_VAR_VALUE, TEST_ASSERTION_NAME};
	public static final String[] EXPECTED_ASSERTION_LABELS_ALT = new String[] {TEST_VAR_VALUE, TEST_ASSERTION_NAME_ALT};
	
	public static BaseCollectorConfig simpleCounterCfg() {
		BaseCollectorConfig cfg = new BaseCollectorConfig();
		cfg.setMetricName("simple_counter");
		cfg.setType(JMeterCollectorType.COUNTER.toString());
		cfg.setHelp("some helpe message");
		
		return cfg;
	}
	
	public static ListenerCollectorConfig listenerCounterCfg(String name, Measurable measurable, String listenTo) {
		BaseCollectorConfig base = TestUtilities.simpleCounterCfg();
		base.setLabels(TestUtilities.TEST_ASSERTION_LABELS);
		ListenerCollectorConfig cfg = new ListenerCollectorConfig(base);
		cfg.setMetricName(name);
		cfg.setMeasuring(measurable.toString());
		cfg.setListenTo(listenTo);
		
		return cfg;
	}
	
	public static ListenerCollectorConfig listenerSuccessRatioCfg(String name, String listenTo) {
		ListenerCollectorConfig cfg = new ListenerCollectorConfig();
		cfg.setLabels(TestUtilities.TEST_ASSERTION_LABELS);
		cfg.setMetricName(name);
		cfg.setType(JMeterCollectorType.SUCCESS_RATIO.toString());
		cfg.setHelp("some helpe message");
		cfg.setListenTo(listenTo);
		
		return cfg;
	}
	
	public static BaseCollectorConfig simpleSummaryCfg() {
		BaseCollectorConfig cfg = new BaseCollectorConfig();
		cfg.setMetricName("simple_summary");
		cfg.setType(JMeterCollectorType.SUMMARY.toString());
		cfg.setHelp("some helpe message");
		
		return cfg;
	}
	
	public static BaseCollectorConfig simpleHistogramCfg() {
		BaseCollectorConfig cfg = new BaseCollectorConfig();
		cfg.setMetricName("simple_histogram");
		cfg.setType(JMeterCollectorType.HISTOGRAM.toString());
		cfg.setHelp("some helpe message");
		
		return cfg;
	}
	
	public static BaseCollectorConfig simpleGaugeCfg() {
		BaseCollectorConfig cfg = new BaseCollectorConfig();
		cfg.setMetricName("simple_gauge");
		cfg.setType(JMeterCollectorType.GAUGE.toString());
		cfg.setHelp("some helpe message");
		
		return cfg;
	}
	
	public static BaseCollectorConfig simpleSuccessRatioCfg() {
		BaseCollectorConfig cfg = new BaseCollectorConfig();
		cfg.setMetricName("simple_ratio");
		cfg.setType(JMeterCollectorType.SUCCESS_RATIO.toString());
		cfg.setHelp("some helpe message");
		
		return cfg;
	}

    public static void createJmeterEnv() {
    	
        JMeterUtils.setJMeterHome("src/test/resources");
        JMeterUtils.setLocale(Locale.ENGLISH);
        JMeterUtils.loadJMeterProperties("src/test/resources/bin/jmeter.properties");
        
        try {
			SaveService.loadProperties();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}

        JMeterTreeModel jMeterTreeModel = new JMeterTreeModel();
        JMeterTreeListener jMeterTreeListener = new JMeterTreeListener();
        jMeterTreeListener.setModel(jMeterTreeModel);
        
        JMeterContextService.getContext().setVariables(new JMeterVariables());
        StandardJMeterEngine engine = new StandardJMeterEngine();
        JMeterContextService.getContext().setEngine(engine);
        
        JMeterThreadMonitor monitor = new NOOPThreadMonitor();
        
        
        HashTree hashtree = new HashTree();
        hashtree.add(new LoopController());
        
        JMeterThread thread = new JMeterThread(hashtree, monitor, null);
        thread.setThreadName("test thread");
        JMeterContextService.getContext().setThread(thread);
        
        
        ThreadGroup tg1 = new ThreadGroup();
        tg1.setName("tg1");
        JMeterContextService.getContext().setThreadGroup(tg1);
        
    }
    
    public static List<BaseCollectorConfig> simpleListConfig() {
    	List<BaseCollectorConfig> list = new ArrayList<BaseCollectorConfig>(); 
    	
    	list.add(simpleGaugeCfg());
    	list.add(simpleHistogramCfg());
    	list.add(simpleSummaryCfg());
    	list.add(simpleCounterCfg());
    	list.add(simpleSuccessRatioCfg());
    	
    	return list;
    }
    
    public static List<ListenerCollectorConfig> simpleListListener() {
    	List<ListenerCollectorConfig> list = new ArrayList<ListenerCollectorConfig>(); 
    	
    	list.add(new ListenerCollectorConfig(simpleGaugeCfg()));
    	list.add(new ListenerCollectorConfig(simpleHistogramCfg()));
    	list.add(new ListenerCollectorConfig(simpleSummaryCfg()));
    	list.add(new ListenerCollectorConfig(simpleCounterCfg()));
    	list.add(new ListenerCollectorConfig(simpleSuccessRatioCfg()));
    	
    	return list;
    }
    
    public static List<ListenerCollectorConfig> fullListListener(){
    	List<ListenerCollectorConfig> list = new ArrayList<ListenerCollectorConfig>();
    	
    	// ---------- counters and success ratio
    	ListenerCollectorConfig cfg = new ListenerCollectorConfig(simpleCounterCfg());
    	cfg = redoNameAndMeasuring(cfg, "test_count_total", Measurable.CountTotal);
    	list.add(cfg);
    	
    	cfg = redoNameAndMeasuring(cfg, "test_failure_total", Measurable.FailureTotal);
    	list.add(cfg);
    	
    	cfg = redoNameAndMeasuring(cfg, "test_success_total", Measurable.SuccessTotal);
    	list.add(cfg);
    	
    	cfg = new ListenerCollectorConfig(simpleSuccessRatioCfg());
    	cfg = redoNameAndMeasuring(cfg, "test_ratio", Measurable.SuccessRatio);
    	list.add(cfg);
    	
    	// ------- histograms
    	cfg = new ListenerCollectorConfig(simpleHistogramCfg());
    	cfg = redoNameAndMeasuring(cfg, "test_hist_rtime", Measurable.ResponseTime);
    	list.add(cfg);
    
    	cfg = redoNameAndMeasuring(cfg, "test_hist_rsize", Measurable.ResponseSize);
    	list.add(cfg);
    	
    	cfg = redoNameAndMeasuring(cfg, "test_hist_latency", Measurable.Latency);
    	list.add(cfg);
    	
    	cfg = redoNameAndMeasuring(cfg, "test_hist_idle_time", Measurable.IdleTime);
    	list.add(cfg);
    	
    	cfg = redoNameAndMeasuring(cfg, "test_hist_connect_time", Measurable.ConnectTime);
    	list.add(cfg);
    
    	// -------- summaries
    	cfg = new ListenerCollectorConfig(simpleSummaryCfg());
    	cfg = redoNameAndMeasuring(cfg, "test_summary_rtime", Measurable.ResponseTime);
    	list.add(cfg);
    
    	cfg = redoNameAndMeasuring(cfg, "test_summary_rsize", Measurable.ResponseSize);
    	list.add(cfg);
    	
    	cfg = redoNameAndMeasuring(cfg, "test_summary_latency", Measurable.Latency);
    	list.add(cfg);    	
    	
    	cfg = redoNameAndMeasuring(cfg, "test_summary_idle_time", Measurable.IdleTime);
    	list.add(cfg);
    	
    	cfg = redoNameAndMeasuring(cfg, "test_summary_connect_time", Measurable.ConnectTime);
    	list.add(cfg);
    	
    	return list;
    }
    
    public static ListenerCollectorConfig redoNameAndMeasuring(ListenerCollectorConfig cfg, String name, Measurable m) {
    	ListenerCollectorConfig clone = (ListenerCollectorConfig) cfg.clone();
    	
    	clone.setMetricName(name);
    	clone.setMeasuring(m.toString());
    	   	
    	return clone;
    }
    
    public static ResultAndVariables resultWithLabels() {
    	SampleResult result = new SampleResult();
    	
    	result.setSampleLabel(TEST_SAMPLER_NAME);
    	result.setResponseCode(TEST_SAMPLER_CODE);
    	
    	JMeterVariables vars = new JMeterVariables();
		vars.put(TEST_VAR_NAME, TEST_VAR_VALUE);
		JMeterContextService.getContext().setVariables(vars);
		
		return new ResultAndVariables(result, vars);
    }
    
    public static class ResultAndVariables {
    	public SampleResult result;
    	public JMeterVariables vars;
    	
    	public ResultAndVariables(SampleResult result, JMeterVariables vars) {
    		this.result = result;
    		this.vars = vars;
    	}
    }
    
    
}
