115 lines
3.3 KiB
C++
115 lines
3.3 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
* You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
/*
|
|
* Test that sync messages preempt other messages in the expected way.
|
|
*/
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "mozilla/_ipdltest/IPDLUnitTest.h"
|
|
#include "mozilla/_ipdltest/PTestUrgencyChild.h"
|
|
#include "mozilla/_ipdltest/PTestUrgencyParent.h"
|
|
|
|
using namespace mozilla::ipc;
|
|
|
|
namespace mozilla::_ipdltest {
|
|
|
|
enum {
|
|
kFirstTestBegin = 1,
|
|
kFirstTestGotReply,
|
|
kSecondTestBegin,
|
|
kSecondTestGotReply,
|
|
};
|
|
|
|
class TestUrgencyParent : public PTestUrgencyParent {
|
|
NS_INLINE_DECL_REFCOUNTING(TestUrgencyParent, override)
|
|
private:
|
|
IPCResult RecvTest1(uint32_t* value) final override {
|
|
EXPECT_TRUE(SendReply1(value));
|
|
EXPECT_EQ(*value, (uint32_t)99) << "unexpected value";
|
|
return IPC_OK();
|
|
}
|
|
|
|
IPCResult RecvTest2() final override {
|
|
uint32_t value;
|
|
inreply_ = true;
|
|
EXPECT_TRUE(SendReply2(&value));
|
|
inreply_ = false;
|
|
EXPECT_EQ(value, (uint32_t)500) << "unexpected value";
|
|
return IPC_OK();
|
|
}
|
|
|
|
IPCResult RecvTest3(uint32_t* value) final override {
|
|
EXPECT_FALSE(inreply_) << "nested non-urgent on top of urgent message";
|
|
*value = 1000;
|
|
return IPC_OK();
|
|
}
|
|
|
|
IPCResult RecvFinalTest_Begin() final override { return IPC_OK(); }
|
|
|
|
~TestUrgencyParent() = default;
|
|
|
|
bool inreply_ = false;
|
|
};
|
|
|
|
class TestUrgencyChild : public PTestUrgencyChild {
|
|
NS_INLINE_DECL_REFCOUNTING(TestUrgencyChild, override)
|
|
private:
|
|
IPCResult RecvStart() final override {
|
|
uint32_t result;
|
|
|
|
// Send a synchronous message, expect to get an urgent message while
|
|
// blocked.
|
|
test_ = kFirstTestBegin;
|
|
EXPECT_TRUE(SendTest1(&result));
|
|
EXPECT_EQ(result, (uint32_t)99) << "wrong value from SendTest1";
|
|
EXPECT_EQ(test_, kFirstTestGotReply)
|
|
<< "never received first urgent message";
|
|
|
|
// Initiate the next test by sending an asynchronous message, then becoming
|
|
// blocked. This tests that the urgent message is still delivered properly,
|
|
// and that the parent does not try to service the sync
|
|
test_ = kSecondTestBegin;
|
|
EXPECT_TRUE(SendTest2());
|
|
EXPECT_TRUE(SendTest3(&result));
|
|
EXPECT_EQ(test_, kSecondTestGotReply)
|
|
<< "never received second urgent message";
|
|
EXPECT_EQ(result, (uint32_t)1000) << "wrong value from SendTest3";
|
|
|
|
EXPECT_TRUE(SendFinalTest_Begin());
|
|
|
|
Close();
|
|
|
|
return IPC_OK();
|
|
}
|
|
|
|
IPCResult RecvReply1(uint32_t* reply) final override {
|
|
EXPECT_EQ(test_, kFirstTestBegin) << "wrong test state in RecvReply1";
|
|
|
|
*reply = 99;
|
|
test_ = kFirstTestGotReply;
|
|
return IPC_OK();
|
|
}
|
|
|
|
IPCResult RecvReply2(uint32_t* reply) final override {
|
|
EXPECT_EQ(test_, kSecondTestBegin) << "wrong test state in RecvReply2";
|
|
|
|
*reply = 500;
|
|
test_ = kSecondTestGotReply;
|
|
return IPC_OK();
|
|
}
|
|
|
|
~TestUrgencyChild() = default;
|
|
|
|
uint32_t test_ = 0;
|
|
};
|
|
|
|
// Only run cross-process because we need to send nested sync messages (this can
|
|
// only be done from the main thread).
|
|
IPDL_TEST_ON(CROSSPROCESS, TestUrgency) { EXPECT_TRUE(mActor->SendStart()); }
|
|
|
|
} // namespace mozilla::_ipdltest
|