package com.cloud.network.dao; import java.util.ArrayList; import java.util.List; import junit.framework.TestCase; import com.cloud.network.Network.GuestIpType; import com.cloud.network.NetworkVO; import com.cloud.network.Networks.BroadcastDomainType; import com.cloud.network.Networks.Mode; import com.cloud.network.Networks.TrafficType; import com.cloud.utils.component.ComponentLocator; public class NetworkDaoTest extends TestCase { public void testTags() { NetworkDaoImpl dao = ComponentLocator.inject(NetworkDaoImpl.class); dao.expunge(1001l); NetworkVO network = new NetworkVO(1001, TrafficType.Control, GuestIpType.Direct, Mode.Dhcp, BroadcastDomainType.Native, 1, 1, 1, 1, 1001, "Name", "DisplayText", false, true, null); network.setGuruName("guru_name"); List tags = new ArrayList(); tags.add("a"); tags.add("b"); network.setTags(tags); network = dao.persist(network); List saveTags = network.getTags(); assert(saveTags.size() == 2 && saveTags.contains("a") && saveTags.contains("b")); NetworkVO retrieved = dao.findById(1001l); List retrievedTags = retrieved.getTags(); assert(retrievedTags.size() == 2 && retrievedTags.contains("a") && retrievedTags.contains("b")); List updateTags = new ArrayList(); updateTags.add("e"); updateTags.add("f"); retrieved.setTags(updateTags); dao.update(retrieved.getId(), retrieved); retrieved = dao.findById(1001l); retrievedTags = retrieved.getTags(); assert(retrievedTags.size() == 2 && retrievedTags.contains("e") && retrievedTags.contains("f")); dao.expunge(1001l); } public void testListBy() { NetworkDaoImpl dao = ComponentLocator.inject(NetworkDaoImpl.class); dao.listBy(1l, 1l, 1l, "192.168.192.0/24"); } }