2222import com .cloud .network .Networks ;
2323import com .cloud .utils .db .SearchBuilder ;
2424import com .cloud .utils .db .SearchCriteria ;
25+ import com .cloud .utils .db .TransactionLegacy ;
2526import org .junit .Assert ;
2627import org .junit .Test ;
2728import org .junit .runner .RunWith ;
2829import org .mockito .Mock ;
2930import org .mockito .Mockito ;
3031import org .mockito .junit .MockitoJUnitRunner ;
3132
33+
3234import java .util .List ;
3335
3436@ RunWith (MockitoJUnitRunner .class )
@@ -44,21 +46,26 @@ public class NetworkDaoImplTest {
4446 List <NetworkVO > listNetworkVoMock ;
4547
4648 @ Test
47- public void listByPhysicalNetworkTrafficTypeTestSetParametersValidation () {
49+ public void listByPhysicalNetworkTrafficTypeTestSetParametersValidation () throws Exception {
4850 NetworkDaoImpl networkDaoImplSpy = Mockito .spy (NetworkDaoImpl .class );
49- networkDaoImplSpy .AllFieldsSearch = searchBuilderNetworkVoMock ;
50- Mockito .doReturn (searchCriteriaNetworkVoMock ).when (searchBuilderNetworkVoMock ).create ();
51- Mockito .doNothing ().when (searchCriteriaNetworkVoMock ).setParameters (Mockito .anyString (), Mockito .any ());
52- Mockito .doReturn (listNetworkVoMock ).when (networkDaoImplSpy ).listBy (Mockito .any (SearchCriteria .class ));
51+ TransactionLegacy txn = TransactionLegacy .open ("runNetworkDaoImplTest" );
52+ try {
53+ networkDaoImplSpy .AllFieldsSearch = searchBuilderNetworkVoMock ;
54+ Mockito .doReturn (searchCriteriaNetworkVoMock ).when (searchBuilderNetworkVoMock ).create ();
55+ Mockito .doNothing ().when (searchCriteriaNetworkVoMock ).setParameters (Mockito .anyString (), Mockito .any ());
56+ Mockito .doReturn (listNetworkVoMock ).when (networkDaoImplSpy ).listBy (Mockito .any (SearchCriteria .class ));
5357
54- long expectedPhysicalNetwork = 2513l ;
58+ long expectedPhysicalNetwork = 2513l ;
5559
56- for (Networks .TrafficType trafficType : Networks .TrafficType .values ()) {
57- List <NetworkVO > result = networkDaoImplSpy .listByPhysicalNetworkTrafficType (expectedPhysicalNetwork , trafficType );
58- Assert .assertEquals (listNetworkVoMock , result );
59- Mockito .verify (searchCriteriaNetworkVoMock ).setParameters ("trafficType" , trafficType );
60- }
60+ for (Networks .TrafficType trafficType : Networks .TrafficType .values ()) {
61+ List <NetworkVO > result = networkDaoImplSpy .listByPhysicalNetworkTrafficType (expectedPhysicalNetwork , trafficType );
62+ Assert .assertEquals (listNetworkVoMock , result );
63+ Mockito .verify (searchCriteriaNetworkVoMock ).setParameters ("trafficType" , trafficType );
64+ }
6165
62- Mockito .verify (searchCriteriaNetworkVoMock , Mockito .times (Networks .TrafficType .values ().length )).setParameters ("physicalNetworkId" , expectedPhysicalNetwork );
66+ Mockito .verify (searchCriteriaNetworkVoMock , Mockito .times (Networks .TrafficType .values ().length )).setParameters ("physicalNetwork" , expectedPhysicalNetwork );
67+ } finally {
68+ txn .close ();
69+ }
6370 }
6471}
0 commit comments