From ddc6ccec4f7700c4538f56a40683dba643923e7f Mon Sep 17 00:00:00 2001
From: ali <aliomerkeser@gmail.com>
Date: Tue, 9 Jan 2024 14:28:03 +0300
Subject: [PATCH] increase test coverage

---
 drivers/r2/driver_test.go | 95 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 95 insertions(+)

diff --git a/drivers/r2/driver_test.go b/drivers/r2/driver_test.go
index eac4dc7..28b2dff 100644
--- a/drivers/r2/driver_test.go
+++ b/drivers/r2/driver_test.go
@@ -3,8 +3,10 @@ package r2
 import (
 	"bytes"
 	"context"
+	"fmt"
 	"io"
 	"testing"
+	"time"
 
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/service/s3"
@@ -127,3 +129,96 @@ func (s *DriverTestSuite) TestWriter() {
 	s.r.NoError(writer.Commit())
 	s.r.NoError(writer.Close())
 }
+
+func (s *DriverTestSuite) TestPutContent() {
+	s.r2Client.EXPECT().PutObject(gomock.Any(), gomock.Any()).Return(&s3.PutObjectOutput{}, nil)
+
+	err := s.driver.PutContent(context.Background(), testPath, []byte("1"))
+	s.r.NoError(err)
+}
+
+func (s *DriverTestSuite) TestStat() {
+	s.r2Client.EXPECT().ListObjectsV2(gomock.Any(), gomock.Any(), gomock.Any()).
+		Return(&s3.ListObjectsV2Output{
+			Contents: []types.Object{{
+				Key:          aws.String(fmt.Sprintf("test-path/x")),
+				Size:         aws.Int64(123),
+				LastModified: aws.Time(time.Now()),
+			}},
+		}, nil)
+
+	stat, err := s.driver.Stat(context.Background(), testPath)
+	s.r.NoError(err)
+	s.r.NotNil(stat)
+}
+
+func (s *DriverTestSuite) TestList() {
+	s.r2Client.EXPECT().ListObjectsV2(gomock.Any(), gomock.Any(), gomock.Any()).
+		Return(&s3.ListObjectsV2Output{
+			IsTruncated: aws.Bool(false),
+			Contents: []types.Object{{
+				Key:          aws.String(fmt.Sprintf("test-path/x")),
+				Size:         aws.Int64(123),
+				LastModified: aws.Time(time.Now()),
+			}},
+		}, nil)
+
+	list, err := s.driver.List(context.Background(), testPath)
+	s.r.NoError(err)
+	s.r.Equal([]string{"/test-path/x"}, list)
+}
+
+func (s *DriverTestSuite) TestMove() {
+	s.r2Client.EXPECT().ListObjectsV2(gomock.Any(), gomock.Any(), gomock.Any()).
+		Return(&s3.ListObjectsV2Output{
+			Contents: []types.Object{{
+				Key:          aws.String(fmt.Sprintf("test-path/x")),
+				Size:         aws.Int64(123),
+				LastModified: aws.Time(time.Now()),
+			}},
+		}, nil)
+
+	s.r2Client.EXPECT().CopyObject(gomock.Any(), gomock.Any(), gomock.Any()).
+		Return(&s3.CopyObjectOutput{}, nil)
+
+	s.r2Client.EXPECT().ListObjectsV2(gomock.Any(), gomock.Any(), gomock.Any()).
+		Return(&s3.ListObjectsV2Output{
+			Contents: []types.Object{{
+				Key:          aws.String(fmt.Sprintf("test-path/x")),
+				Size:         aws.Int64(123),
+				LastModified: aws.Time(time.Now()),
+			}},
+		}, nil)
+
+	s.r2Client.EXPECT().DeleteObjects(gomock.Any(), gomock.Any(), gomock.Any()).
+		Return(&s3.DeleteObjectsOutput{}, nil)
+
+	s.r.NoError(s.driver.Move(context.Background(), testPath, testPath+"1"))
+}
+
+func (s *DriverTestSuite) TestDelete() {
+	s.r2Client.EXPECT().ListObjectsV2(gomock.Any(), gomock.Any()).
+		Return(&s3.ListObjectsV2Output{
+			Contents: []types.Object{{
+				Key: aws.String(fmt.Sprintf("test-path/x")),
+			}},
+		}, nil)
+	s.r2Client.EXPECT().DeleteObjects(gomock.Any(), gomock.Any()).Return(&s3.DeleteObjectsOutput{}, nil)
+
+	s.r.NoError(s.driver.Delete(context.Background(), testPath))
+}
+
+func (s *DriverTestSuite) TestWalk() {
+	s.r2Client.EXPECT().ListObjectsV2(gomock.Any(), gomock.Any(), gomock.Any()).
+		Return(&s3.ListObjectsV2Output{
+			Contents: []types.Object{{
+				Key:          aws.String(fmt.Sprintf("test-path/x")),
+				Size:         aws.Int64(123),
+				LastModified: aws.Time(time.Now()),
+			}},
+		}, nil)
+
+	s.r.NoError(s.driver.Walk(context.Background(), testPath, func(fileInfo storagedriver.FileInfo) error {
+		return nil
+	}))
+}