lostluck commented on a change in pull request #16942:
URL: https://github.com/apache/beam/pull/16942#discussion_r815032440
##########
File path: sdks/go/pkg/beam/util/gcsx/gcs_test.go
##########
@@ -13,35 +13,62 @@
// See the License for the specific language governing permissions and
// limitations under the License.
-package gcsx_test
+package gcsx
import (
- "context"
- "time"
+ "testing"
- "cloud.google.com/go/storage"
- "github.com/apache/beam/sdks/v2/go/pkg/beam/util/gcsx"
+ "github.com/apache/beam/sdks/v2/go/pkg/beam/internal/errors"
)
-func Example() {
- ctx := context.Background()
- c, err := gcsx.NewClient(ctx, storage.ScopeReadOnly)
- if err != nil {
- // do something
+func TestMakeObject(t *testing.T) {
+ if got, want := MakeObject("some-bucket", "some/path"),
"gs://some-bucket/some/path"; got != want {
+ t.Fatalf("MakeObject() Got: %v Want: %v", got, want)
}
+}
- buckets, object, err := gcsx.ParseObject("gs://some-bucket/some-object")
- if err != nil {
- // do something
+func TestParseObject(t *testing.T) {
+ tests := []struct {
+ object string
+ bucket string
+ path string
+ err error
+ }{
+ {
+ object: "gs://some-bucket/some-object",
+ bucket: "some-bucket",
+ path: "some-object",
+ err: nil,
+ },
+ {
+ object: "gs://some-bucket",
+ bucket: "some-bucket",
+ path: "",
+ err: nil,
+ },
+ {
+ object: "gs://",
+ bucket: "",
+ path: "",
+ err: errors.Errorf("object gs:// must have bucket"),
+ },
+ {
+ object: "other://some-bucket/some-object",
+ bucket: "",
+ path: "",
+ err: errors.Errorf("object
other://some-bucket/some-object must have 'gs' scheme"),
+ },
}
- ctx, cancel := context.WithTimeout(ctx, 30*time.Second)
- defer cancel()
-
- bytes, err := gcsx.ReadObject(ctx, c, buckets, object)
- if err != nil {
- // do something
+ for _, test := range tests {
+ if bucket, path, err := ParseObject(test.object); bucket !=
test.bucket || path != test.path || (err != nil && test.err == nil) || (err ==
nil && test.err != nil) {
+ t.Errorf("ParseObject(%v) Got: %v, %v, %v Want: %v, %v,
%v", test.object, bucket, path, err, test.bucket, test.path, test.err)
+ }
}
+}
- _ = bytes
+func TestJoin(t *testing.T) {
+ if got, want := Join("gs://some-bucket/some-object", "some/path",
"more/pathing"), "gs://some-bucket/some-object/some/path/more/pathing"; got !=
want {
+ t.Fatalf("MakeObject() Got: %v Want: %v", got, want)
Review comment:
I will note that it's possible to expect and test panics:
```
defer func(t *testing.T) {
if r := recover(); r != nil {
// ... validate r matches the value passed to panic ...
return
}
t.Fatal("expected a panic")
}(t)
```
As you can imagine, it's best for those cases to be isolated.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
