package Ġcom . test kn ight . services Ċ Ċ import Ġcom . intellij . psi . Psi Method Ċ import Ġcom . test kn ight . extensions . Test K night TestCase Ċ import Ġcom . test kn ight . models . TestClass Data Ċ import Ġcom . test kn ight . models . TestMethod Data Ċ import Ġjunit . framework . TestCase Ċ import Ġorg . junit . Test Ċ Ċ internal Ġclass ĠLoad Tests Service Test Ġ: ĠTest K night TestCase () Ġ{ ĊĊĠĠĠ Ġprivate Ġval Ġservice Ġ= ĠLoad Tests Service () ĊĊĠĠĠ Ġ@ Test ĊĠĠĠ Ġfun Ġtest Load Tests Simple () Ġ{ ĊĠĠĠĠĠĠĠ Ġval Ġdata Ġ= Ġget Basic Test Info ("/ Point Test . java ") ĊĊĠĠĠĠĠĠĠ Ġval Ġresult Ġ= Ġservice . get Tests ( data . psi File ) ĊĠĠĠĠĠĠĠ Ġval Ġexpected Ġ= Ġif Ġ( data . psi Class Ġ!= Ġnull ) ĠlistOf ( ĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data (" translate Test ", Ġ" Point Test ", Ġdata . psi Class !!. find Methods ByName (" translate Test ")[ 0 ] Ġas ĠPsi Method ), ĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data (" set X Test ", Ġ" Point Test ", Ġdata . psi Class !!. find Methods ByName (" set X Test ")[ 0 ] Ġas ĠPsi Method ), ĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data (" set Y Test ", Ġ" Point Test ", Ġdata . psi Class !!. find Methods ByName (" set Y Test ")[ 0 ] Ġas ĠPsi Method ), ĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data (" parameter ized Test ", Ġ" Point Test ", Ġdata . psi Class !!. find Methods ByName (" parameter ized Test ")[ 0 ] Ġas ĠPsi Method ) ĊĠĠĠĠĠĠĠ Ġ) Ġelse ĠlistOf < TestMethod Data >() ĊĊĠĠĠĠĠĠĠ ĠTestCase . assertEquals ( expected , Ġresult ) ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġ@ Test ĊĠĠĠ Ġfun Ġtest Load Tests Full Annotations () Ġ{ ĊĠĠĠĠĠĠĠ Ġval Ġdata Ġ= Ġget Basic Test Info ("/ Point Test Full Annotations . java ") ĊĠĠĠĠĠĠĠ Ġval Ġresult Ġ= Ġservice . get Tests ( data . psi File ) ĊĠĠĠĠĠĠĠ Ġval Ġexpected Ġ= Ġif Ġ( data . psi Class Ġ!= Ġnull ) ĠlistOf ( ĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data (" translate Test ", Ġ" Point Test ", Ġdata . psi Class !!. find Methods ByName (" translate Test ")[ 0 ] Ġas ĠPsi Method ), ĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data (" set X Test ", Ġ" Point Test ", Ġdata . psi Class !!. find Methods ByName (" set X Test ")[ 0 ] Ġas ĠPsi Method ), ĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data (" set Y Test ", Ġ" Point Test ", Ġdata . psi Class !!. find Methods ByName (" set Y Test ")[ 0 ] Ġas ĠPsi Method ), ĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data (" parameter ized Test ", Ġ" Point Test ", Ġdata . psi Class !!. find Methods ByName (" parameter ized Test ")[ 0 ] Ġas ĠPsi Method ) ĊĠĠĠĠĠĠĠ Ġ) Ġelse ĠlistOf < TestMethod Data >() ĊĠĠĠĠĠĠĠ ĠTestCase . assertEquals ( expected , Ġresult ) ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġ@ Test ĊĠĠĠ Ġfun Ġtest Load Tests From TestClass With Two Tests () Ġ{ ĊĠĠĠĠĠĠĠ Ġval Ġdata Ġ= Ġget Basic Test Info ("/ Two Test Classes Test . java ") ĊĊĠĠĠĠĠĠĠ Ġval Ġfirst Class Ġ= Ġdata . test Classes . elementAt ( 0 ) ĊĠĠĠĠĠĠĠ Ġval Ġsecond Class Ġ= Ġdata . test Classes . elementAt ( 1 ) ĊĠĠĠĠĠĠĠ Ġval Ġresult Ġ= Ġservice . get Tests ( data . psi File ) ĊĠĠĠĠĠĠĠ Ġval Ġexpected Ġ= Ġif Ġ( first Class Ġ!= Ġnull Ġ&& Ġsecond Class Ġ!= Ġnull ) ĠlistOf ( ĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data (" first Test ", Ġ" First Test ", Ġfirst Class . find Methods ByName (" first Test ")[ 0 ] Ġas ĠPsi Method ), ĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data (" second Test ", Ġ" Second Test ", Ġsecond Class . find Methods ByName (" second Test ")[ 0 ] Ġas ĠPsi Method ) ĊĠĠĠĠĠĠĠ Ġ) Ġelse ĠlistOf < TestMethod Data >() ĊĠĠĠĠĠĠĠ ĠTestCase . assertEquals ( expected , Ġresult ) ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġ@ Test ĊĠĠĠ Ġfun Ġtest Load Tests Tree From TestClass With Two Tests () Ġ{ ĊĠĠĠĠĠĠĠ Ġval Ġdata Ġ= Ġget Basic Test Info ("/ Two Test Classes Test . java ") ĊĊĠĠĠĠĠĠĠ Ġval Ġfirst Class Ġ= Ġdata . test Classes . elementAt ( 0 ) ĊĠĠĠĠĠĠĠ Ġval Ġsecond Class Ġ= Ġdata . test Classes . elementAt ( 1 ) ĊĠĠĠĠĠĠĠ Ġval Ġresult Ġ= Ġservice . get Tests Tree ( data . psi File ) ĊĊĠĠĠĠĠĠĠ Ġval Ġexpected Ġ= ĠlistOf ( ĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Class Data ( ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ" First Test ", ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ ĠlistOf ( ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data ( ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ" first Test ", ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ" First Test ", ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġfirst Class . find Methods ByName (" first Test ")[ 0 ] Ġas ĠPsi Method ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ), ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġfirst Class ĊĠĠĠĠĠĠĠĠĠĠĠ Ġ), ĊĊĠĠĠĠĠĠĠĠĠĠĠ ĠTest Class Data ( ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ" Second Test ", ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ ĠlistOf ( ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ ĠTest Method Data ( ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ" second Test ", ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ" Second Test ", ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġsecond Class . find Methods ByName (" second Test ")[ 0 ] Ġas ĠPsi Method ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ), ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġsecond Class ĊĠĠĠĠĠĠĠĠĠĠĠ Ġ) ĊĠĠĠĠĠĠĠ Ġ) ĊĊĠĠĠĠĠĠĠ ĠTestCase . assertEquals ( expected , Ġresult ) ĊĠĠĠ Ġ} Ċ } Ċ