This version works where both sides are the same message, messageview or messagemut type, but not any mix of them (e.g. cannot compare a message against its corresponding view). PiperOrigin-RevId: 671091099pull/18099/head
parent
e1dbeb0e61
commit
5d341470da
7 changed files with 247 additions and 15 deletions
@ -0,0 +1,35 @@ |
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2024 Google LLC. All rights reserved.
|
||||
//
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file or at
|
||||
// https://developers.google.com/open-source/licenses/bsd
|
||||
|
||||
use googletest::description::Description; |
||||
use googletest::matcher::{Matcher, MatcherBase, MatcherResult}; |
||||
use protobuf::__internal::MatcherEq; |
||||
|
||||
#[derive(MatcherBase)] |
||||
pub struct MessageMatcher<T: MatcherEq> { |
||||
expected: T, |
||||
} |
||||
|
||||
impl<T> Matcher<&T> for MessageMatcher<T> |
||||
where |
||||
T: MatcherEq, |
||||
{ |
||||
fn matches(&self, actual: &T) -> MatcherResult { |
||||
actual.matches(&self.expected).into() |
||||
} |
||||
|
||||
fn describe(&self, matcher_result: MatcherResult) -> Description { |
||||
match matcher_result { |
||||
MatcherResult::Match => format!("is equal to {:?}", self.expected).into(), |
||||
MatcherResult::NoMatch => format!("is not equal to {:?}", self.expected).into(), |
||||
} |
||||
} |
||||
} |
||||
|
||||
pub fn proto_eq<T: MatcherEq>(expected: T) -> MessageMatcher<T> { |
||||
MessageMatcher { expected } |
||||
} |
@ -0,0 +1,59 @@ |
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2024 Google LLC. All rights reserved.
|
||||
//
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file or at
|
||||
// https://developers.google.com/open-source/licenses/bsd
|
||||
|
||||
use googletest::prelude::*; |
||||
use paste::paste; |
||||
use protobuf_gtest_matchers::proto_eq; |
||||
use unittest_edition_rust_proto::TestAllTypes as TestAllTypesEditions; |
||||
use unittest_proto3_rust_proto::TestAllTypes as TestAllTypesProto3; |
||||
use unittest_rust_proto::TestAllTypes as TestAllTypesProto2; |
||||
|
||||
macro_rules! generate_eq_msgs_tests { |
||||
($(($type: ident, $name_ext: ident)),*) => { |
||||
paste! {$( |
||||
#[gtest] |
||||
fn [<expect_eq_msgs_ $name_ext>]() { |
||||
let mut msg = [< $type >]::new(); |
||||
let mut msg2 = [< $type >]::new(); |
||||
msg.set_optional_int32(1); |
||||
msg2.set_optional_int32(1); |
||||
assert_that!(&msg.as_view(), proto_eq(msg2.as_view())); |
||||
assert_that!(&msg.as_mut(), proto_eq(msg2.as_mut())); |
||||
assert_that!(msg, proto_eq(msg2)); |
||||
} |
||||
)*} |
||||
} |
||||
} |
||||
|
||||
macro_rules! generate_not_eq_msgs_tests { |
||||
($(($type: ident, $name_ext: ident)),*) => { |
||||
paste! {$( |
||||
#[gtest] |
||||
fn [<expect_not_eq_msgs_ $name_ext>]() { |
||||
let mut msg = [< $type >]::new(); |
||||
let mut msg2 = [< $type >]::new(); |
||||
msg.set_optional_int32(1); |
||||
msg2.set_optional_int32(0); |
||||
assert_that!(&msg.as_view(), not(proto_eq(msg2.as_view()))); |
||||
assert_that!(&msg.as_mut(), not(proto_eq(msg2.as_mut()))); |
||||
assert_that!(&msg, not(proto_eq(msg2))); |
||||
} |
||||
)*} |
||||
} |
||||
} |
||||
|
||||
generate_eq_msgs_tests!( |
||||
(TestAllTypesEditions, editions), |
||||
(TestAllTypesProto3, proto3), |
||||
(TestAllTypesProto2, proto2) |
||||
); |
||||
|
||||
generate_not_eq_msgs_tests!( |
||||
(TestAllTypesEditions, editions), |
||||
(TestAllTypesProto3, proto3), |
||||
(TestAllTypesProto2, proto2) |
||||
); |
Loading…
Reference in new issue