Skip to content

Commit 0dd0f1b

Browse files
Daniel Augusto Veronezi SalvadorSadiJr
authored andcommitted
Correção na deleção de _traffic types_
1 parent ebc5744 commit 0dd0f1b

2 files changed

Lines changed: 13 additions & 20 deletions

File tree

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
@@ -569,7 +569,7 @@ public List<NetworkVO> listByPhysicalNetwork(final long physicalNetworkId) {
569569
public List<NetworkVO> listByPhysicalNetworkTrafficType(final long physicalNetworkId, final TrafficType trafficType) {
570570
final SearchCriteria<NetworkVO> sc = AllFieldsSearch.create();
571571
sc.setParameters("trafficType", trafficType);
572-
sc.setParameters("physicalNetwork", physicalNetworkId);
572+
sc.setParameters("physicalNetworkId", physicalNetworkId);
573573
return listBy(sc);
574574
}
575575

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)