11789: Merge branch 'master' into 11789-arvput-exclude-flag

[arvados.git] / sdk / go / arvados / collection_fs_test.go

diff --git a/sdk/go/arvados/collection_fs_test.go b/sdk/go/arvados/collection_fs_test.go
index 8cfd21e357773803d0f0a9ccf26dc60d71a8af69..f51d1eb3dd1d58d80b278f3859676addb9b0cab5 100644 (file)
--- a/sdk/go/arvados/collection_fs_test.go
+++ b/sdk/go/arvados/collection_fs_test.go
@@ -1,3 +1,7 @@
+// Copyright (C) The Arvados Authors. All rights reserved.
+//
+// SPDX-License-Identifier: Apache-2.0
+
 package arvados
 
 import (
@@ -45,15 +49,22 @@ func (s *CollectionFSSuite) TestReaddirFull(c *check.C) {
 func (s *CollectionFSSuite) TestReaddirLimited(c *check.C) {
        f, err := s.fs.Open("./dir1")
        c.Assert(err, check.IsNil)
-       for i := 0; i < 2; i++ {
-               fis, err := f.Readdir(1)
-               c.Check(err, check.IsNil)
-               c.Check(len(fis), check.Equals, 1)
-               if len(fis) > 0 {
-                       c.Check(fis[0].Size(), check.Equals, int64(3))
-               }
-       }
+
        fis, err := f.Readdir(1)
+       c.Check(err, check.IsNil)
+       c.Check(len(fis), check.Equals, 1)
+       if len(fis) > 0 {
+               c.Check(fis[0].Size(), check.Equals, int64(3))
+       }
+
+       fis, err = f.Readdir(1)
+       c.Check(err, check.Equals, io.EOF)
+       c.Check(len(fis), check.Equals, 1)
+       if len(fis) > 0 {
+               c.Check(fis[0].Size(), check.Equals, int64(3))
+       }
+
+       fis, err = f.Readdir(1)
        c.Check(len(fis), check.Equals, 0)
        c.Check(err, check.NotNil)
        c.Check(err, check.Equals, io.EOF)
@@ -65,7 +76,7 @@ func (s *CollectionFSSuite) TestReaddirLimited(c *check.C) {
        c.Assert(err, check.IsNil)
        fis, err = f.Readdir(2)
        c.Check(len(fis), check.Equals, 1)
-       c.Assert(err, check.IsNil)
+       c.Assert(err, check.Equals, io.EOF)
        fis, err = f.Readdir(2)
        c.Check(len(fis), check.Equals, 0)
        c.Assert(err, check.Equals, io.EOF)