|
9 | 9 |
|
10 | 10 | "github.com/oklog/ulid/v2" |
11 | 11 | "github.com/stretchr/testify/require" |
| 12 | + "gopkg.in/guregu/null.v4" |
12 | 13 |
|
13 | 14 | "github.com/frain-dev/convoy/config" |
14 | 15 | "github.com/frain-dev/convoy/database" |
@@ -85,6 +86,133 @@ func TestCreateDeliveryAttempt_ValidRequest(t *testing.T) { |
85 | 86 | require.Equal(t, attempt.Status, fetched.Status) |
86 | 87 | } |
87 | 88 |
|
| 89 | +func TestCreateDeliveryAttempt_PersistsRequestedAndRespondedAt(t *testing.T) { |
| 90 | + db, ctx := setupTestDB(t) |
| 91 | + defer db.Close() |
| 92 | + |
| 93 | + project := seedTestData(t, db, ctx) |
| 94 | + endpoint := seedEndpoint(t, db, ctx, project) |
| 95 | + eventDelivery := seedEventDelivery(t, db, ctx, project, endpoint) |
| 96 | + |
| 97 | + service := New(log.New("convoy", log.LevelError), db) |
| 98 | + |
| 99 | + requestedAt := time.Now().UTC().Truncate(time.Millisecond) |
| 100 | + respondedAt := requestedAt.Add(75 * time.Millisecond) |
| 101 | + |
| 102 | + attempt := &datastore.DeliveryAttempt{ |
| 103 | + UID: ulid.Make().String(), |
| 104 | + URL: "https://example.com/webhook", |
| 105 | + Method: "POST", |
| 106 | + APIVersion: "2023.12.25", |
| 107 | + EndpointID: endpoint.UID, |
| 108 | + EventDeliveryId: eventDelivery.UID, |
| 109 | + ProjectId: project.UID, |
| 110 | + HttpResponseCode: "200", |
| 111 | + Status: true, |
| 112 | + RequestedAt: null.TimeFrom(requestedAt), |
| 113 | + RespondedAt: null.TimeFrom(respondedAt), |
| 114 | + } |
| 115 | + |
| 116 | + err := service.CreateDeliveryAttempt(ctx, attempt) |
| 117 | + require.NoError(t, err) |
| 118 | + |
| 119 | + fetched, err := service.FindDeliveryAttemptById(ctx, eventDelivery.UID, attempt.UID) |
| 120 | + require.NoError(t, err) |
| 121 | + require.True(t, fetched.RequestedAt.Valid) |
| 122 | + require.True(t, fetched.RespondedAt.Valid) |
| 123 | + require.True(t, fetched.RequestedAt.Time.Equal(requestedAt)) |
| 124 | + require.True(t, fetched.RespondedAt.Time.Equal(respondedAt)) |
| 125 | + |
| 126 | + // And via the list path used by the meta-event listener. |
| 127 | + attempts, err := service.FindDeliveryAttempts(ctx, eventDelivery.UID) |
| 128 | + require.NoError(t, err) |
| 129 | + require.Len(t, attempts, 1) |
| 130 | + require.True(t, attempts[0].RequestedAt.Valid) |
| 131 | + require.True(t, attempts[0].RespondedAt.Valid) |
| 132 | +} |
| 133 | + |
| 134 | +func TestCreateDeliveryAttempt_NoResponseLeavesRespondedAtNull(t *testing.T) { |
| 135 | + db, ctx := setupTestDB(t) |
| 136 | + defer db.Close() |
| 137 | + |
| 138 | + project := seedTestData(t, db, ctx) |
| 139 | + endpoint := seedEndpoint(t, db, ctx, project) |
| 140 | + eventDelivery := seedEventDelivery(t, db, ctx, project, endpoint) |
| 141 | + |
| 142 | + service := New(log.New("convoy", log.LevelError), db) |
| 143 | + |
| 144 | + attempt := &datastore.DeliveryAttempt{ |
| 145 | + UID: ulid.Make().String(), |
| 146 | + URL: "https://example.com/webhook", |
| 147 | + Method: "POST", |
| 148 | + APIVersion: "2023.12.25", |
| 149 | + EndpointID: endpoint.UID, |
| 150 | + EventDeliveryId: eventDelivery.UID, |
| 151 | + ProjectId: project.UID, |
| 152 | + Error: "connection refused", |
| 153 | + Status: false, |
| 154 | + RequestedAt: null.TimeFrom(time.Now().UTC()), |
| 155 | + // RespondedAt intentionally left zero (no HTTP response). |
| 156 | + } |
| 157 | + |
| 158 | + err := service.CreateDeliveryAttempt(ctx, attempt) |
| 159 | + require.NoError(t, err) |
| 160 | + |
| 161 | + fetched, err := service.FindDeliveryAttemptById(ctx, eventDelivery.UID, attempt.UID) |
| 162 | + require.NoError(t, err) |
| 163 | + require.True(t, fetched.RequestedAt.Valid) |
| 164 | + require.False(t, fetched.RespondedAt.Valid) |
| 165 | +} |
| 166 | + |
| 167 | +func TestPartitionRoundTrip_PreservesRequestedAndRespondedAt(t *testing.T) { |
| 168 | + db, ctx := setupTestDB(t) |
| 169 | + defer db.Close() |
| 170 | + |
| 171 | + project := seedTestData(t, db, ctx) |
| 172 | + endpoint := seedEndpoint(t, db, ctx, project) |
| 173 | + eventDelivery := seedEventDelivery(t, db, ctx, project, endpoint) |
| 174 | + |
| 175 | + service := New(log.New("convoy", log.LevelError), db) |
| 176 | + |
| 177 | + requestedAt := time.Now().UTC().Truncate(time.Millisecond) |
| 178 | + respondedAt := requestedAt.Add(60 * time.Millisecond) |
| 179 | + |
| 180 | + attempt := &datastore.DeliveryAttempt{ |
| 181 | + UID: ulid.Make().String(), |
| 182 | + URL: "https://example.com/webhook", |
| 183 | + Method: "POST", |
| 184 | + APIVersion: "2023.12.25", |
| 185 | + EndpointID: endpoint.UID, |
| 186 | + EventDeliveryId: eventDelivery.UID, |
| 187 | + ProjectId: project.UID, |
| 188 | + HttpResponseCode: "200", |
| 189 | + Status: true, |
| 190 | + RequestedAt: null.TimeFrom(requestedAt), |
| 191 | + RespondedAt: null.TimeFrom(respondedAt), |
| 192 | + } |
| 193 | + require.NoError(t, service.CreateDeliveryAttempt(ctx, attempt)) |
| 194 | + |
| 195 | + // Partition rebuild must carry the new columns over. |
| 196 | + require.NoError(t, service.PartitionDeliveryAttemptsTable(ctx)) |
| 197 | + |
| 198 | + fetched, err := service.FindDeliveryAttemptById(ctx, eventDelivery.UID, attempt.UID) |
| 199 | + require.NoError(t, err) |
| 200 | + require.True(t, fetched.RequestedAt.Valid) |
| 201 | + require.True(t, fetched.RespondedAt.Valid) |
| 202 | + require.True(t, fetched.RequestedAt.Time.Equal(requestedAt)) |
| 203 | + require.True(t, fetched.RespondedAt.Time.Equal(respondedAt)) |
| 204 | + |
| 205 | + // Un-partition rebuild must also carry them over. |
| 206 | + require.NoError(t, service.UnPartitionDeliveryAttemptsTable(ctx)) |
| 207 | + |
| 208 | + fetched, err = service.FindDeliveryAttemptById(ctx, eventDelivery.UID, attempt.UID) |
| 209 | + require.NoError(t, err) |
| 210 | + require.True(t, fetched.RequestedAt.Valid) |
| 211 | + require.True(t, fetched.RespondedAt.Valid) |
| 212 | + require.True(t, fetched.RequestedAt.Time.Equal(requestedAt)) |
| 213 | + require.True(t, fetched.RespondedAt.Time.Equal(respondedAt)) |
| 214 | +} |
| 215 | + |
88 | 216 | func TestCreateDeliveryAttempt_WithHeaders(t *testing.T) { |
89 | 217 | db, ctx := setupTestDB(t) |
90 | 218 | defer db.Close() |
|
0 commit comments