| Lianhao Lu | 432bca4 | 2018-03-24 22:36:08 +0800 | [diff] [blame] | 1 | # Copyright (c) 2018 Intel Corp. All rights reserved. |
| 2 | # |
| 3 | # Licensed under the Apache License, Version 2.0 (the "License"); you may |
| 4 | # not use this file except in compliance with the License. You may obtain |
| 5 | # a copy of the License at |
| 6 | # |
| 7 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | # |
| 9 | # Unless required by applicable law or agreed to in writing, software |
| 10 | # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| 11 | # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| 12 | # License for the specific language governing permissions and limitations |
| 13 | # under the License. |
| 14 | # |
| 15 | |
| 16 | import os |
| 17 | |
| 18 | from vnfsdk_pkgtools.packager import utils |
| 19 | |
| 20 | CONTENT = "needToBeHashed" |
| 21 | SHA256 = "20a480339aa4371099f9503511dcc5a8051ce3884846678ced5611ec64bbfc9c" |
| 22 | SHA512 = "dbed8672e752d51d0c7ca42050f67faf1534e58470bba96e787df5c4cf6a4f8ecf7ad45fb9307adbc5b9dec8432627d86b3eb1d3d43ee9c5e93f754ff2825320" |
| 23 | |
| 24 | def test_cal_file_hash(tmpdir): |
| 25 | p = tmpdir.join("file_to_hash.txt") |
| 26 | p.write(CONTENT) |
| 27 | assert SHA512 == utils.cal_file_hash("", str(p), 'SHA512') |
| 28 | assert SHA256 == utils.cal_file_hash(p.dirname, p.basename, 'sha256') |
| Lianhao Lu | 979471f | 2018-07-26 17:50:28 +0800 | [diff] [blame^] | 29 | |
| 30 | def test_cal_file_hash_remote(mocker): |
| 31 | class FakeRequest(object): |
| 32 | def __init__(self, *args): |
| 33 | self.status_code = 200 |
| 34 | self.content = CONTENT |
| 35 | mocker.patch('requests.get', new=FakeRequest) |
| 36 | assert SHA256 == utils.cal_file_hash("", "http://fake", 'sha256') |