feat: support envelopes in integration test

getsentry · Nov 27, 2023 · 029ec39 · 029ec39
1 parent fe18d6e
commit 029ec39
Show file tree

Hide file tree

Showing 4 changed files with 55 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,11 @@
 # Changelog
 
+## Unreleased
+
+### Features
+
+- Sentry-CLI integration test action: support envelopes ([#58](https://github.com/getsentry/github-workflows/pull/58))
+
 ## 2.7.0
 
 ### Features

diff --git a/sentry-cli/integration-test/action.psm1 b/sentry-cli/integration-test/action.psm1
@@ -17,6 +17,28 @@ class InvokeSentryResult
         return @($this.ServerStdOut | Where-Object { $_.StartsWith($prefix) } | ForEach-Object { $_.Substring($prefix.Length).Trim() })
     }
 
+    # Envelopes are collected to a list, each envelope body a single item.
+    [string[]]Envelopes()
+    {
+        $envelopes = @()
+        $this.ServerStdOut | ForEach-Object {
+            if ($_.Trim() -eq "envelope start")
+            {
+                $envelope = ''
+            }
+            elseif ($_ -eq "envelope end")
+            {
+                $envelopes += $envelope
+                $envelope = $null
+            }
+            elseif ($null -ne $envelope)
+            {
+                $envelope += $_ + "\n"
+            }
+        }
+        return $envelopes
+    }
+
     [bool]HasErrors()
     {
         return $this.ServerStdErr.Length -gt 0

diff --git a/sentry-cli/integration-test/sentry-server.py b/sentry-cli/integration-test/sentry-server.py
@@ -92,6 +92,11 @@ def do_POST(self):
             self.writeJSON('{ }')
         elif self.isApi('api/0/organizations/{}/chunk-upload/'.format(apiOrg)):
             self.writeJSON('{ }')
+        elif self.isApi('api/0/envelope'):
+            sys.stdout.write("     envelope start\n")
+            sys.stdout.write(self.body)
+            sys.stdout.write("\n     envelope end\n")
+            self.writeJSON('{ }')
         else:
             self.writeNoApiMatchesError()
 

diff --git a/sentry-cli/integration-test/tests/action.Tests.ps1 b/sentry-cli/integration-test/tests/action.Tests.ps1
@@ -42,4 +42,26 @@ Describe 'Invoke-SentryServer' {
         Should -ActualValue $result.HasErrors() -BeFalse
         $result.UploadedDebugFiles() | Should -Be @('file3.dylib', 'file2.so', 'file1.dll')
     }
+
+    It "collects envelopes" {
+        $result = Invoke-SentryServer {
+            Param([string]$url)
+            Invoke-WebRequest -Uri "$url/api/0/envelope" -Method Post -Body @'
+{"event_id":"9ec79c33ec9942ab8353589fcb2e04dc","dsn":"https://e12d836b15bb49d7bbf99e64295d995b:@sentry.io/42"}
+{"type":"attachment","length":10,"content_type":"text/plain","filename":"hello.txt"}
+\xef\xbb\xbfHello\r\n
+{"type":"event","length":41,"content_type":"application/json","filename":"application.log"}
+{"message":"hello world","level":"error"}
+'@
+            Invoke-WebRequest -Uri "$url/api/0/envelope" -Method Post -Body @'
+{"event_id":"9ec79c33ec9942ab8353589fcb2e04dc"}
+{"type":"attachment"}
+helloworld
+'@
+        }
+        Should -ActualValue $result.HasErrors() -BeFalse
+        $result.Envelopes().Length | Should -Be 2
+        $result.Envelopes()[0].Length | Should -Be 357
+        $result.Envelopes()[1].Length | Should -Be 84
+    }
 }