Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 32dc6dd5 authored by Daniel Zheng's avatar Daniel Zheng
Browse files

Adding correct task formed test 

Test: test passes
Change-Id: I07b591a2c72d207d794dd3eb044a5fdfc048eb70
parent 935ee1f9

fastboot/task_test.cpp

+15 −0
Original line number Diff line number Diff line
@@ -16,6 +16,7 @@ 


#include "task.h"

#include "fastboot.h"

#include "fastboot_driver_mock.h"



#include <gtest/gtest.h>

#include <fstream>
@@ -24,6 +25,7 @@ 
#include <unordered_map>

#include "android-base/strings.h"

using android::base::Split;

using testing::_;



class ParseTest : public ::testing ::Test {

  protected:
@@ -121,3 +123,16 @@ TEST_F(ParseTest, BAD_FASTBOOT_INFO_INPUT) { 
    ASSERT_EQ(ParseCommand(fp.get(), "erase dtbo dtbo"), nullptr);

    ASSERT_EQ(ParseCommand(fp.get(), "wipe this"), nullptr);

}



TEST_F(ParseTest, CORRECT_TASK_FORMED) {

    std::vector<std::string> commands = {"flash dtbo", "flash --slot-other system system_other.img",

                                         "reboot bootloader", "update-super", "erase cache"};

    std::vector<std::unique_ptr<Task>> tasks = collectTasks(fp.get(), commands);



    ASSERT_TRUE(tasks[0]->AsFlashTask());

    ASSERT_TRUE(tasks[0]->AsFlashTask());

    ASSERT_TRUE(tasks[1]->AsFlashTask());

    ASSERT_TRUE(tasks[2]->AsRebootTask());

    ASSERT_TRUE(tasks[3]->AsUpdateSuperTask());

    ASSERT_TRUE(tasks[4]->AsWipeTask());

}