mirror of
https://github.com/wassname/ray.git
synced 2026-07-20 12:40:20 +08:00
31 lines
1.1 KiB
C++
31 lines
1.1 KiB
C++
#include <cstring>
|
|
#include <string>
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "queue/message.h"
|
|
using namespace ray;
|
|
using namespace ray::streaming;
|
|
|
|
TEST(ProtoBufTest, MessageCommonTest) {
|
|
JobID job_id = JobID::FromInt(0);
|
|
TaskID task_id = TaskID::ForDriverTask(job_id);
|
|
ray::ActorID actor_id = ray::ActorID::Of(job_id, task_id, 0);
|
|
ray::ActorID peer_actor_id = ray::ActorID::Of(job_id, task_id, 1);
|
|
ObjectID queue_id = ray::ObjectID::FromRandom();
|
|
|
|
uint8_t data[128];
|
|
std::shared_ptr<LocalMemoryBuffer> buffer =
|
|
std::make_shared<LocalMemoryBuffer>(data, 128, true);
|
|
DataMessage msg(actor_id, peer_actor_id, queue_id, 100, 1000, 2000, buffer, true);
|
|
std::unique_ptr<LocalMemoryBuffer> serilized_buffer = msg.ToBytes();
|
|
std::shared_ptr<DataMessage> msg2 = DataMessage::FromBytes(serilized_buffer->Data());
|
|
EXPECT_EQ(msg.ActorId(), msg2->ActorId());
|
|
EXPECT_EQ(msg.PeerActorId(), msg2->PeerActorId());
|
|
EXPECT_EQ(msg.QueueId(), msg2->QueueId());
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|