add UT

aliyun · May 28, 2024 · 842f534 · 842f534
1 parent 572bb7d
commit 842f534
Showing 1 changed file with 39 additions and 0 deletions.
diff --git a/logstore_test.go b/logstore_test.go
@@ -229,6 +229,45 @@ func (s *LogstoreTestSuite) TestPullLogs() {
 	s.Nil(err)
 }
 
+func (s *LogstoreTestSuite) TestGetLogByteWithError() {
+	c := &LogContent{
+		Key:   proto.String("error code"),
+		Value: proto.String("InternalServerError"),
+	}
+	l := &Log{
+		Time: proto.Uint32(uint32(time.Now().Unix())),
+		Contents: []*LogContent{
+			c,
+		},
+	}
+	lg := &LogGroup{
+		Topic:  proto.String("demo topic"),
+		Source: proto.String("10.230.201.117"),
+		Logs: []*Log{
+			l,
+		},
+	}
+
+	shards, err := s.Logstore.ListShards()
+	s.True(len(shards) > 0)
+
+	err = s.Logstore.PutLogs(lg)
+	s.Nil(err)
+
+	cursor, err := s.Logstore.GetCursor(0, "begin")
+	s.Nil(err)
+	endCursor, err := s.Logstore.GetCursor(0, "end")
+	s.Nil(err)
+
+	_, _, err = s.Logstore.GetLogsBytes(1000, cursor, "", 10)
+	s.Contains(err.Error(), "ShardNotExist")
+	s.NotNil(err)
+
+	_, _, err = s.Logstore.GetLogsBytes(1000, cursor, endCursor, 10)
+	s.Contains(err.Error(), "ShardNotExist")
+	s.NotNil(err)
+}
+
 func (s *LogstoreTestSuite) TestGetLogs() {
 	idx, err := s.Logstore.GetIndex()
 	if err != nil {