private async Task <IMembershipTable> GetMembershipTable(GlobalConfiguration.LivenessProviderType membershipType)
{
string runId = Guid.NewGuid().ToString("N");
var config = new GlobalConfiguration {
LivenessType = membershipType, DeploymentId = runId
};
IMembershipTable membership;
switch (membershipType)
{
case GlobalConfiguration.LivenessProviderType.AzureTable:
config.DataConnectionString = StorageTestConstants.DataConnectionString;
membership = await AzureBasedMembershipTable.GetMembershipTable(config, true);
break;
case GlobalConfiguration.LivenessProviderType.SqlServer:
config.DataConnectionString = StorageTestConstants.GetSqlConnectionString(TestContext.DeploymentDirectory);
membership = await SqlMembershipTable.GetMembershipTable(config, true);
break;
default:
throw new NotImplementedException(membershipType.ToString());
}
return(membership);
}
public override TestCluster CreateTestCluster()
{
var options = new TestClusterOptions(2);
options.ClusterConfiguration.Globals.DataConnectionString = StorageTestConstants.GetZooKeeperConnectionString();
options.ClusterConfiguration.Globals.LivenessType = GlobalConfiguration.LivenessProviderType.ZooKeeper;
return(new TestCluster(options));
}
public static void ClassInitialize(TestContext testContext)
{
hostName = Dns.GetHostName();
TraceLogger.GetLogger("MembershipTablePluginTests", TraceLogger.LoggerType.Application);
var cfg = new ClusterConfiguration();
cfg.LoadFromFile("OrleansConfigurationForUnitTests.xml");
TraceLogger.Initialize(cfg.GetConfigurationForNode("Primary"));
StorageTestConstants.Init();
TraceLogger.AddTraceLevelOverride("AzureTableDataManager", Logger.Severity.Verbose3);
TraceLogger.AddTraceLevelOverride("OrleansSiloInstanceManager", Logger.Severity.Verbose3);
TraceLogger.AddTraceLevelOverride("Storage", Logger.Severity.Verbose3);
}
private async Task Initialize()
{
deploymentId = "test-" + Guid.NewGuid();
int generation = SiloAddress.AllocateNewGeneration();
siloAddress = SiloAddress.NewLocalAddress(generation);
logger.Info("DeploymentId={0} Generation={1}", deploymentId, generation);
GlobalConfiguration config = new GlobalConfiguration
{
DeploymentId = deploymentId,
DataConnectionString = StorageTestConstants.GetZooKeeperConnectionString()
};
var mbr = AssemblyLoader.LoadAndCreateInstance <IMembershipTable>(Constants.ORLEANS_ZOOKEEPER_UTILS_DLL, logger);
await mbr.InitializeMembershipTable(config, true, logger).WithTimeout(timeout);
membership = mbr;
}
protected override string GetConnectionString()
{
return(StorageTestConstants.GetZooKeeperConnectionString());
}
public static void ClassInitialize(TestContext testContext)
{
StorageTestConstants.Init();
}
