public IHttpActionResult LimitMemory(string handle, MemoryLimits limits)
{
var container = containerService.GetContainerByHandle(handle);
if (container == null)
{
return(NotFound());
}
if (limits.LimitInBytes != null)
{
container.LimitMemory(limits.LimitInBytes.Value);
}
return(Ok());
}
public Limits()
{
CpuLimits = new CpuLimits();
MemoryLimits = new MemoryLimits();
DiskLimits = new DiskLimits();
}
private void describe_()
{
Mock<IContainerService> mockContainerService = null;
LimitsController LimitsController = null;
string handle = null;
Mock<IContainer> mockContainer = null;
before = () =>
{
mockContainerService = new Mock<IContainerService>();
LimitsController = new LimitsController(mockContainerService.Object);
handle = Guid.NewGuid().ToString();
mockContainer = new Mock<IContainer>();
mockContainerService.Setup(x => x.GetContainerByHandle(handle)).Returns(mockContainer.Object);
};
describe["#LimitMemory"] = () =>
{
const ulong limitInBytes = 876;
MemoryLimits limits = null;
IHttpActionResult result = null;
before = () =>
{
limits = new MemoryLimits { LimitInBytes = limitInBytes };
};
act = () =>
{
result = LimitsController.LimitMemory(handle, limits);
};
it["sets limits on the container"] = () =>
{
mockContainer.Verify(x => x.LimitMemory(limitInBytes));
};
context["when the container does not exist"] = () =>
{
before = () =>
{
mockContainerService.Setup(x => x.GetContainerByHandle(It.IsAny<string>())).Returns(null as IContainer);
};
it["Returns not found"] = () =>
{
result.should_cast_to<NotFoundResult>();
};
};
};
describe["#CurrentMemoryLimit"] = () =>
{
it["returns the current limit on the container"] = () =>
{
mockContainer.Setup(x => x.CurrentMemoryLimit()).Returns(3072);
var result = LimitsController.CurrentMemoryLimit(handle);
var jsonResult = result.should_cast_to<JsonResult<MemoryLimits>>();
jsonResult.Content.LimitInBytes.should_be(3072);
};
context["when the container does not exist"] = () =>
{
before = () =>
{
mockContainerService.Setup(x => x.GetContainerByHandle(It.IsAny<string>())).Returns(null as IContainer);
};
it["Returns not found"] = () =>
{
var result = LimitsController.CurrentMemoryLimit(handle);
result.should_cast_to<NotFoundResult>();
};
};
};
describe["#LimitCpu"] = () =>
{
IHttpActionResult result = null;
const int weight = 5;
act = () =>
{
var limits = new CpuLimits {Weight = weight};
result = LimitsController.LimitCpu(handle, limits);
};
it["sets limits on the container"] = () =>
{
mockContainer.Verify(x => x.LimitCpu(weight));
};
context["when the container does not exist"] = () =>
{
before = () =>
{
mockContainerService.Setup(x => x.GetContainerByHandle(It.IsAny<string>())).Returns(null as IContainer);
};
it["Returns not found"] = () =>
{
result.should_cast_to<NotFoundResult>();
};
};
};
describe["#CurrentCpuLimit"] = () =>
{
it["returns the current limit on the container"] = () =>
{
mockContainer.Setup(x => x.CurrentCpuLimit()).Returns(6);
var result = LimitsController.CurrentCpuLimit(handle);
var jsonResult = result.should_cast_to<JsonResult<CpuLimits>>();
jsonResult.Content.Weight.should_be(6);
};
context["when the container does not exist"] = () =>
{
before = () =>
{
mockContainerService.Setup(x => x.GetContainerByHandle(It.IsAny<string>())).Returns(null as IContainer);
};
it["Returns not found"] = () =>
{
var result = LimitsController.CurrentCpuLimit(handle);
result.should_cast_to<NotFoundResult>();
};
};
};
describe["#LimitDisk"] = () =>
{
IHttpActionResult result = null;
const ulong bytes = 5;
act = () =>
{
var limits = new DiskLimits { ByteHard = bytes };
result = LimitsController.LimitDisk(handle, limits);
};
it["sets limits on the container"] = () =>
{
mockContainer.Verify(x => x.LimitDisk(bytes));
};
context["when the container does not exist"] = () =>
{
before = () =>
{
mockContainerService.Setup(x => x.GetContainerByHandle(It.IsAny<string>())).Returns(null as IContainer);
};
it["Returns not found"] = () =>
{
result.should_cast_to<NotFoundResult>();
};
};
};
describe["#CurrentDiskLimit"] = () =>
{
it["returns the current limit on the container"] = () =>
{
mockContainer.Setup(x => x.CurrentDiskLimit()).Returns(6);
var result = LimitsController.CurrentDiskLimit(handle);
var jsonResult = result.should_cast_to<JsonResult<DiskLimits>>();
jsonResult.Content.ByteHard.should_be(6);
};
context["when the container does not exist"] = () =>
{
before = () =>
{
mockContainerService.Setup(x => x.GetContainerByHandle(It.IsAny<string>()))
.Returns(null as IContainer);
};
it["Returns not found"] = () =>
{
var result = LimitsController.CurrentDiskLimit(handle);
result.should_cast_to<NotFoundResult>();
};
};
};
}
public IHttpActionResult LimitMemory(string handle, MemoryLimits limits)
{
var container = containerService.GetContainerByHandle(handle);
if (container == null)
{
return NotFound();
}
if (limits.LimitInBytes != null) container.LimitMemory(limits.LimitInBytes.Value);
return Ok();
}