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

Commit 71c918aa authored by Daniel Zheng's avatar Daniel Zheng Committed by Gerrit Code Review
Browse files

Merge "Added Mock function call tests"

parents 3b41a5b4 0d2d631d

fastboot/fastboot_driver_mock.h

+1 −2
Original line number Diff line number Diff line
@@ -22,8 +22,7 @@ namespace fastboot { 


class MockFastbootDriver : public IFastBootDriver {

  public:

    MOCK_METHOD(RetCode, FlashPartition,

                (const std::string& partition, android::base::borrowed_fd fd, uint32_t sz),

    MOCK_METHOD(RetCode, FlashPartition, (const std::string&, android::base::borrowed_fd, uint32_t),

                (override));

    MOCK_METHOD(RetCode, DeletePartition, (const std::string&), (override));

    MOCK_METHOD(RetCode, Reboot, (std::string*, std::vector<std::string>*), (override));

fastboot/task_test.cpp

+16 −0
Original line number Diff line number Diff line
@@ -136,3 +136,19 @@ TEST_F(ParseTest, CorrectTaskFormed) { 
    ASSERT_TRUE(tasks[3]->AsUpdateSuperTask());

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

}



TEST_F(ParseTest, CorrectDriverCalls) {

    fastboot::MockFastbootDriver fb;

    fp->fb = &fb;



    EXPECT_CALL(fb, RebootTo(_, _, _)).Times(1);

    EXPECT_CALL(fb, Reboot(_, _)).Times(1);

    EXPECT_CALL(fb, WaitForDisconnect()).Times(2);



    std::vector<std::string> commands = {"reboot bootloader", "reboot"};

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



    for (auto& task : tasks) {

        task->Run();

    }

}