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

Commit a6d6714d authored by Daniel Zheng's avatar Daniel Zheng Committed by Automerger Merge Worker
Browse files

Merge "Added Mock function call tests" am: 71c918aa 

Original change: https://android-review.googlesource.com/c/platform/system/core/+/2530985



Change-Id: I08f20227925e58e7d920ffdd69d903c164cb17e5
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents dceb71de 71c918aa

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();

    }

}