Skip to content

Commit d32f8c2

Browse files
committed
Fix query filter and units tests
1 parent 2600965 commit d32f8c2

File tree

2 files changed

+13
-20
lines changed

2 files changed

+13
-20
lines changed

engine/schema/src/main/java/com/cloud/network/dao/NetworkDaoImpl.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -598,7 +598,7 @@ public List<NetworkVO> listByPhysicalNetwork(final long physicalNetworkId) {
598598
public List<NetworkVO> listByPhysicalNetworkTrafficType(final long physicalNetworkId, final TrafficType trafficType) {
599599
final SearchCriteria<NetworkVO> sc = AllFieldsSearch.create();
600600
sc.setParameters("trafficType", trafficType);
601-
sc.setParameters("physicalNetwork", physicalNetworkId);
601+
sc.setParameters("physicalNetworkId", physicalNetworkId);
602602
return listBy(sc);
603603
}
604604

engine/schema/src/test/java/com/cloud/network/dao/NetworkDaoImplTest.java

Lines changed: 12 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -22,15 +22,13 @@
2222
import com.cloud.network.Networks;
2323
import com.cloud.utils.db.SearchBuilder;
2424
import com.cloud.utils.db.SearchCriteria;
25-
import com.cloud.utils.db.TransactionLegacy;
2625
import org.junit.Assert;
2726
import org.junit.Test;
2827
import org.junit.runner.RunWith;
2928
import org.mockito.Mock;
3029
import org.mockito.Mockito;
3130
import org.mockito.junit.MockitoJUnitRunner;
3231

33-
3432
import 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

Comments
 (0)