github.com/cs3org/reva/v2@v2.27.7/pkg/storage/utils/downloader/mock/downloader_mock.go (about)

     1  // Copyright 2018-2021 CERN
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  //     http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  // See the License for the specific language governing permissions and
    13  // limitations under the License.
    14  //
    15  // In applying this license, CERN does not waive the privileges and immunities
    16  // granted to it by virtue of its status as an Intergovernmental Organization
    17  // or submit itself to any jurisdiction.
    18  
    19  package mock
    20  
    21  import (
    22  	"bufio"
    23  	"context"
    24  	"io"
    25  	"os"
    26  
    27  	"github.com/cs3org/reva/v2/pkg/storage/utils/downloader"
    28  
    29  	providerv1beta1 "github.com/cs3org/go-cs3apis/cs3/storage/provider/v1beta1"
    30  )
    31  
    32  type mockDownloader struct{}
    33  
    34  // NewDownloader creates a mock downloader that implements the Downloader interface
    35  // supposed to be used for testing
    36  func NewDownloader() downloader.Downloader {
    37  	return &mockDownloader{}
    38  }
    39  
    40  // Download copies the content of a local file into the dst Writer
    41  func (m *mockDownloader) Download(ctx context.Context, id *providerv1beta1.ResourceId, dst io.Writer) error {
    42  	f, err := os.Open(id.OpaqueId)
    43  	if err != nil {
    44  		return err
    45  	}
    46  	defer f.Close()
    47  	fr := bufio.NewReader(f)
    48  	_, err = io.Copy(dst, fr)
    49  	return err
    50  }