コード例 #1
0
        private static void RewriteWebConfig(AutoRemovableDirectory tempFiles, ConfigReplacements replacements)
        {
            var webConfig      = tempFiles.ReadFile("Web.config");
            var configRewriter = new ConfigRewriter(new ConfigSources(new Dictionary <string, string>()));

            var rewrittenWebConfig = configRewriter.Rewrite(webConfig, replacements);

            tempFiles.WriteFile("Web.config", rewrittenWebConfig);
        }
コード例 #2
0
        public void HeartbeatService_Basic()
        {
            // Verify that heartbeat service works by modifying its configuration file
            // to monitor two simulated services and then verify that the monitoring
            // works.

            HttpServer     httpServer = null;
            Process        svcProcess = null;
            ConfigRewriter rewriter;
            Assembly       assembly;
            string         iniPath;

            assembly = typeof(LillTek.Datacenter.HeartbeatService.Program).Assembly;
            iniPath  = Config.GetConfigPath(assembly);
            rewriter = new ConfigRewriter(iniPath);

            try
            {
                // Initialize the local HTTP server that will simulate the monitored services.

                httpServer = new HttpServer(new IPEndPoint[] { new IPEndPoint(IPAddress.Any, 666) }, new IHttpModule[] { this }, 10, 10, 8196);
                httpServer.Start();

                host2Health = new Dictionary <string, bool>(StringComparer.OrdinalIgnoreCase);
                host2Health["testservice01.lilltek.com"] = true;
                host2Health["testservice02.lilltek.com"] = true;
                host2Health["testservice03.lilltek.com"] = true;

                // Rewrite the config file to have it start in hub mode
                // and then start the router service as a form application.

                rewriter.Rewrite(new ConfigRewriteTag[0]);

                // $hack(jeff.lill):
                //
                // Append test related settings to the configuration file.  Since these settings
                // appear after the defaults, the written settings will override.

                using (var file = File.AppendText(iniPath))
                {
                    const string prefix = "LillTek.Datacenter.HeartbeatService.";

                    file.WriteLine(prefix + "MonitorUri[0] = http://testservice01.lilltek.com:666/Heartbeat.aspx?global=0");
                    file.WriteLine(prefix + "MonitorUri[1] = http://testservice02.lilltek.com:666/Heartbeat.aspx?global=0");
                    file.WriteLine(prefix + "MonitorUri[2] = http://testservice03.lilltek.com:666/Heartbeat.aspx?global=0");
                    file.WriteLine(prefix + "BkInterval = 1s");
                    file.WriteLine(prefix + "PollInterval = 1s");
                }

                svcProcess = Helper.StartProcess(assembly, "-mode:form -start");
                Thread.Sleep(10000);    // Give the process a chance to spin up

                // Give the service a chance to ping the test a few times and then
                // make a request to the heartbeat handler to verify that it indicates
                // that the server is healthy.

                Thread.Sleep(3000);
                Assert.IsTrue(GetHealth());

                // Now indicate that the first target service is unhealthy, give the heartbeat
                // handler a chance to ping the service, and verify that it determined
                // the server is unhealthy.

                host2Health["testservice01.lilltek.com"] = false;
                Thread.Sleep(3000);
                Assert.IsFalse(GetHealth());

                // Now indicate that the first target service is healthy, give the heartbeat
                // handler a chance to ping the service, and verify that it determined
                // the server is healthy.

                host2Health["testservice01.lilltek.com"] = true;
                Thread.Sleep(3000);
                Assert.IsTrue(GetHealth());

                // Now indicate that the  second and third target services are unhealthy, give the heartbeat
                // handler a chance to ping the service, and verify that it determined
                // the server is unhealthy.

                host2Health["testservice02.lilltek.com"] = false;
                host2Health["testservice03.lilltek.com"] = false;
                Thread.Sleep(3000);
                Assert.IsFalse(GetHealth());

                // Now mark the second service as healthy and verify that the
                // server is still unhealthy (due to the third service).

                host2Health["testservice02.lilltek.com"] = true;
                Thread.Sleep(3000);
                Assert.IsFalse(GetHealth());

                // Now mark all services as healthy and verify that
                // the server is considered healthy.

                host2Health["testservice03.lilltek.com"] = true;
                Thread.Sleep(3000);
                Assert.IsTrue(GetHealth());

                // Stop the simulated target HTTP server and verify that the handler
                // detects the failure.

                httpServer.Stop();
                httpServer = null;
                Thread.Sleep(3000);
                Assert.IsFalse(GetHealth());
            }
            finally
            {
                rewriter.Restore();

                if (httpServer != null)
                {
                    httpServer.Stop();
                    httpServer = null;
                }

                if (svcProcess != null)
                {
                    svcProcess.Kill();
                    svcProcess.Close();
                }
            }
        }
コード例 #3
0
        public void ConfigRewriter_Test()
        {
            string         fname    = Path.GetTempPath() + "ConfigRewriteTest.ini";
            ConfigRewriter rewriter = new ConfigRewriter(fname);
            StreamReader   reader   = null;
            StreamWriter   writer   = null;

            string orgText =
                @"
// This is a test
foo=bar
// $replace(tag1)
   -- $REPLACE(TAG2)
bar=foo
  // $(
";
            string newText =
                @"
// This is a test
foo=bar
key1=10
key2=20
bar=foo
  // $(
";

            try
            {
                writer = new StreamWriter(fname);
                writer.Write(orgText);
                writer.Close();
                writer = null;

                rewriter.Restore();     // This shouldn't case a problem

                rewriter.Rewrite(new ConfigRewriteTag[] { new ConfigRewriteTag("tag1", "key1=10"), new ConfigRewriteTag("tag2", "key2=20\r\n") });

                reader = new StreamReader(fname);
                Assert.AreEqual(newText, reader.ReadToEnd());
                reader.Close();
                reader = null;

                rewriter.Restore();

                reader = new StreamReader(fname);
                Assert.AreEqual(orgText, reader.ReadToEnd());
                reader.Close();
                reader = null;

                rewriter.Restore();     // This shouldn't case a problem
            }
            finally
            {
                if (reader != null)
                {
                    reader.Close();
                }

                if (writer != null)
                {
                    writer.Close();
                }

                if (File.Exists(fname))
                {
                    File.Delete(fname);
                }
            }
        }
コード例 #4
0
        public void RouterService_RootMode()
        {
            // Verify that the router service actually deploys a root router
            // by starting an instance of the application and a couple of
            // hub routers and making sure that the service is routing messages
            // between the two hubs.

            Process        svcProcess = null;
            HubRouter      hub1       = null;
            HubRouter      hub2       = null;
            ConfigRewriter rewriter;
            Assembly       assembly;
            string         iniPath;

            assembly = typeof(LillTek.Datacenter.RouterService.Program).Assembly;
            iniPath  = Config.GetConfigPath(assembly);
            rewriter = new ConfigRewriter(iniPath);

            try
            {
                // Rewrite the config file to have it start in root mode
                // and then start the router service as a form application.

                rewriter.Rewrite(new ConfigRewriteTag[] { new ConfigRewriteTag("Mode", "#undef HUBMODE\r\n#define ROOTMODE\r\n#define HUBNAME hub\r\n") });
                svcProcess = Helper.StartProcess(assembly, "-mode:form -start");
                Thread.Sleep(10000);    // Give the process a chance to spin up

                // Crank up a couple of hub routers and send a message from
                // one to the other.

                hub1 = CreateHub(Helper.MachineName + ":" + Const.DCRootPort.ToString(), "hub1", group1);
                hub2 = CreateHub(Helper.MachineName + ":" + Const.DCRootPort.ToString(), "hub2", group2);
                Thread.Sleep(2000);

                recvMsg = null;
                hub1.SendTo("logical://hub2", new PropertyMsg());
                Thread.Sleep(1000);

                Assert.IsNotNull(recvMsg);
            }
            finally
            {
                rewriter.Restore();

                if (svcProcess != null)
                {
                    svcProcess.Kill();
                    svcProcess.Close();
                }

                if (hub1 != null)
                {
                    hub1.Stop();
                }

                if (hub2 != null)
                {
                    hub2.Stop();
                }
            }
        }
コード例 #5
0
        public void RouterService_HubMode()
        {
            // Verify that the router service actually deploys a hub router
            // by starting an instance of the application and a couple of
            // non-P2P leaf routers and making sure that the service is
            // routing messages between the two leaves.

            Process        svcProcess = null;
            LeafRouter     leaf1      = null;
            LeafRouter     leaf2      = null;
            ConfigRewriter rewriter;
            Assembly       assembly;
            string         iniPath;

            assembly = typeof(LillTek.Datacenter.RouterService.Program).Assembly;
            iniPath  = Config.GetConfigPath(assembly);
            rewriter = new ConfigRewriter(iniPath);

            try
            {
                // Rewrite the config file to have it start in hub mode
                // and then start the router service as a form application.

                rewriter.Rewrite(new ConfigRewriteTag[] { new ConfigRewriteTag("Mode", "#define HUBMODE\r\n#undef ROOTMODE\r\n#define HUBNAME hub\r\n") });
                svcProcess = Helper.StartProcess(assembly, "-mode:form -start");
                Thread.Sleep(10000);    // Give the process a chance to spin up

                // Crank up a couple of leaf routers and send a message from
                // one to the other.

                leaf1 = CreateLeaf("detached", Const.DCDefHubName, "leaf1", Const.DCCloudEP.ToString(), false);
                leaf2 = CreateLeaf("detached", Const.DCDefHubName, "leaf2", Const.DCCloudEP.ToString(), false);
                Thread.Sleep(2000);

                recvMsg = null;
                leaf1.SendTo("logical://leaf2", new PropertyMsg());
                Thread.Sleep(1000);

                Assert.IsNotNull(recvMsg);
            }
            finally
            {
                rewriter.Restore();

                if (svcProcess != null)
                {
                    svcProcess.Kill();
                    svcProcess.Close();
                }

                if (leaf1 != null)
                {
                    leaf1.Stop();
                }

                if (leaf2 != null)
                {
                    leaf2.Stop();
                }
            }
        }