github.com/hugh712/snapd@v0.0.0-20200910133618-1a99902bd583/osutil/digest_test.go (about) 1 // -*- Mode: Go; indent-tabs-mode: t -*- 2 3 /* 4 * Copyright (C) 2014-2015 Canonical Ltd 5 * 6 * This program is free software: you can redistribute it and/or modify 7 * it under the terms of the GNU General Public License version 3 as 8 * published by the Free Software Foundation. 9 * 10 * This program is distributed in the hope that it will be useful, 11 * but WITHOUT ANY WARRANTY; without even the implied warranty of 12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 13 * GNU General Public License for more details. 14 * 15 * You should have received a copy of the GNU General Public License 16 * along with this program. If not, see <http://www.gnu.org/licenses/>. 17 * 18 */ 19 20 package osutil_test 21 22 import ( 23 "crypto" 24 "crypto/sha512" 25 "io/ioutil" 26 "path/filepath" 27 28 . "gopkg.in/check.v1" 29 30 "github.com/snapcore/snapd/osutil" 31 ) 32 33 type FileDigestSuite struct{} 34 35 var _ = Suite(&FileDigestSuite{}) 36 37 func (ts *FileDigestSuite) TestFileDigest(c *C) { 38 exData := []byte("hashmeplease") 39 40 tempdir := c.MkDir() 41 fn := filepath.Join(tempdir, "ex.file") 42 err := ioutil.WriteFile(fn, exData, 0644) 43 c.Assert(err, IsNil) 44 45 digest, size, err := osutil.FileDigest(fn, crypto.SHA512) 46 c.Assert(err, IsNil) 47 c.Check(size, Equals, uint64(len(exData))) 48 h512 := sha512.Sum512(exData) 49 c.Check(digest, DeepEquals, h512[:]) 50 }