public void Vm_getSizeCategory_shouldReturnEmptyString()
{
var result = VmSizeUtil.GetDisplayTextSizeForDropdown(null);
var expectedResult = "unknown";
Assert.Equal(expectedResult, result);
}
public void Vm_getSizeCategory()
{
var vmSize = new VmSize()
{
Key = "first", NumberOfCores = 4, MemoryGB = 3, OsDiskSizeInMB = 1000, MaxDataDiskCount = 4
};
var result = VmSizeUtil.GetDisplayTextSizeForDropdown(vmSize);
var expectedResult = "first (4 cores, 3 GB Memory, os disk: 1000, max data disks: 4)";
Assert.Equal(expectedResult, result);
}
public async Task UpdateVmSizeCache(CancellationToken cancellationToken = default)
{
var currentUser = await _userService.GetCurrentUserAsync();
var regionsFromDb = await _db.Regions.Include(r => r.VmSizeAssociations).ThenInclude(va => va.VmSize).Where(r => !r.Disabled).ToListAsync();
if (regionsFromDb == null || (regionsFromDb != null & regionsFromDb.Count() == 0))
{
throw new Exception($"Could not update Vm Size Cache, No regions found in DB");
}
foreach (var curRegionFromDb in regionsFromDb)
{
_logger.LogInformation($"Updating VM Size Cache for Region: {curRegionFromDb.Key}");
try
{
var resourceSkusFromAzure = await _azureResourceSkuService.GetSKUsForRegion(curRegionFromDb.Key, "virtualMachines", filterBasedOnResponseRestrictions : true, cancellationToken);
if (resourceSkusFromAzure == null || (resourceSkusFromAzure != null && resourceSkusFromAzure.Count() == 0))
{
throw new Exception($"No VM SKUs found in Azure for region {curRegionFromDb.Key}");
}
_logger.LogInformation($"Updating VM Size Cache for Region: {curRegionFromDb.Key}. Found {resourceSkusFromAzure.Count()} SKUs for region");
var existingSizeItemsForRegion = curRegionFromDb.VmSizeAssociations.ToDictionary(r => r.VmSize.Key, r => r.VmSize);
var validSkusFromAzure = new HashSet <string>();
VmSize curVmSizeInDb;
foreach (var curAzureSku in resourceSkusFromAzure)
{
if (existingSizeItemsForRegion.TryGetValue(curAzureSku.Name, out curVmSizeInDb))
{
curVmSizeInDb.Category = AzureVmUtil.GetSizeCategory(curAzureSku.Name);
if (ShouldBeExcluded(curVmSizeInDb))
{
var toRemoveFromDb = curRegionFromDb.VmSizeAssociations.FirstOrDefault(ra => ra.VmSizeKey == curVmSizeInDb.Key);
curRegionFromDb.VmSizeAssociations.Remove(toRemoveFromDb);
await _db.SaveChangesAsync();
continue;
}
else
{
PopulateVmSizeProps(curAzureSku, curVmSizeInDb);
curVmSizeInDb.DisplayText = VmSizeUtil.GetDisplayTextSizeForDropdown(curVmSizeInDb);
//Get updated price for VM Size and set on db entry
var regionAssociation = curVmSizeInDb.RegionAssociations.Where(ra => ra.RegionKey == curRegionFromDb.Key).SingleOrDefault();
regionAssociation.Price = await _azureCostManagementService.GetVmPrice(curRegionFromDb.Key, curVmSizeInDb.Key);
await _db.SaveChangesAsync();
validSkusFromAzure.Add(curVmSizeInDb.Key);
}
}
else
{
//Size item might exist in db for other region
curVmSizeInDb = await _db.VmSizes.FirstOrDefaultAsync(r => r.Key == curAzureSku.Name);
if (curVmSizeInDb == null)
{
curVmSizeInDb = new VmSize()
{
Key = curAzureSku.Name, CreatedBy = currentUser.UserName, Category = AzureVmUtil.GetSizeCategory(curAzureSku.Name)
};
}
else
{
curVmSizeInDb.Category = AzureVmUtil.GetSizeCategory(curAzureSku.Name);
}
if (ShouldBeExcluded(curVmSizeInDb))
{
//Dont want to include these
continue;
}
PopulateVmSizeProps(curAzureSku, curVmSizeInDb);
curVmSizeInDb.DisplayText = VmSizeUtil.GetDisplayTextSizeForDropdown(curVmSizeInDb);
//Add to lookup
existingSizeItemsForRegion.Add(curAzureSku.Name, curVmSizeInDb);
var priceOfVm = await _azureCostManagementService.GetVmPrice(curRegionFromDb.Key, curVmSizeInDb.Key);
//Add to DB
curRegionFromDb.VmSizeAssociations.Add(new RegionVmSize()
{
Region = curRegionFromDb, VmSize = curVmSizeInDb, Price = priceOfVm
});
await _db.SaveChangesAsync();
validSkusFromAzure.Add(curVmSizeInDb.Key);
}
}
//Delete those that are no longer present in Azure, or that does not pass the filter
foreach (var curDbSize in existingSizeItemsForRegion.Values)
{
if (!validSkusFromAzure.Contains(curDbSize.Key))
{
var toRemoveFromDb = curRegionFromDb.VmSizeAssociations.FirstOrDefault(ra => ra.VmSizeKey == curDbSize.Key);
if (toRemoveFromDb != null)
{
curRegionFromDb.VmSizeAssociations.Remove(toRemoveFromDb);
}
}
}
await _db.SaveChangesAsync();
_logger.LogInformation($"Done updating VM Size Cache for Region: {curRegionFromDb.Name}");
}
catch (Exception ex)
{
_logger.LogError($"Update VM Size cache: Unable to update Size cache for Region {curRegionFromDb.Key}. Se inner exception for details", ex);
continue;
}
}
_logger.LogInformation($"Deleting Vm Size entries not associated with any region");
foreach (var curVmSize in await _db.VmSizes.Include(s => s.RegionAssociations).ToListAsync())
{
if (curVmSize.RegionAssociations == null || (curVmSize.RegionAssociations != null && curVmSize.RegionAssociations.Count == 0))
{
_db.VmSizes.Remove(curVmSize);
}
}
await _db.SaveChangesAsync();
//TODO: Delete size records with no associated region?
_logger.LogInformation($"Done updating VM Size Cache");
}
