1 package org.testng.xml; 2 3 import org.testng.TestNGException; 4 import org.testng.reporters.XMLStringBuffer; 5 import org.testng.xml.dom.OnElement; 6 7 import java.util.Properties; 8 9 /** 10 * This class describes the tag <method-selector> in testng.xml. 11 * 12 * Created on Sep 26, 2005 13 * @author cbeust 14 * @author <a href='mailto:the_mindstorm[at]evolva[dot]ro'>Alexandru Popescu</a> 15 */ 16 public class XmlMethodSelector { 17 // Either this: 18 private String m_className; 19 private int m_priority; 20 21 // Or that: 22 private XmlScript m_script = new XmlScript(); 23 24 // For YAML setClassName(String s)25 public void setClassName(String s) { 26 m_className = s; 27 } 28 getClassName()29 public String getClassName() { 30 return m_className; 31 } 32 33 // For YAML 34 @OnElement(tag = "selector-class", attributes = { "name", "priority" }) setElement(String name, String priority)35 public void setElement(String name, String priority) { 36 setName(name); 37 setPriority(Integer.parseInt(priority)); 38 } 39 setName(String name)40 public void setName(String name) { 41 m_className = name; 42 } 43 setScript(XmlScript script)44 public void setScript(XmlScript script) { 45 m_script = script; 46 } 47 48 /** 49 * @return Returns the expression. 50 */ getExpression()51 public String getExpression() { 52 return m_script.getScript(); 53 } 54 55 /** 56 * @param expression The expression to set. 57 */ setExpression(String expression)58 public void setExpression(String expression) { 59 m_script.setScript(expression); 60 } 61 62 /** 63 * @return Returns the language. 64 */ getLanguage()65 public String getLanguage() { 66 return m_script.getLanguage(); 67 } 68 69 /** 70 * @param language The language to set. 71 */ 72 // @OnElement(tag = "script", attributes = "language") setLanguage(String language)73 public void setLanguage(String language) { 74 m_script.setLanguage(language); 75 // m_language = language; 76 } 77 getPriority()78 public int getPriority() { 79 return m_priority; 80 } 81 setPriority(int priority)82 public void setPriority(int priority) { 83 m_priority = priority; 84 } 85 ppp(String s)86 private void ppp(String s) { 87 System.out.println("[XmlMethodSelector] " + s); 88 } 89 toXml(String indent)90 public String toXml(String indent) { 91 XMLStringBuffer xsb = new XMLStringBuffer(indent); 92 93 xsb.push("method-selector"); 94 95 if (null != m_className) { 96 Properties clsProp = new Properties(); 97 clsProp.setProperty("name", getClassName()); 98 if(getPriority() != -1) { 99 clsProp.setProperty("priority", String.valueOf(getPriority())); 100 } 101 xsb.addEmptyElement("selector-class", clsProp); 102 } 103 else if (getLanguage() != null) { 104 Properties scriptProp = new Properties(); 105 scriptProp.setProperty("language", getLanguage()); 106 xsb.push("script", scriptProp); 107 xsb.addCDATA(getExpression()); 108 xsb.pop("script"); 109 } 110 else { 111 throw new TestNGException("Invalid Method Selector: found neither class name nor language"); 112 } 113 114 xsb.pop("method-selector"); 115 116 return xsb.toXML(); 117 } 118 119 @Override hashCode()120 public int hashCode() { 121 final int prime = 31; 122 int result = 1; 123 result = prime * result 124 + ((m_className == null) ? 0 : m_className.hashCode()); 125 result = prime * result 126 + ((getExpression() == null) ? 0 : getExpression().hashCode()); 127 result = prime * result 128 + ((getLanguage() == null) ? 0 : getLanguage().hashCode()); 129 result = prime * result + m_priority; 130 return result; 131 } 132 133 @Override equals(Object obj)134 public boolean equals(Object obj) { 135 if (this == obj) 136 return true; 137 if (obj == null) 138 return XmlSuite.f(); 139 if (getClass() != obj.getClass()) 140 return XmlSuite.f(); 141 XmlMethodSelector other = (XmlMethodSelector) obj; 142 if (m_className == null) { 143 if (other.m_className != null) 144 return XmlSuite.f(); 145 } else if (!m_className.equals(other.m_className)) 146 return XmlSuite.f(); 147 if (getExpression() == null) { 148 if (other.getExpression() != null) 149 return XmlSuite.f(); 150 } else if (!getExpression().equals(other.getExpression())) 151 return XmlSuite.f(); 152 if (getLanguage() == null) { 153 if (other.getLanguage() != null) 154 return XmlSuite.f(); 155 } else if (!getLanguage().equals(other.getLanguage())) 156 return XmlSuite.f(); 157 if (m_priority != other.m_priority) 158 return XmlSuite.f(); 159 return true; 160 } 161 } 162