dd-trace-csharp/src/DataDog.Tracing.Sql.Tests/TraceDbCommandTests.cs at 8c08070dff88c0d7565fbe6e7528ef0562610366 · bcuff/dd-trace-csharp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
using System;
using System.Collections.Generic;
using System.Data;
using System.Reactive;
using System.Text;
using FluentAssertions;
using Microsoft.Data.Sqlite;
using NUnit.Framework;
using SQLitePCL;

namespace DataDog.Tracing.Sql.Tests
{
    [TestFixture]
    public class TraceDbCommandTests
    {
        RootSpan _root;
        SqliteConnection _conn;

        [SetUp]
        public void SetUp()
        {
            _conn = new SqliteConnection("Filename=./test.db");
            _conn.Open();
            _root = new RootSpan();
        }

        [Test]
        [TestCase(CommandBehavior.Default)]
        [TestCase(CommandBehavior.CloseConnection)]
        public void ExecuteReader_is_traced(CommandBehavior commandBehavior)
        {
            var customers = new List<Customer>();
            using (var command = new TraceDbCommand(_conn.CreateCommand(), _root))
            {
                command.CommandText = "SELECT * FROM Customers";
                command.CommandType = CommandType.Text;
                using (var reader = command.ExecuteReader(commandBehavior))
                {
                    while (reader.Read())
                    {
                        customers.Add(new Customer(reader));
                    }
                }
            }
            customers.Count.Should().Be(2);
            _root.Spans[1].Name.Should().Be("sql.ExecuteDbDataReader");
            _root.Spans[1].Service.Should().Be("sql");
            _root.Spans[1].Resource.Should().Be("main");
            _root.Spans[1].Type.Should().Be("sql");
            _root.Spans[1].Error.Should().Be(0);
            _root.Spans[1].Meta["sql.CommandBehavior"].Should().Be(commandBehavior.ToString("x"));
            _root.Spans[1].Meta["sql.CommandText"].Should().Be("SELECT * FROM Customers");
            _root.Spans[1].Meta["sql.CommandType"].Should().Be("Text");
        }

        [Test]
        public void ExecuteNonQuery_is_traced()
        {
            int rows;
            using (var command = new TraceDbCommand(_conn.CreateCommand(), _root))
            {
                command.CommandText = "SELECT * FROM Customers";
                command.CommandType = CommandType.Text;
                rows = command.ExecuteNonQuery();
            }
            _root.Spans[1].Name.Should().Be("sql." + nameof(IDbCommand.ExecuteNonQuery));
            _root.Spans[1].Service.Should().Be("sql");
            _root.Spans[1].Resource.Should().Be("main");
            _root.Spans[1].Type.Should().Be("sql");
            _root.Spans[1].Error.Should().Be(0);
            _root.Spans[1].Meta["sql.RowsAffected"].Should().Be(rows.ToString());
            _root.Spans[1].Meta["sql.CommandText"].Should().Be("SELECT * FROM Customers");
            _root.Spans[1].Meta["sql.CommandType"].Should().Be("Text");
        }

        [Test]
        public void ExecuteScalar_is_traced()
        {
            object result;
            using (var command = new TraceDbCommand(_conn.CreateCommand(), _root))
            {
                command.CommandText = "SELECT COUNT(*) FROM Customers";
                command.CommandType = CommandType.Text;
                result = command.ExecuteScalar();
            }
            result.Should().Be(2L);
            _root.Spans[1].Name.Should().Be("sql." + nameof(IDbCommand.ExecuteScalar));
            _root.Spans[1].Service.Should().Be("sql");
            _root.Spans[1].Resource.Should().Be("main");
            _root.Spans[1].Type.Should().Be("sql");
            _root.Spans[1].Error.Should().Be(0);
            _root.Spans[1].Meta["sql.CommandText"].Should().Be("SELECT COUNT(*) FROM Customers");
            _root.Spans[1].Meta["sql.CommandType"].Should().Be("Text");
        }

        [TearDown]
        public void TearDown()
        {
            _conn.Close();
        }
    }
}