2222import com .cloud .network .Networks ;
2323import com .cloud .utils .db .SearchBuilder ;
2424import com .cloud .utils .db .SearchCriteria ;
25- import com .cloud .utils .db .TransactionLegacy ;
2625import org .junit .Assert ;
2726import org .junit .Test ;
2827import org .junit .runner .RunWith ;
2928import org .mockito .Mock ;
3029import org .mockito .Mockito ;
3130import org .mockito .junit .MockitoJUnitRunner ;
3231
33-
3432import java .util .List ;
3533
3634@ RunWith (MockitoJUnitRunner .class )
@@ -46,26 +44,21 @@ public class NetworkDaoImplTest {
4644 List <NetworkVO > listNetworkVoMock ;
4745
4846 @ Test
49- public void listByPhysicalNetworkTrafficTypeTestSetParametersValidation () throws Exception {
47+ public void listByPhysicalNetworkTrafficTypeTestSetParametersValidation () {
5048 NetworkDaoImpl networkDaoImplSpy = Mockito .spy (NetworkDaoImpl .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 ));
57-
58- long expectedPhysicalNetwork = 2513l ;
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 ));
5953
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- }
54+ long expectedPhysicalNetwork = 2513l ;
6555
66- Mockito .verify (searchCriteriaNetworkVoMock , Mockito .times (Networks .TrafficType .values ().length )).setParameters ("physicalNetwork" , expectedPhysicalNetwork );
67- } finally {
68- txn .close ();
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 );
6960 }
61+
62+ Mockito .verify (searchCriteriaNetworkVoMock , Mockito .times (Networks .TrafficType .values ().length )).setParameters ("physicalNetworkId" , expectedPhysicalNetwork );
7063 }
7164}
0 commit comments