16
16
* limitations under the License.
17
17
*/
18
18
19
- #include < lib/dnssd/minimal_mdns/records/ResourceRecord.h>
20
- #include < lib/support/UnitTestRegistration.h>
19
+ #include < gtest/gtest.h>
21
20
22
- #include < nlunit-test .h>
21
+ #include < lib/dnssd/minimal_mdns/records/ResourceRecord .h>
23
22
24
23
namespace {
25
24
@@ -41,7 +40,7 @@ class FakeResourceRecord : public ResourceRecord
41
40
const char * mData ;
42
41
};
43
42
44
- void CanWriteSimpleRecord (nlTestSuite * inSuite, void * inContext )
43
+ TEST (TestResourceRecord, CanWriteSimpleRecord )
45
44
{
46
45
uint8_t headerBuffer[HeaderRef::kSizeBytes ];
47
46
uint8_t dataBuffer[128 ];
@@ -68,15 +67,15 @@ void CanWriteSimpleRecord(nlTestSuite * inSuite, void * inContext)
68
67
' s' , ' o' , ' m' , ' e' , ' d' , ' a' , ' t' , ' a' ,
69
68
};
70
69
71
- NL_TEST_ASSERT (inSuite, record.Append (header, ResourceType::kAnswer , writer));
72
- NL_TEST_ASSERT (inSuite, header.GetAnswerCount () == 1 );
73
- NL_TEST_ASSERT (inSuite, header.GetAuthorityCount () == 0 );
74
- NL_TEST_ASSERT (inSuite, header.GetAdditionalCount () == 0 );
75
- NL_TEST_ASSERT (inSuite, output.Needed () == sizeof (expectedOutput));
76
- NL_TEST_ASSERT (inSuite, memcmp (dataBuffer, expectedOutput, sizeof (expectedOutput)) == 0 );
70
+ EXPECT_TRUE ( record.Append (header, ResourceType::kAnswer , writer));
71
+ EXPECT_EQ ( header.GetAnswerCount (), 1 );
72
+ EXPECT_EQ ( header.GetAuthorityCount (), 0 );
73
+ EXPECT_EQ ( header.GetAdditionalCount (), 0 );
74
+ EXPECT_EQ ( output.Needed (), sizeof (expectedOutput));
75
+ EXPECT_EQ ( memcmp (dataBuffer, expectedOutput, sizeof (expectedOutput)), 0 );
77
76
}
78
77
79
- void CanWriteMultipleRecords (nlTestSuite * inSuite, void * inContext )
78
+ TEST (TestResourceRecord, CanWriteMultipleRecords )
80
79
{
81
80
uint8_t headerBuffer[HeaderRef::kSizeBytes ];
82
81
uint8_t dataBuffer[128 ];
@@ -118,26 +117,26 @@ void CanWriteMultipleRecords(nlTestSuite * inSuite, void * inContext)
118
117
' x' , ' y' , ' z' ,
119
118
};
120
119
121
- NL_TEST_ASSERT (inSuite, record1.Append (header, ResourceType::kAnswer , writer));
122
- NL_TEST_ASSERT (inSuite, header.GetAnswerCount () == 1 );
123
- NL_TEST_ASSERT (inSuite, header.GetAuthorityCount () == 0 );
124
- NL_TEST_ASSERT (inSuite, header.GetAdditionalCount () == 0 );
120
+ EXPECT_TRUE ( record1.Append (header, ResourceType::kAnswer , writer));
121
+ EXPECT_EQ ( header.GetAnswerCount (), 1 );
122
+ EXPECT_EQ ( header.GetAuthorityCount (), 0 );
123
+ EXPECT_EQ ( header.GetAdditionalCount (), 0 );
125
124
126
- NL_TEST_ASSERT (inSuite, record2.Append (header, ResourceType::kAuthority , writer));
127
- NL_TEST_ASSERT (inSuite, header.GetAnswerCount () == 1 );
128
- NL_TEST_ASSERT (inSuite, header.GetAuthorityCount () == 1 );
129
- NL_TEST_ASSERT (inSuite, header.GetAdditionalCount () == 0 );
125
+ EXPECT_TRUE ( record2.Append (header, ResourceType::kAuthority , writer));
126
+ EXPECT_EQ ( header.GetAnswerCount (), 1 );
127
+ EXPECT_EQ ( header.GetAuthorityCount (), 1 );
128
+ EXPECT_EQ ( header.GetAdditionalCount (), 0 );
130
129
131
- NL_TEST_ASSERT (inSuite, record3.Append (header, ResourceType::kAdditional , writer));
132
- NL_TEST_ASSERT (inSuite, header.GetAnswerCount () == 1 );
133
- NL_TEST_ASSERT (inSuite, header.GetAuthorityCount () == 1 );
134
- NL_TEST_ASSERT (inSuite, header.GetAdditionalCount () == 1 );
130
+ EXPECT_TRUE ( record3.Append (header, ResourceType::kAdditional , writer));
131
+ EXPECT_EQ ( header.GetAnswerCount (), 1 );
132
+ EXPECT_EQ ( header.GetAuthorityCount (), 1 );
133
+ EXPECT_EQ ( header.GetAdditionalCount (), 1 );
135
134
136
- NL_TEST_ASSERT (inSuite, output.Needed () == sizeof (expectedOutput));
137
- NL_TEST_ASSERT (inSuite, memcmp (dataBuffer, expectedOutput, sizeof (expectedOutput)) == 0 );
135
+ EXPECT_EQ ( output.Needed (), sizeof (expectedOutput));
136
+ EXPECT_EQ ( memcmp (dataBuffer, expectedOutput, sizeof (expectedOutput)), 0 );
138
137
}
139
138
140
- void RecordOrderIsEnforced (nlTestSuite * inSuite, void * inContext )
139
+ TEST (TestResourceRecord, RecordOrderIsEnforced )
141
140
{
142
141
uint8_t headerBuffer[HeaderRef::kSizeBytes ];
143
142
uint8_t dataBuffer[128 ];
@@ -151,15 +150,15 @@ void RecordOrderIsEnforced(nlTestSuite * inSuite, void * inContext)
151
150
152
151
header.Clear ();
153
152
header.SetAuthorityCount (1 );
154
- NL_TEST_ASSERT (inSuite, record.Append (header, ResourceType::kAnswer , writer) == false );
153
+ EXPECT_EQ ( record.Append (header, ResourceType::kAnswer , writer), false );
155
154
156
155
header.Clear ();
157
156
header.SetAdditionalCount (1 );
158
- NL_TEST_ASSERT (inSuite, record.Append (header, ResourceType::kAnswer , writer) == false );
159
- NL_TEST_ASSERT (inSuite, record.Append (header, ResourceType::kAuthority , writer) == false );
157
+ EXPECT_EQ ( record.Append (header, ResourceType::kAnswer , writer), false );
158
+ EXPECT_EQ ( record.Append (header, ResourceType::kAuthority , writer), false );
160
159
}
161
160
162
- void ErrorsOutOnSmallBuffers (nlTestSuite * inSuite, void * inContext )
161
+ TEST (TestResourceRecord, ErrorsOutOnSmallBuffers )
163
162
{
164
163
uint8_t headerBuffer[HeaderRef::kSizeBytes ];
165
164
uint8_t dataBuffer[123 ];
@@ -191,23 +190,23 @@ void ErrorsOutOnSmallBuffers(nlTestSuite * inSuite, void * inContext)
191
190
BufferWriter output (dataBuffer, i);
192
191
RecordWriter writer (&output);
193
192
194
- NL_TEST_ASSERT (inSuite, record.Append (header, ResourceType::kAnswer , writer) == false );
193
+ EXPECT_EQ ( record.Append (header, ResourceType::kAnswer , writer), false );
195
194
196
195
// header untouched
197
- NL_TEST_ASSERT (inSuite, memcmp (headerBuffer, clearHeader, HeaderRef::kSizeBytes ) == 0 );
196
+ EXPECT_EQ ( memcmp (headerBuffer, clearHeader, HeaderRef::kSizeBytes ), 0 );
198
197
}
199
198
200
199
memset (dataBuffer, 0 , sizeof (dataBuffer));
201
200
BufferWriter output (dataBuffer, sizeof (expectedOutput));
202
201
RecordWriter writer (&output);
203
202
204
- NL_TEST_ASSERT (inSuite, record.Append (header, ResourceType::kAnswer , writer));
205
- NL_TEST_ASSERT (inSuite, output.Needed () == sizeof (expectedOutput));
206
- NL_TEST_ASSERT (inSuite, memcmp (dataBuffer, expectedOutput, sizeof (expectedOutput)) == 0 );
207
- NL_TEST_ASSERT (inSuite, memcmp (headerBuffer, clearHeader, HeaderRef::kSizeBytes ) != 0 );
203
+ EXPECT_TRUE ( record.Append (header, ResourceType::kAnswer , writer));
204
+ EXPECT_EQ ( output.Needed (), sizeof (expectedOutput));
205
+ EXPECT_EQ ( memcmp (dataBuffer, expectedOutput, sizeof (expectedOutput)), 0 );
206
+ EXPECT_NE ( memcmp (headerBuffer, clearHeader, HeaderRef::kSizeBytes ), 0 );
208
207
}
209
208
210
- void RecordCount (nlTestSuite * inSuite, void * inContext )
209
+ TEST (TestResourceRecord, RecordCount )
211
210
{
212
211
constexpr int kAppendCount = 10 ;
213
212
uint8_t headerBuffer[HeaderRef::kSizeBytes ];
@@ -223,69 +222,49 @@ void RecordCount(nlTestSuite * inSuite, void * inContext)
223
222
BufferWriter output (dataBuffer, sizeof (dataBuffer));
224
223
RecordWriter writer (&output);
225
224
226
- NL_TEST_ASSERT (inSuite, record.Append (header, ResourceType::kAnswer , writer));
227
- NL_TEST_ASSERT (inSuite, header.GetAnswerCount () == i + 1 );
228
- NL_TEST_ASSERT (inSuite, header.GetAuthorityCount () == 0 );
229
- NL_TEST_ASSERT (inSuite, header.GetAdditionalCount () == 0 );
225
+ EXPECT_TRUE ( record.Append (header, ResourceType::kAnswer , writer));
226
+ EXPECT_EQ ( header.GetAnswerCount (), i + 1 );
227
+ EXPECT_EQ ( header.GetAuthorityCount (), 0 );
228
+ EXPECT_EQ ( header.GetAdditionalCount (), 0 );
230
229
}
231
230
232
231
for (int i = 0 ; i < kAppendCount ; i++)
233
232
{
234
233
BufferWriter output (dataBuffer, sizeof (dataBuffer));
235
234
RecordWriter writer (&output);
236
235
237
- NL_TEST_ASSERT (inSuite, record.Append (header, ResourceType::kAuthority , writer));
238
- NL_TEST_ASSERT (inSuite, header.GetAnswerCount () == kAppendCount );
239
- NL_TEST_ASSERT (inSuite, header.GetAuthorityCount () == i + 1 );
240
- NL_TEST_ASSERT (inSuite, header.GetAdditionalCount () == 0 );
236
+ EXPECT_TRUE ( record.Append (header, ResourceType::kAuthority , writer));
237
+ EXPECT_EQ ( header.GetAnswerCount (), kAppendCount );
238
+ EXPECT_EQ ( header.GetAuthorityCount (), i + 1 );
239
+ EXPECT_EQ ( header.GetAdditionalCount (), 0 );
241
240
}
242
241
243
242
for (int i = 0 ; i < kAppendCount ; i++)
244
243
{
245
244
BufferWriter output (dataBuffer, sizeof (dataBuffer));
246
245
RecordWriter writer (&output);
247
246
248
- NL_TEST_ASSERT (inSuite, record.Append (header, ResourceType::kAdditional , writer));
249
- NL_TEST_ASSERT (inSuite, header.GetAnswerCount () == kAppendCount );
250
- NL_TEST_ASSERT (inSuite, header.GetAuthorityCount () == kAppendCount );
251
- NL_TEST_ASSERT (inSuite, header.GetAdditionalCount () == i + 1 );
247
+ EXPECT_TRUE ( record.Append (header, ResourceType::kAdditional , writer));
248
+ EXPECT_EQ ( header.GetAnswerCount (), kAppendCount );
249
+ EXPECT_EQ ( header.GetAuthorityCount (), kAppendCount );
250
+ EXPECT_EQ ( header.GetAdditionalCount (), i + 1 );
252
251
}
253
252
}
254
- void CacheFlushBit (nlTestSuite * inSuite, void * inContext )
253
+ TEST (TestResourceRecord, CacheFlushBit )
255
254
{
256
255
FakeResourceRecord record (" somedata" );
257
256
// No cache flush bit by default.
258
- NL_TEST_ASSERT (inSuite, record.GetClass () == QClass::IN);
259
- NL_TEST_ASSERT (inSuite, record.GetCacheFlush () == false );
257
+ EXPECT_EQ ( record.GetClass (), QClass::IN);
258
+ EXPECT_EQ ( record.GetCacheFlush (), false );
260
259
261
260
// Check we can set flush bit and the class marker reflects that.
262
261
record.SetCacheFlush (true );
263
- NL_TEST_ASSERT (inSuite, record.GetClass () == QClass::IN_FLUSH);
264
- NL_TEST_ASSERT (inSuite, record.GetCacheFlush () == true );
262
+ EXPECT_EQ ( record.GetClass (), QClass::IN_FLUSH);
263
+ EXPECT_EQ ( record.GetCacheFlush (), true );
265
264
266
265
// Check we can unset.
267
266
record.SetCacheFlush (false );
268
- NL_TEST_ASSERT (inSuite, record.GetClass () == QClass::IN);
269
- NL_TEST_ASSERT (inSuite, record.GetCacheFlush () == false );
267
+ EXPECT_EQ ( record.GetClass (), QClass::IN);
268
+ EXPECT_EQ ( record.GetCacheFlush (), false );
270
269
}
271
-
272
- const nlTest sTests [] = {
273
- NL_TEST_DEF (" CanWriteSimpleRecord" , CanWriteSimpleRecord), //
274
- NL_TEST_DEF (" CanWriteMultipleRecords" , CanWriteMultipleRecords), //
275
- NL_TEST_DEF (" RecordOrderIsEnforced" , RecordOrderIsEnforced), //
276
- NL_TEST_DEF (" ErrorsOutOnSmallBuffers" , ErrorsOutOnSmallBuffers), //
277
- NL_TEST_DEF (" RecordCount" , RecordCount), //
278
- NL_TEST_DEF (" CacheFlushBit" , CacheFlushBit), //
279
- NL_TEST_SENTINEL () //
280
- };
281
-
282
270
} // namespace
283
-
284
- int TestResourceRecord ()
285
- {
286
- nlTestSuite theSuite = { " ResourceRecord" , sTests , nullptr , nullptr };
287
- nlTestRunner (&theSuite, nullptr );
288
- return nlTestRunnerStats (&theSuite);
289
- }
290
-
291
- CHIP_REGISTER_TEST_SUITE (TestResourceRecord)
0 commit comments