/***************************************************************************
* Copyright (c) 2012-2014 VMware, Inc. All Rights Reserved.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
***************************************************************************/
package com.vmware.bdd.service.resmgmt.impl;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import mockit.Expectations;
import mockit.Mocked;
import mockit.NonStrictExpectations;
import mockit.Verifications;
import org.testng.Assert;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;
import com.vmware.bdd.apitypes.DatastoreRead;
import com.vmware.bdd.apitypes.Datastore.DatastoreType;
import com.vmware.bdd.dal.IClusterDAO;
import com.vmware.bdd.dal.IDatastoreDAO;
import com.vmware.bdd.entity.VcDatastoreEntity;
import com.vmware.bdd.exception.VcProviderException;
import com.vmware.bdd.service.resmgmt.IResourceService;
public class DatastoreServiceTest extends BaseResourceTest {
@Mocked
private IDatastoreDAO dsDao;
@Mocked
private IResourceService resService;
@Mocked
private IClusterDAO clusterDao;
private DatastoreService datastoreSvc;
private List<VcDatastoreEntity> dss;
private final VcDatastoreEntity ds = new VcDatastoreEntity();
@BeforeClass
public void beforeClass() {
datastoreSvc = new DatastoreService();
dss = new ArrayList<VcDatastoreEntity>();
ds.setName("testDS");
ds.setType(DatastoreType.SHARED);
ds.setVcDatastore("datastore");
dss.add(ds);
}
@Test(groups = { "res-mgmt" })
public void testGetSharedDatastoresByNames() {
new Expectations() {
{
dsDao.findByNameAndType(DatastoreType.SHARED, anyString);
result = dss;
}
};
datastoreSvc.setDsDao(dsDao);
List<String> names = new ArrayList<String>();
names.add("ds01");
Set<String> patterns = datastoreSvc.getSharedDatastoresByNames(names);
Assert.assertTrue(patterns.size() == 1);
}
@Test(groups = { "res-mgmt" })
public void testGetLocalDatastoresByNames() {
new Expectations() {
{
dsDao.findByNameAndType(DatastoreType.LOCAL, anyString);
result = dss;
}
};
datastoreSvc.setDsDao(dsDao);
List<String> names = new ArrayList<String>();
names.add("ds01");
Set<String> patterns = datastoreSvc.getLocalDatastoresByNames(names);
Assert.assertTrue(patterns.size() == 1);
}
@Test(groups = { "res-mgmt" })
public void testAddDataStores() {
new NonStrictExpectations() {
{
dsDao.nameExisted(anyString);
result = false;
resService.isDatastoreExistInVC(anyString);
result = true;
}
};
datastoreSvc.setDsDao(dsDao);
datastoreSvc.setResService(resService);
List<String> dsSpec = new ArrayList<String>();
dsSpec.add("datastore");
datastoreSvc.addDatastores("testDS", DatastoreType.SHARED, dsSpec, false);
new Verifications() {
{
VcDatastoreEntity entity = new VcDatastoreEntity();
dsDao.insert(withAny(entity));
}
};
}
@Test(groups = { "res-mgmt" })
public void testDeleteDatastore() {
new NonStrictExpectations() {
{
dsDao.findByName(anyString);
result = dss;
Set<String> names = new HashSet<String>();
names.add("datastore1");
List<String> clusters = new ArrayList<String>();
clusterDao.findClustersByUsedDatastores(names);
result = clusters;
}
};
datastoreSvc.setDsDao(dsDao);
datastoreSvc.setClusterDao(clusterDao);
datastoreSvc.deleteDatastore("testDS");
new Verifications() {
{
dsDao.delete(withAny(ds));
}
};
}
@Test(groups = { "res-mgmt" }, expectedExceptions = VcProviderException.class)
public void testDeleteDatastoreWithReference() {
new NonStrictExpectations() {
{
dsDao.findByName(anyString);
result = dss;
Set<String> names = new HashSet<String>();
names.add("datastore");
List<String> clusters = new ArrayList<String>();
clusters.add("cluster1");
clusterDao.findClustersByUsedDatastores(names);
returns(clusters);
}
};
datastoreSvc.setDsDao(dsDao);
datastoreSvc.setClusterDao(clusterDao);
datastoreSvc.deleteDatastore("testDS");
}
@Test(groups = { "res-mgmt" })
public void testGetAllDataStoreName() {
new Expectations() {
{
dsDao.findAllSortByName();
result = dss;
}
};
datastoreSvc.setDsDao(dsDao);
Set<String> names = datastoreSvc.getAllDatastoreNames();
Assert.assertNotNull(names);
Assert.assertEquals(names.size(), 1);
}
@Test(groups = { "res-mgmt" })
public void testGetAllDatastoreReads() {
new Expectations() {
{
dsDao.findByName(anyString);
result = dss;
}
};
datastoreSvc.setDsDao(dsDao);
DatastoreRead ds = datastoreSvc.getDatastoreRead("testDS");
Assert.assertNotNull(ds);
}
}