Files
snapd/randutil/rand_test.go
2020-03-21 08:34:16 +01:00

63 lines
1.5 KiB
Go

// -*- Mode: Go; indent-tabs-mode: t -*-
/*
* Copyright (C) 2014-2020 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
package randutil_test
import (
"math/rand"
"testing"
"time"
. "gopkg.in/check.v1"
"github.com/snapcore/snapd/randutil"
)
func Test(t *testing.T) { TestingT(t) }
type randutilSuite struct{}
var _ = Suite(&randutilSuite{})
func (s *randutilSuite) TestRandomString(c *C) {
// for our tests
rand.Seed(1)
s1 := randutil.RandomString(10)
c.Assert(s1, Equals, "pw7MpXh0JB")
s2 := randutil.RandomString(5)
c.Assert(s2, Equals, "4PQyl")
}
func (s *randutilSuite) TestRandomDuration(c *C) {
// ensure moreMixedSeed is done
d := randutil.RandomDuration(time.Hour)
c.Check(d < time.Hour, Equals, true)
// for our tests
rand.Seed(1)
d1 := randutil.RandomDuration(time.Hour)
c.Assert(d1, Equals, time.Duration(1991947779410))
d2 := randutil.RandomDuration(4 * time.Hour)
c.Assert(d2, Equals, time.Duration(4423082153551))
}