uboot.h

#pragma once

#include <iostream>
#include <functional>
#include "command.h"

#ifdef TESTING
#include <gtest/gtest.h>
#endif

class UBoot
{
private:
    static const std::string EP200_1GB_BINARY;
    static const std::string EP200_4GB_BINARY;
    static const std::string TV100_BINARY;
    static const std::string DTD200_BINARY;

public:
    UBoot();
    virtual ~UBoot();
    static bool updateUBoot(const std::string &zipLocation);
    static bool getCurrentVersion(std::string &current_version);

private:
    static inline std::function<std::string(std::string)> mCmdFunc = Command::GetStdoutFromCommand;
    static bool getCurrentBootPartition(std::string &current_part);
    static bool getBinaryPath(const std::string &zipLocation, std::string &binaryPath);
    static unsigned long int getTotalKBMemory();

#ifdef TESTING
public:
    friend class UpdateUBoot_test;
    FRIEND_TEST(UpdateUBoot_test,getCurrentBootPartition0);
    FRIEND_TEST(UpdateUBoot_test,getCurrentBootPartition1);
    FRIEND_TEST(UpdateUBoot_test,getCurrentBootPartitionNotValidPattern);
    FRIEND_TEST(UpdateUBoot_test,getCurrentBootPartitionNotValidPattern2);
    FRIEND_TEST(UpdateUBoot_test,getCurrentBootPartitionEmptyResponse);
    FRIEND_TEST(UpdateUBoot_test,getCurrentVersionOK);
    FRIEND_TEST(UpdateUBoot_test,getCurrentVersionNotValidPattern);
    FRIEND_TEST(UpdateUBoot_test,getCurrentVersionEmptyResponse);
    FRIEND_TEST(UpdateUBoot_test,zipError);
    FRIEND_TEST(UpdateUBoot_test,updateUBootVersionsMatch);
    FRIEND_TEST(UpdateUBoot_test,updateUBootVersionError);
#endif //TESTING
};