1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chromeos/ash/components/data_migration/rts_receiver.h"
#include <cstdint>
#include <vector>
#include "base/functional/callback_helpers.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "chromeos/ash/components/data_migration/constants.h"
#include "chromeos/ash/components/data_migration/pending_file_transfer_queue.h"
#include "chromeos/ash/components/data_migration/testing/connection_barrier.h"
#include "chromeos/ash/components/data_migration/testing/fake_nearby_connections.h"
#include "chromeos/ash/components/data_migration/testing/fake_nearby_process_manager.h"
#include "chromeos/ash/components/nearby/common/connections_manager/nearby_connections_manager.h"
#include "chromeos/ash/components/nearby/common/connections_manager/nearby_connections_manager_impl.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace data_migration {
namespace {
constexpr char kRemoteEndpointId[] = "test-remote-endpoint";
class RtsReceiverTest : public testing::Test {
protected:
RtsReceiverTest()
: nearby_process_manager_(kRemoteEndpointId),
nearby_connections_manager_(&nearby_process_manager_, kServiceId) {}
void SetUp() override {
ConnectionBarrier connection_barrier(&nearby_connections_manager_);
NearbyConnection* nearby_connection = connection_barrier.Wait();
ASSERT_TRUE(nearby_connection);
rts_receiver_.emplace(nearby_connection, &pending_file_transfer_queue_);
}
base::test::TaskEnvironment task_environment_;
FakeNearbyProcessManager nearby_process_manager_;
NearbyConnectionsManagerImpl nearby_connections_manager_;
PendingFileTransferQueue pending_file_transfer_queue_;
std::optional<RtsReceiver> rts_receiver_;
};
TEST_F(RtsReceiverTest, PushesPendingFilesToQueue) {
ASSERT_TRUE(
nearby_process_manager_.fake_nearby_connections().SendBytesPayload(
/*payload_id=*/1, /*file payload id*/ "100"));
ASSERT_TRUE(
nearby_process_manager_.fake_nearby_connections().SendBytesPayload(
/*payload_id=*/2, /*file payload id*/ "200"));
ASSERT_TRUE(
nearby_process_manager_.fake_nearby_connections().SendBytesPayload(
/*payload_id=*/3, /*file payload id*/ "300"));
base::test::TestFuture<int64_t> future;
pending_file_transfer_queue_.Pop(future.GetCallback());
EXPECT_EQ(future.Get<0>(), 100);
future.Clear();
pending_file_transfer_queue_.Pop(future.GetCallback());
EXPECT_EQ(future.Get<0>(), 200);
future.Clear();
pending_file_transfer_queue_.Pop(future.GetCallback());
EXPECT_EQ(future.Get<0>(), 300);
future.Clear();
pending_file_transfer_queue_.Pop(future.GetCallback());
task_environment_.RunUntilIdle();
EXPECT_FALSE(future.IsReady());
}
TEST_F(RtsReceiverTest, IgnoresMalformedBytes) {
ASSERT_TRUE(
nearby_process_manager_.fake_nearby_connections().SendBytesPayload(
/*payload_id=*/1, "invalid-file-payload-id"));
base::test::TestFuture<int64_t> future;
pending_file_transfer_queue_.Pop(future.GetCallback());
task_environment_.RunUntilIdle();
EXPECT_FALSE(future.IsReady());
// Should keep receiving after receiving a malformed RTS.
ASSERT_TRUE(
nearby_process_manager_.fake_nearby_connections().SendBytesPayload(
/*payload_id=*/2, /*file payload id*/ "200"));
EXPECT_EQ(future.Get<0>(), 200);
}
TEST_F(RtsReceiverTest, HandlesDisconnectFromRemoteDevice) {
ASSERT_TRUE(
nearby_process_manager_.fake_nearby_connections().SendBytesPayload(
/*payload_id=*/1, /*file payload id*/ "100"));
base::test::TestFuture<int64_t> future;
pending_file_transfer_queue_.Pop(future.GetCallback());
EXPECT_EQ(future.Get<0>(), 100);
ASSERT_TRUE(nearby_process_manager_.fake_nearby_connections()
.SimulateRemoteDisconnect());
future.Clear();
pending_file_transfer_queue_.Pop(future.GetCallback());
task_environment_.RunUntilIdle();
EXPECT_FALSE(future.IsReady());
}
} // namespace
} // namespace data_migration
|
