Gunnar Mills | 01d55c3 | 2017-04-20 10:52:15 -0500 | [diff] [blame] | 1 | #include "version.hpp" |
| 2 | #include <gtest/gtest.h> |
| 3 | #include <experimental/filesystem> |
| 4 | #include <stdlib.h> |
| 5 | #include <fstream> |
| 6 | #include <iostream> |
| 7 | #include <sstream> |
| 8 | #include <string> |
Saqib Khan | 26a960d | 2017-09-19 14:23:28 -0500 | [diff] [blame] | 9 | #include <openssl/sha.h> |
Gunnar Mills | 01d55c3 | 2017-04-20 10:52:15 -0500 | [diff] [blame] | 10 | |
| 11 | using namespace phosphor::software::manager; |
| 12 | namespace fs = std::experimental::filesystem; |
| 13 | |
Gunnar Mills | 01d55c3 | 2017-04-20 10:52:15 -0500 | [diff] [blame] | 14 | class VersionTest : public testing::Test |
| 15 | { |
Adriana Kobylak | 2285fe0 | 2018-02-27 15:36:59 -0600 | [diff] [blame^] | 16 | protected: |
| 17 | virtual void SetUp() |
| 18 | { |
| 19 | char versionDir[] = "./versionXXXXXX"; |
| 20 | _directory = mkdtemp(versionDir); |
Gunnar Mills | 01d55c3 | 2017-04-20 10:52:15 -0500 | [diff] [blame] | 21 | |
Adriana Kobylak | 2285fe0 | 2018-02-27 15:36:59 -0600 | [diff] [blame^] | 22 | if (_directory.empty()) |
Gunnar Mills | 01d55c3 | 2017-04-20 10:52:15 -0500 | [diff] [blame] | 23 | { |
Adriana Kobylak | 2285fe0 | 2018-02-27 15:36:59 -0600 | [diff] [blame^] | 24 | throw std::bad_alloc(); |
Gunnar Mills | 01d55c3 | 2017-04-20 10:52:15 -0500 | [diff] [blame] | 25 | } |
Adriana Kobylak | 2285fe0 | 2018-02-27 15:36:59 -0600 | [diff] [blame^] | 26 | } |
Gunnar Mills | 01d55c3 | 2017-04-20 10:52:15 -0500 | [diff] [blame] | 27 | |
Adriana Kobylak | 2285fe0 | 2018-02-27 15:36:59 -0600 | [diff] [blame^] | 28 | virtual void TearDown() |
| 29 | { |
| 30 | fs::remove_all(_directory); |
| 31 | } |
Gunnar Mills | 01d55c3 | 2017-04-20 10:52:15 -0500 | [diff] [blame] | 32 | |
Adriana Kobylak | 2285fe0 | 2018-02-27 15:36:59 -0600 | [diff] [blame^] | 33 | std::string _directory; |
Gunnar Mills | 01d55c3 | 2017-04-20 10:52:15 -0500 | [diff] [blame] | 34 | }; |
| 35 | |
| 36 | /** @brief Make sure we correctly get the version and purpose from getValue()*/ |
| 37 | TEST_F(VersionTest, TestGetValue) |
| 38 | { |
| 39 | auto manifestFilePath = _directory + "/" + "MANIFEST"; |
| 40 | auto version = "test-version"; |
| 41 | auto purpose = "BMC"; |
| 42 | |
| 43 | std::ofstream file; |
| 44 | file.open(manifestFilePath, std::ofstream::out); |
| 45 | ASSERT_TRUE(file.is_open()); |
| 46 | |
| 47 | file << "version=" << version << std::endl; |
| 48 | file << "purpose=" << purpose << std::endl; |
| 49 | file.close(); |
| 50 | |
| 51 | EXPECT_EQ(Version::getValue(manifestFilePath, "version"), version); |
| 52 | EXPECT_EQ(Version::getValue(manifestFilePath, "purpose"), purpose); |
| 53 | } |
| 54 | |
| 55 | /** @brief Make sure we correctly get the Id from getId()*/ |
| 56 | TEST_F(VersionTest, TestGetId) |
| 57 | { |
Gunnar Mills | 01d55c3 | 2017-04-20 10:52:15 -0500 | [diff] [blame] | 58 | auto version = "test-id"; |
Saqib Khan | 26a960d | 2017-09-19 14:23:28 -0500 | [diff] [blame] | 59 | unsigned char digest[SHA512_DIGEST_LENGTH]; |
| 60 | SHA512_CTX ctx; |
| 61 | SHA512_Init(&ctx); |
| 62 | SHA512_Update(&ctx, version, strlen(version)); |
| 63 | SHA512_Final(digest, &ctx); |
Adriana Kobylak | 2285fe0 | 2018-02-27 15:36:59 -0600 | [diff] [blame^] | 64 | char mdString[SHA512_DIGEST_LENGTH * 2 + 1]; |
Saqib Khan | 26a960d | 2017-09-19 14:23:28 -0500 | [diff] [blame] | 65 | for (int i = 0; i < SHA512_DIGEST_LENGTH; i++) |
| 66 | { |
Adriana Kobylak | 2285fe0 | 2018-02-27 15:36:59 -0600 | [diff] [blame^] | 67 | snprintf(&mdString[i * 2], 3, "%02x", (unsigned int)digest[i]); |
Saqib Khan | 26a960d | 2017-09-19 14:23:28 -0500 | [diff] [blame] | 68 | } |
| 69 | std::string hexId = std::string(mdString); |
| 70 | hexId = hexId.substr(0, 8); |
| 71 | EXPECT_EQ(Version::getId(version), hexId); |
Gunnar Mills | 01d55c3 | 2017-04-20 10:52:15 -0500 | [diff] [blame] | 72 | } |