public void TestClientsForProject()
{
// This test will ensure that when subscription is added, only events for that subscription get sent to it
var state = new InMemoryAssetStatusState() as IAssetStatusState;
Assert.NotNull(state);
const string clientId1 = "AF284FC0-4D38-4337-A782-564ABD65D5CC";
const string clientId2 = "C670850B-BF18-4AA3-963F-9820C47481C8";
var projectUid1 = Guid.Parse("59436FBE-909F-4ED9-A038-9403CAA5782D");
var projectUid2 = Guid.Parse("860706B6-E1B9-4E33-B74A-B2691430A0D7");
var customerUid1 = Guid.Parse("67543B02-FF2A-4AE4-96DF-826BD29E3942");
var customerUid2 = Guid.Parse("E59522CD-C0E3-4475-AD16-9C0C3035932B");
// First test the same customer uid, different project uid
state.AddSubscription(clientId1, new AssetUpdateSubscriptionModel()
{
CustomerUid = customerUid1,
ProjectUid = projectUid1
});
state.AddSubscription(clientId2, new AssetUpdateSubscriptionModel()
{
CustomerUid = customerUid1,
ProjectUid = projectUid2
});
// Now we should only get one clientid back
{
var clients = state.GetClientsForProject(customerUid1, projectUid1).Result;
Assert.True(clients.Count == 1, $"Expected a single client but got ${clients.Count}");
Assert.Equal(clientId1, clients[0]);
}
// Now same project UID but different customer UID (also this should replace the existing subscription)
state.AddSubscription(clientId1, new AssetUpdateSubscriptionModel()
{
CustomerUid = customerUid1,
ProjectUid = projectUid2
});
state.AddSubscription(clientId2, new AssetUpdateSubscriptionModel()
{
CustomerUid = customerUid2,
ProjectUid = projectUid2
});
// We should have 2 subscriptions, but only one for the project / customer
{
var subs = state.GetSubscriptions().Result;
var clients = state.GetClientsForProject(customerUid2, projectUid2).Result;
Assert.True(subs.Count == 2, $"Subs don't match, expected two got ${subs.Count}");
Assert.True(clients.Count == 1, $"Expected a single client but got ${clients.Count}");
Assert.Equal(clientId2, clients[0]);
}
// Now update the subscriptions to be the same
state.AddSubscription(clientId1, new AssetUpdateSubscriptionModel()
{
CustomerUid = customerUid1,
ProjectUid = projectUid1
});
state.AddSubscription(clientId2, new AssetUpdateSubscriptionModel()
{
CustomerUid = customerUid1,
ProjectUid = projectUid1
});
// We should have 2 subscriptions, both for the same proejct / customer
{
var subs = state.GetSubscriptions().Result;
var clients = state.GetClientsForProject(customerUid1, projectUid1).Result;
Assert.True(subs.Count == 2, $"Subs don't match, expected two got ${subs.Count}");
Assert.True(clients.Count == 2, $"Expected two clients but got ${clients.Count}");
Assert.Contains(clientId1, clients);
Assert.Contains(clientId2, clients);
}
// Now test a combo that returns no results
{
var clients = state.GetClientsForProject(customerUid2, projectUid2).Result;
Assert.True(clients.Count == 0, $"Expected no clients but got ${clients.Count}");
}
}
public void TestAddAndRemove()
{
var state = new InMemoryAssetStatusState() as IAssetStatusState;
Assert.NotNull(state);
{
var count = state.GetSubscriptions().Result.Count;
Assert.True(count == 0, $"Expected a zero count with a new state, got ${count} instead");
}
const string clientId1 = "307CA37C-2422-4AD9-A512-34DFB3BC7507";
const string clientId2 = "B0CCDF38-AA7E-4BCB-8692-53C898D6A469";
var model1 = new AssetUpdateSubscriptionModel()
{
ProjectUid = Guid.Parse("2A131E56-880E-44E8-901F-86D3CD3CBDBB")
};
var model2 = new AssetUpdateSubscriptionModel()
{
ProjectUid = Guid.Parse("D5C5A0FA-2480-4ACB-BDCE-0CDF10998BF8")
};
var model3 = new AssetUpdateSubscriptionModel()
{
ProjectUid = Guid.Parse("BDE14152-1B1B-4809-85B2-384DA1D91B43")
};
// Add the first model
state.AddSubscription(clientId1, model1);
// count should be 1
{
var items = state.GetSubscriptions().Result;
Assert.True(items.Count == 1, $"Added a model, count should be one but it is ${items.Count}");
Assert.Equal(model1, items[0]);
}
// Add the second model, with the same identifier (this should replace the existing one)
state.AddSubscription(clientId1, model2);
// count should be 1
{
var items = state.GetSubscriptions().Result;
Assert.True(items.Count == 1, $"Added a model with the same identifier, count should be one but it is ${items.Count}");
Assert.Equal(model2, items[0]);
}
// Add another model, with another identifier
state.AddSubscription(clientId2, model3);
// Count should be 2
{
var items = state.GetSubscriptions().Result;
Assert.True(items.Count == 2, $"Added a model with a new identifier, count should be two but it is ${items.Count}");
Assert.Contains(model2, items);
Assert.Contains(model3, items);
}
// Remove the first subscription and an unknown subscription
state.RemoveSubscription(clientId1);
state.RemoveSubscription("does not exist id");
// Count should be 1
{
var items = state.GetSubscriptions().Result;
Assert.True(items.Count == 1, $"Removed a subscription, count should be one but it is ${items.Count}");
Assert.Contains(model3, items);
}
state.RemoveSubscription(clientId2);
// should be empty
{
var count = state.GetSubscriptions().Result.Count;
Assert.True(count == 0, $"Expected a zero count with all clients removed, got ${count} instead");
}
}
