tequilaOS/platform_system_core
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge "Added Mock function call tests"

Browse source
This commit is contained in:
Daniel Zheng 2023-04-26 19:57:52 +00:00 committed by Gerrit Code Review
commit 71c918aa59
2 changed files with 17 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
fastboot/fastboot_driver_mock.h
View file

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

16
fastboot/task_test.cpp
View file

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