internal InsertInstanceEvent(LogRecord logRecord) : base(logRecord)
{
Debug.Assert(IsInsertInstanceEvent(logRecord));
var request = logRecord.ProtoPayload.Request;
if (request != null)
{
var disks = request["disks"];
if (disks != null)
{
foreach (var disk in ((JArray)disks))
{
if (disk["boot"] != null && (bool)disk["boot"])
{
var initializeParams = disk["initializeParams"];
if (initializeParams != null)
{
this.Image = GlobalResourceReference.FromString(
initializeParams["sourceImage"].Value <string>());
}
}
}
}
}
}
public void WhenCreatedFromPath_ThenToStringReturnsPath()
{
var path = "projects/project-1/global/images/image-1";
Assert.AreEqual(
path,
GlobalResourceReference.FromString(path).ToString());
}
public void WhenPathInvalid_FromStringThrowsArgumentException()
{
Assert.Throws <ArgumentException>(() => GlobalResourceReference.FromString(
"projects/project-1/notglobal/images/image-1"));
Assert.Throws <ArgumentException>(() => GlobalResourceReference.FromString(
"/project-1/global/images/image-1"));
Assert.Throws <ArgumentException>(() => GlobalResourceReference.FromString(
"/"));
}
public void WhenQualifiedByGoogleapisHost_FromStringReturnsObject()
{
var ref1 = GlobalResourceReference.FromString(
"https://www.googleapis.com/compute/v1/projects/windows-cloud/global/licenses/windows-server-core");
Assert.AreEqual("licenses", ref1.ResourceType);
Assert.AreEqual("windows-server-core", ref1.ResourceName);
Assert.AreEqual("windows-cloud", ref1.ProjectId);
}
public void WhenQualifiedByComputeGoogleapisHost_FromStringReturnsObject()
{
var ref1 = GlobalResourceReference.FromString(
"https://compute.googleapis.com/compute/v1/projects/debian-cloud/global/images/family/debian-9");
Assert.AreEqual("images", ref1.ResourceType);
Assert.AreEqual("family/debian-9", ref1.ResourceName);
Assert.AreEqual("debian-cloud", ref1.ProjectId);
}
public void WhenResourceNameCotainsSlash_FromStringReturnsObject()
{
var ref1 = GlobalResourceReference.FromString(
"projects/debian-cloud/global/images/family/debian-9");
Assert.AreEqual("images", ref1.ResourceType);
Assert.AreEqual("family/debian-9", ref1.ResourceName);
Assert.AreEqual("debian-cloud", ref1.ProjectId);
}
public void WhenPathIsValid_FromStringReturnsObject()
{
var ref1 = GlobalResourceReference.FromString(
"projects/project-1/global/images/image-1");
Assert.AreEqual("images", ref1.ResourceType);
Assert.AreEqual("image-1", ref1.ResourceName);
Assert.AreEqual("project-1", ref1.ProjectId);
}
public void WhenCreatedFromUrl_ThenToStringReturnsPath()
{
var path = "projects/project-1/global/images/image-1";
Assert.AreEqual(
path,
GlobalResourceReference.FromString(
"https://www.googleapis.com/compute/v1/" + path).ToString());
}
public void WhenReadingSample3_ThenHistoryIsRestored()
{
var set = BuildHistoryFromResource("instance-3.json");
Assert.AreEqual(1, set.Instances.Count());
var instance = set.Instances.First();
Assert.AreEqual(
GlobalResourceReference.FromString("projects/project-1/global/images/windows-server"),
instance.Image);
Assert.AreEqual(2, instance.Placements.Count());
var firstPlacement = instance.Placements.First();
Assert.AreEqual("413db7b32a208e7ccb4ee62acedee725", firstPlacement.ServerId);
Assert.AreEqual(Tenancy.SoleTenant, firstPlacement.Tenancy);
// Insert..
Assert.AreEqual(DateTime.Parse("2020-05-05T08:31:40.864Z").ToUniversalTime(), firstPlacement.From);
// ..till TerminateOnHostMaintenance.
Assert.AreEqual(DateTime.Parse("2020-05-06T16:10:46.781Z").ToUniversalTime(), firstPlacement.To);
var secondPlacement = instance.Placements.Last();
Assert.AreEqual("413db7b32a208e7ccb4ee62acedee725", secondPlacement.ServerId);
Assert.AreEqual(Tenancy.SoleTenant, secondPlacement.Tenancy);
// Insert..
Assert.AreEqual(DateTime.Parse("2020-05-06T16:36:01.441Z").ToUniversalTime(), secondPlacement.From);
// ..till GuestTerminate.
Assert.AreEqual(DateTime.Parse("2020-05-06T17:39:34.635Z").ToUniversalTime(), secondPlacement.To);
}
public void WhenReadingSample2_ThenHistoryIsRestored()
{
var set = BuildHistoryFromResource("instance-2.json");
Assert.AreEqual(1, set.Instances.Count());
var instance = set.Instances.First();
Assert.AreEqual(
GlobalResourceReference.FromString("projects/windows-cloud/global/images/windows-server"),
instance.Image);
Assert.AreEqual(1, instance.Placements.Count());
var placement = instance.Placements.First();
Assert.AreEqual("15934ff9aee7d8c5719fad1053b7fc7d", placement.ServerId);
Assert.AreEqual(Tenancy.SoleTenant, placement.Tenancy);
// Insert..
Assert.AreEqual(DateTime.Parse("2020-05-06T14:57:46.557Z").ToUniversalTime(), placement.From);
// ..till GuestTerminate.
Assert.AreEqual(DateTime.Parse("2020-05-06T16:03:06.484Z").ToUniversalTime(), placement.To);
}
public async Task AddExistingInstances(
InstancesResource instancesResource,
DisksResource disksResource,
string projectId)
{
// NB. Instances.list returns the disks associated with each
// instance, but lacks the information about the source image.
// Therefore, load disks separately.
using (TraceSources.LogAnalysis.TraceMethod().WithParameters(projectId))
{
// TODO: Paging
var disks = await disksResource.AggregatedList(projectId).ExecuteAsync();
var sourceImagaesByDisk = disks.Items == null
? new Dictionary <string, string>()
: disks.Items.Values
.Where(v => v.Disks != null)
.EnsureNotNull()
.SelectMany(v => v.Disks)
.EnsureNotNull()
.ToDictionary(d => d.SelfLink, d => d.SourceImage);
// TODO: Paging
var instances = await instancesResource.AggregatedList(projectId).ExecuteAsync();
if (instances.Items != null)
{
foreach (var list in instances.Items.Values)
{
if (list.Instances == null)
{
continue;
}
foreach (var instance in list.Instances)
{
TraceSources.LogAnalysis.TraceVerbose("Adding {0}", instance.Id);
var bootDiskUrl = instance.Disks
.EnsureNotNull()
.Where(d => d.Boot != null && d.Boot.Value)
.EnsureNotNull()
.Select(d => d.Source)
.EnsureNotNull()
.FirstOrDefault();
GlobalResourceReference image = null;
if (bootDiskUrl != null &&
sourceImagaesByDisk.TryGetValue(bootDiskUrl, out string imageUrl) &&
imageUrl != null)
{
image = GlobalResourceReference.FromString(imageUrl);
}
AddExistingInstance(
(ulong)instance.Id.Value,
new VmInstanceReference(
projectId,
ShortZoneIdFromUrl(instance.Zone),
instance.Name),
image,
instance.Status == "RUNNING"
? InstanceState.Running
: InstanceState.Terminated,
DateTime.Now,
instance.Scheduling.NodeAffinities != null && instance.Scheduling.NodeAffinities.Any()
? Tenancy.SoleTenant
: Tenancy.Fleet);
}
}
}
}
}
public void WhenPathLacksProject_FromStringThrowsArgumentException()
{
Assert.Throws <ArgumentException>(() => GlobalResourceReference.FromString(
"/project-1/project-1/global/images/image-1"));
}
