예제 #1
0
        public void Should_LogFiveTimes_When_ThereAreMoreThanFiveErrorsOnStartupMessageSend()
        {
            _listener = new TestTraceListener();
            Trace.Listeners.Add(_listener);
            Cassandra.Diagnostics.CassandraTraceSwitch.Level = TraceLevel.Verbose;

            var cluster = GetCluster(false);
            var session = GetSession(cluster);

            using (var target = InsightsClientTests.GetInsightsClient(cluster, session))
            {
                Expression <Func <IControlConnection, Task <Response> > > mockExpression =
                    cc => cc.UnsafeSendQueryRequestAsync(
                        "CALL InsightsRpc.reportInsight(?)",
                        It.IsAny <QueryProtocolOptions>());
                Mock.Get(cluster.Metadata.ControlConnection).Setup(mockExpression).ReturnsAsync((Response)null);

                target.Init();

                TestHelper.RetryAssert(
                    () =>
                {
                    Mock.Get(cluster.Metadata.ControlConnection).Verify(mockExpression, Times.AtLeast(10));
                },
                    30);

                Trace.Flush();
                Assert.AreEqual(5, _listener.Queue.Count, string.Join(" ; ", _listener.Queue.ToArray()));
                var messages = _listener.Queue.ToArray();
                Assert.AreEqual(messages.Count(m => m.Contains("Could not send insights startup event. Exception:")), 5);
            }
        }
예제 #2
0
        public void Should_InvokeRpcCallCorrectlyAndPeriodically_When_SendStatusMessageIsInvoked()
        {
            const string expectedJson =
                "{\"metadata\":{" +
                "\"name\":\"driver.status\"," +
                "\"timestamp\":124219041," +
                "\"tags\":{\"language\":\"csharp\"}," +
                "\"insightType\":\"EVENT\"," +
                "\"insightMappingId\":\"v1\"}," +
                "\"data\":{" +
                "\"clientId\":\"becfe098-e462-47e7-b6a7-a21cd316d4c0\"," +
                "\"sessionId\":\"e21eab96-d91e-4790-80bd-1d5fb5472258\"," +
                "\"controlConnection\":\"10.10.10.10:9011\"," +
                "\"connectedNodes\":{" +
                "\"127.0.0.1:9042\":{\"connections\":3,\"inFlightQueries\":1}," +
                "\"127.0.0.2:9042\":{\"connections\":4,\"inFlightQueries\":2}}}}";
            var cluster = GetCluster(false);
            var session = GetSession(cluster);

            using (var target = InsightsClientTests.GetInsightsClient(cluster, session))
            {
                var queryProtocolOptions = new ConcurrentQueue <QueryProtocolOptions>();
                Mock.Get(cluster.Metadata.ControlConnection).Setup(cc => cc.UnsafeSendQueryRequestAsync(
                                                                       "CALL InsightsRpc.reportInsight(?)",
                                                                       It.IsAny <QueryProtocolOptions>()))
                .ReturnsAsync(new FakeResultResponse(ResultResponse.ResultResponseKind.Void))
                .Callback <string, QueryProtocolOptions>((query, opts) => { queryProtocolOptions.Enqueue(opts); });

                target.Init();

                TestHelper.RetryAssert(() => { Assert.GreaterOrEqual(queryProtocolOptions.Count, 5); }, 5, 400);
                queryProtocolOptions.TryDequeue(out var result); // ignore startup message
                queryProtocolOptions.TryPeek(out result);
                Assert.AreEqual(expectedJson, result.Values[0], "Actual: " + Environment.NewLine + result.Values[0]);
            }
        }
예제 #3
0
        public void Should_InvokeRpcCallCorrectlyAndImmediatelyWithExecutionProfiles_When_SendStartupMessageIsInvoked()
        {
            const string expectedJson =
                "{\"metadata\":{" +
                "\"name\":\"driver.startup\"," +
                "\"timestamp\":124219041," +
                "\"tags\":{\"language\":\"csharp\"}," +
                "\"insightType\":\"EVENT\"," +
                "\"insightMappingId\":\"v1\"}," +
                "\"data\":{" +
                "\"clientId\":\"becfe098-e462-47e7-b6a7-a21cd316d4c0\"," +
                "\"sessionId\":\"e21eab96-d91e-4790-80bd-1d5fb5472258\"," +
                "\"applicationName\":\"appname\"," +
                "\"applicationVersion\":\"appv1\"," +
                "\"contactPoints\":{\"localhost\":[\"127.0.0.1:9042\"]}," +
                "\"initialControlConnection\":\"10.10.10.10:9011\"," +
                "\"protocolVersion\":4," +
                "\"localAddress\":\"10.10.10.2:9015\"," +
                "\"executionProfiles\":{" +
                "\"default\":{" +
                "\"readTimeout\":1505," +
                "\"retry\":{" +
                "\"type\":\"DowngradingConsistencyRetryPolicy\"," +
                "\"namespace\":\"Cassandra\"}," +
                "\"loadBalancing\":{" +
                "\"type\":\"RoundRobinPolicy\"," +
                "\"namespace\":\"Cassandra\"}," +
                "\"speculativeExecution\":{" +
                "\"type\":\"ConstantSpeculativeExecutionPolicy\"," +
                "\"namespace\":\"Cassandra\"," +
                "\"options\":{" +
                "\"delay\":1213," +
                "\"maxSpeculativeExecutions\":10}}," +
                "\"consistency\":\"ALL\"," +
                "\"serialConsistency\":\"LOCAL_SERIAL\"," +
                "\"graphOptions\":{" +
                "\"language\":\"gremlin-groovy\"," +
                "\"source\":\"g\"," +
                "\"name\":\"testGraphName\"," +
                "\"readConsistency\":\"ALL\"," +
                "\"writeConsistency\":null," +
                "\"readTimeout\":-1}}," +
                "\"profile2\":{" +
                "\"readTimeout\":501," +
                "\"retry\":{" +
                "\"type\":\"IdempotenceAwareRetryPolicy\"," +
                "\"namespace\":\"Cassandra\"," +
                "\"options\":{" +
                "\"childPolicy\":{" +
                "\"type\":\"DefaultRetryPolicy\"," +
                "\"namespace\":\"Cassandra\"}}}," +
                "\"loadBalancing\":{" +
                "\"type\":\"TokenAwarePolicy\"," +
                "\"namespace\":\"Cassandra\"," +
                "\"options\":{" +
                "\"childPolicy\":{" +
                "\"type\":\"RoundRobinPolicy\"," +
                "\"namespace\":\"Cassandra\"}}}," +
                "\"speculativeExecution\":{" +
                "\"type\":\"ConstantSpeculativeExecutionPolicy\"," +
                "\"namespace\":\"Cassandra\"," +
                "\"options\":{" +
                "\"delay\":230," +
                "\"maxSpeculativeExecutions\":5}}," +
                "\"serialConsistency\":\"SERIAL\"" +
                "}," +
                "\"profile3\":{" +
                "\"consistency\":\"EACH_QUORUM\"" +
                "}" +
                "}," +
                "\"poolSizeByHostDistance\":{" +
                "\"local\":1," +
                "\"remote\":5}," +
                "\"heartbeatInterval\":10000," +
                "\"compression\":\"SNAPPY\"," +
                "\"reconnectionPolicy\":{" +
                "\"type\":\"ConstantReconnectionPolicy\"," +
                "\"namespace\":\"Cassandra\"," +
                "\"options\":{" +
                "\"constantDelayMs\":150}}," +
                "\"ssl\":{\"enabled\":false}," +
                "\"authProvider\":{" +
                "\"type\":\"NoneAuthProvider\"," +
                "\"namespace\":\"Cassandra\"}," +
                "\"configAntiPatterns\":{" +
                "\"downgradingConsistency\":\"Downgrading consistency retry policy in use\"}," +
                "\"periodicStatusInterval\":5," +
                "\"platformInfo\":{" +
                "\"os\":{\"name\":\"os name\",\"version\":\"os version\",\"arch\":\"os arch\"}," +
                "\"cpus\":{\"length\":2,\"model\":\"Awesome CPU\"}," +
                "\"runtime\":{" +
                "\"runtimeFramework\":\"runtime-framework\"," +
                "\"targetFramework\":\"target-framework\"," +
                "\"dependencies\":{" +
                "\"Assembly1\":{" +
                "\"fullName\":\"Assembly1FullName\"," +
                "\"name\":\"Assembly1\"," +
                "\"version\":\"1.2.0\"}}}}," +
                "\"hostName\":\"awesome_hostname\"," +
                "\"driverName\":\"Driver Name\"," +
                "\"applicationNameWasGenerated\":false," +
                "\"driverVersion\":\"1.1.2\"," +
                "\"dataCenters\":[\"dc123\"]}}";
            var cluster = GetCluster(true, eventDelayMilliseconds: 5000);
            var session = GetSession(cluster);

            using (var target = InsightsClientTests.GetInsightsClient(cluster, session))
            {
                var queryProtocolOptions = new ConcurrentQueue <QueryProtocolOptions>();
                Mock.Get(cluster.Metadata.ControlConnection).Setup(cc => cc.UnsafeSendQueryRequestAsync(
                                                                       "CALL InsightsRpc.reportInsight(?)",
                                                                       It.IsAny <QueryProtocolOptions>()))
                .ReturnsAsync(new FakeResultResponse(ResultResponse.ResultResponseKind.Void))
                .Callback <string, QueryProtocolOptions>((query, opts) => { queryProtocolOptions.Enqueue(opts); });

                target.Init();

                TestHelper.RetryAssert(
                    () => { Assert.GreaterOrEqual(queryProtocolOptions.Count, 1); }, 10, 50);
                queryProtocolOptions.TryPeek(out var result);
                Assert.AreEqual(expectedJson, result.Values[0], "Actual: " + Environment.NewLine + result.Values[0]);
            }
        }