Skip to content

Commit 640d8ee

Browse files
michael-zhao459michael-zhao459
committed
lint
1 parent 5476864 commit 640d8ee

File tree

1 file changed

+28
-26
lines changed

1 file changed

+28
-26
lines changed

‎tests/test_dsm.py‎

Lines changed: 28 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -166,9 +166,7 @@ def test_sqs_invalid_datadog_message_attribute(self, mock_dsm_logger):
166166
self.assertEqual(self.mock_checkpoint.call_count, 1)
167167
args, _ = self.mock_checkpoint.call_args
168168
self.assertEqual(args[0], "sqs")
169-
self.assertEqual(
170-
args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue"
171-
)
169+
self.assertEqual(args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue")
172170
carrier_get = args[2]
173171
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), None)
174172

@@ -193,9 +191,7 @@ def test_sqs_invalid_datadog_message_attribute(self, mock_dsm_logger):
193191
self.assertEqual(self.mock_checkpoint.call_count, 1)
194192
args, _ = self.mock_checkpoint.call_args
195193
self.assertEqual(args[0], "sqs")
196-
self.assertEqual(
197-
args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue"
198-
)
194+
self.assertEqual(args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue")
199195
carrier_get = args[2]
200196
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), None)
201197

@@ -253,14 +249,18 @@ def test_sqs_multiple_records(self):
253249
# Check first call
254250
first_call_args = self.mock_checkpoint.call_args_list[0][0]
255251
self.assertEqual(first_call_args[0], "sqs")
256-
self.assertEqual(first_call_args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue")
252+
self.assertEqual(
253+
first_call_args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue"
254+
)
257255
carrier_get = first_call_args[2]
258256
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), "12345")
259257

260258
# Check second call
261259
second_call_args = self.mock_checkpoint.call_args_list[1][0]
262260
self.assertEqual(second_call_args[0], "sqs")
263-
self.assertEqual(second_call_args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue-2")
261+
self.assertEqual(
262+
second_call_args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue-2"
263+
)
264264
carrier_get = second_call_args[2]
265265
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), "12345")
266266

@@ -425,9 +425,7 @@ def test_sns_invalid_datadog_message_attribute(self, mock_dsm_logger):
425425
self.assertEqual(self.mock_checkpoint.call_count, 1)
426426
args, _ = self.mock_checkpoint.call_args
427427
self.assertEqual(args[0], "sns")
428-
self.assertEqual(
429-
args[1], "arn:aws:sns:us-east-1:123456789012:test-topic"
430-
)
428+
self.assertEqual(args[1], "arn:aws:sns:us-east-1:123456789012:test-topic")
431429
carrier_get = args[2]
432430
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), None)
433431

@@ -455,9 +453,7 @@ def test_sns_invalid_datadog_message_attribute(self, mock_dsm_logger):
455453
self.assertEqual(self.mock_checkpoint.call_count, 1)
456454
args, _ = self.mock_checkpoint.call_args
457455
self.assertEqual(args[0], "sns")
458-
self.assertEqual(
459-
args[1], "arn:aws:sns:us-east-1:123456789012:test-topic"
460-
)
456+
self.assertEqual(args[1], "arn:aws:sns:us-east-1:123456789012:test-topic")
461457
carrier_get = args[2]
462458
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), None)
463459

@@ -524,14 +520,18 @@ def test_sns_multiple_records(self):
524520
# Check first call
525521
first_call_args = self.mock_checkpoint.call_args_list[0][0]
526522
self.assertEqual(first_call_args[0], "sns")
527-
self.assertEqual(first_call_args[1], "arn:aws:sns:us-east-1:123456789012:test-topic")
523+
self.assertEqual(
524+
first_call_args[1], "arn:aws:sns:us-east-1:123456789012:test-topic"
525+
)
528526
carrier_get = first_call_args[2]
529527
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), "12345")
530528

531529
# Check second call
532530
second_call_args = self.mock_checkpoint.call_args_list[1][0]
533531
self.assertEqual(second_call_args[0], "sns")
534-
self.assertEqual(second_call_args[1], "arn:aws:sns:us-east-1:123456789012:test-topic-2")
532+
self.assertEqual(
533+
second_call_args[1], "arn:aws:sns:us-east-1:123456789012:test-topic-2"
534+
)
535535
carrier_get = second_call_args[2]
536536
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), "12345")
537537

@@ -725,9 +725,7 @@ def test_sns_to_sqs_invalid_datadog_message_attribute(self, mock_dsm_logger):
725725
self.assertEqual(self.mock_checkpoint.call_count, 1)
726726
args, _ = self.mock_checkpoint.call_args
727727
self.assertEqual(args[0], "sqs")
728-
self.assertEqual(
729-
args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue"
730-
)
728+
self.assertEqual(args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue")
731729
carrier_get = args[2]
732730
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), None)
733731

@@ -760,9 +758,7 @@ def test_sns_to_sqs_invalid_datadog_message_attribute(self, mock_dsm_logger):
760758
self.assertEqual(self.mock_checkpoint.call_count, 1)
761759
args, _ = self.mock_checkpoint.call_args
762760
self.assertEqual(args[0], "sqs")
763-
self.assertEqual(
764-
args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue"
765-
)
761+
self.assertEqual(args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue")
766762
carrier_get = args[2]
767763
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), None)
768764

@@ -832,14 +828,18 @@ def test_sns_to_sqs_multiple_records(self):
832828
# Check first call
833829
first_call_args = self.mock_checkpoint.call_args_list[0][0]
834830
self.assertEqual(first_call_args[0], "sqs")
835-
self.assertEqual(first_call_args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue")
831+
self.assertEqual(
832+
first_call_args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue"
833+
)
836834
carrier_get = first_call_args[2]
837835
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), "12345")
838836

839837
# Check second call
840838
second_call_args = self.mock_checkpoint.call_args_list[1][0]
841839
self.assertEqual(second_call_args[0], "sqs")
842-
self.assertEqual(second_call_args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue-2")
840+
self.assertEqual(
841+
second_call_args[1], "arn:aws:sqs:us-east-1:123456789012:test-queue-2"
842+
)
843843
carrier_get = second_call_args[2]
844844
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), "12345")
845845

@@ -1023,15 +1023,17 @@ def test_kinesis_multiple_records(self):
10231023
first_call_args = self.mock_checkpoint.call_args_list[0][0]
10241024
self.assertEqual(first_call_args[0], "kinesis")
10251025
self.assertEqual(
1026-
first_call_args[1], "arn:aws:kinesis:us-east-1:123456789012:stream/test-stream"
1026+
first_call_args[1],
1027+
"arn:aws:kinesis:us-east-1:123456789012:stream/test-stream",
10271028
)
10281029
carrier_get = first_call_args[2]
10291030
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), "12345")
10301031

10311032
second_call_args = self.mock_checkpoint.call_args_list[1][0]
10321033
self.assertEqual(second_call_args[0], "kinesis")
10331034
self.assertEqual(
1034-
second_call_args[1], "arn:aws:kinesis:us-east-1:123456789012:stream/test-stream-2"
1035+
second_call_args[1],
1036+
"arn:aws:kinesis:us-east-1:123456789012:stream/test-stream-2",
10351037
)
10361038
carrier_get = second_call_args[2]
10371039
self.assertEqual(carrier_get("dd-pathway-ctx-base64"), "12345")

0 commit comments

Comments
 (0)