62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
// Copyright 2020 PingCAP, Inc. Licensed under Apache-2.0.
|
|
|
|
package task
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
. "github.com/pingcap/check"
|
|
backuppb "github.com/pingcap/kvproto/pkg/brpb"
|
|
)
|
|
|
|
var _ = Suite(&testBackupSuite{})
|
|
|
|
func TestT(t *testing.T) {
|
|
TestingT(t)
|
|
}
|
|
|
|
type testBackupSuite struct{}
|
|
|
|
func (s *testBackupSuite) TestParseTSString(c *C) {
|
|
var (
|
|
ts uint64
|
|
err error
|
|
)
|
|
|
|
ts, err = parseTSString("")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(int(ts), Equals, 0)
|
|
|
|
ts, err = parseTSString("400036290571534337")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(int(ts), Equals, 400036290571534337)
|
|
|
|
_, offset := time.Now().Local().Zone()
|
|
ts, err = parseTSString("2018-05-11 01:42:23")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(int(ts), Equals, 400032515489792000-(offset*1000)<<18)
|
|
}
|
|
|
|
func (s *testBackupSuite) TestParseCompressionType(c *C) {
|
|
var (
|
|
ct backuppb.CompressionType
|
|
err error
|
|
)
|
|
ct, err = parseCompressionType("lz4")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(int(ct), Equals, 1)
|
|
|
|
ct, err = parseCompressionType("snappy")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(int(ct), Equals, 2)
|
|
|
|
ct, err = parseCompressionType("zstd")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(int(ct), Equals, 3)
|
|
|
|
ct, err = parseCompressionType("Other Compression (strings)")
|
|
c.Assert(err, ErrorMatches, "invalid compression.*")
|
|
c.Assert(int(ct), Equals, 0)
|
|
}
|