Skip to content

Commit b9a01ea

Browse files
author
Prathik Rao
committed
getlatestversion in catalog tests
1 parent 3395323 commit b9a01ea

1 file changed

Lines changed: 9 additions & 5 deletions

File tree

sdk/cs/test/FoundryLocal.Tests/ModelTests.cs

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ namespace Microsoft.AI.Foundry.Local.Tests;
1515
internal sealed class ModelTests
1616
{
1717
[Test]
18-
public async Task GetLastestVersion_Works()
18+
public async Task AddVariant_And_SelectVariant_Works()
1919
{
2020
var loadManager = new Mock<Detail.IModelLoadManager>();
2121
var coreInterop = new Mock<Detail.ICoreInterop>();
@@ -45,10 +45,14 @@ public async Task GetLastestVersion_Works()
4545
model.AddVariant(variant);
4646
}
4747

48-
var latestA = model.GetLatestVersion(variants[0]);
49-
await Assert.That(latestA).IsEqualTo(variants[0]);
48+
// Model should initially select the first variant
49+
await Assert.That(model.Id).IsEqualTo(variants[0].Id);
5050

51-
var latestB = model.GetLatestVersion(variants[2]);
52-
await Assert.That(latestB).IsEqualTo(variants[1]);
51+
// Selecting a different variant should update the model's active variant
52+
model.SelectVariant(variants[1]);
53+
await Assert.That(model.Id).IsEqualTo(variants[1].Id);
54+
55+
// All variants should be accessible
56+
await Assert.That(model.Variants.Count).IsEqualTo(3);
5357
}
5458
}

0 commit comments

Comments
 (0)