@@ -39,6 +39,11 @@ void* my_gballoc_malloc(size_t size)
39
39
return malloc (size );
40
40
}
41
41
42
+ void * my_gballoc_calloc (size_t nmemb , size_t size )
43
+ {
44
+ return calloc (nmemb , size );
45
+ }
46
+
42
47
void * my_gballoc_realloc (void * ptr , size_t size )
43
48
{
44
49
return realloc (ptr , size );
@@ -140,6 +145,7 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
140
145
ASSERT_ARE_EQUAL (int , 0 , result );
141
146
142
147
REGISTER_GLOBAL_MOCK_HOOK (gballoc_malloc , my_gballoc_malloc );
148
+ REGISTER_GLOBAL_MOCK_HOOK (gballoc_calloc , my_gballoc_calloc );
143
149
REGISTER_GLOBAL_MOCK_FAIL_RETURN (gballoc_malloc , NULL );
144
150
REGISTER_GLOBAL_MOCK_HOOK (gballoc_free , my_gballoc_free );
145
151
@@ -187,7 +193,7 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
187
193
TEST_FUNCTION (dns_resolver__is_complete_repeated_call__succeeds )
188
194
{
189
195
///arrange
190
- DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , NULL );
196
+ DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , 53 , NULL );
191
197
// We're calling this twice
192
198
bool result = dns_resolver_is_lookup_complete (dns );
193
199
uint32_t ipv4 = dns_resolver_get_ipv4 (dns );
@@ -220,7 +226,7 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
220
226
{
221
227
///arrange
222
228
bool result ;
223
- DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , NULL );
229
+ DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , 53 , NULL );
224
230
umock_c_reset_all_calls ();
225
231
STRICT_EXPECTED_CALL (getaddrinfo (IGNORED_PTR_ARG , IGNORED_PTR_ARG , IGNORED_PTR_ARG , IGNORED_PTR_ARG ));
226
232
@@ -240,7 +246,7 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
240
246
///arrange
241
247
bool result ;
242
248
uint32_t ipv4 ;
243
- DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , NULL );
249
+ DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , 53 , NULL );
244
250
umock_c_reset_all_calls ();
245
251
STRICT_EXPECTED_CALL (getaddrinfo (IGNORED_PTR_ARG , IGNORED_PTR_ARG , IGNORED_PTR_ARG , IGNORED_PTR_ARG ));
246
252
result = dns_resolver_is_lookup_complete (dns );
@@ -261,7 +267,7 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
261
267
{
262
268
///arrange
263
269
bool result ;
264
- DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , NULL );
270
+ DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , 53 , NULL );
265
271
umock_c_reset_all_calls ();
266
272
STRICT_EXPECTED_CALL (getaddrinfo (IGNORED_PTR_ARG , IGNORED_PTR_ARG , IGNORED_PTR_ARG , IGNORED_PTR_ARG )).SetReturn (GETADDRINFO_FAIL );
267
273
@@ -281,7 +287,7 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
281
287
///arrange
282
288
bool result ;
283
289
uint32_t ipv4 ;
284
- DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , NULL );
290
+ DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , 53 , NULL );
285
291
umock_c_reset_all_calls ();
286
292
STRICT_EXPECTED_CALL (getaddrinfo (IGNORED_PTR_ARG , IGNORED_PTR_ARG , IGNORED_PTR_ARG , IGNORED_PTR_ARG )).SetReturn (GETADDRINFO_FAIL );
287
293
result = dns_resolver_is_lookup_complete (dns );
@@ -314,7 +320,7 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
314
320
TEST_FUNCTION (dns_resolver__get_ipv4_too_early__fails )
315
321
{
316
322
///arrange
317
- DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , NULL );
323
+ DNSRESOLVER_HANDLE dns = dns_resolver_create ("fake.com" , 53 , NULL );
318
324
319
325
///act
320
326
uint32_t result = dns_resolver_get_ipv4 (dns );
@@ -355,7 +361,7 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
355
361
TEST_FUNCTION (dns_resolver__destroy__success )
356
362
{
357
363
///arrange
358
- DNSRESOLVER_HANDLE result = dns_resolver_create ("fake.com" , NULL );
364
+ DNSRESOLVER_HANDLE result = dns_resolver_create ("fake.com" , 53 , NULL );
359
365
umock_c_reset_all_calls ();
360
366
361
367
STRICT_EXPECTED_CALL (gballoc_free (IGNORED_NUM_ARG )); // copy hostname
@@ -373,11 +379,11 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
373
379
{
374
380
///arrange
375
381
DNSRESOLVER_HANDLE result ;
376
- STRICT_EXPECTED_CALL (gballoc_malloc ( IGNORED_NUM_ARG )); // copy hostname
382
+ STRICT_EXPECTED_CALL (gballoc_calloc ( IGNORED_NUM_ARG , IGNORED_NUM_ARG )); // copy hostname
377
383
STRICT_EXPECTED_CALL (gballoc_malloc (IGNORED_NUM_ARG )); // instance
378
384
379
385
///act
380
- result = dns_resolver_create ("fake.com" , NULL );
386
+ result = dns_resolver_create ("fake.com" , 53 , NULL );
381
387
382
388
///assert
383
389
ASSERT_IS_NOT_NULL (result );
@@ -395,7 +401,7 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
395
401
int negativeTestsInitResult = umock_c_negative_tests_init ();
396
402
ASSERT_ARE_EQUAL (int , 0 , negativeTestsInitResult );
397
403
398
- STRICT_EXPECTED_CALL (gballoc_malloc ( IGNORED_NUM_ARG )); // copy hostname
404
+ STRICT_EXPECTED_CALL (gballoc_calloc ( IGNORED_NUM_ARG , IGNORED_NUM_ARG )); // copy hostname
399
405
STRICT_EXPECTED_CALL (gballoc_malloc (IGNORED_NUM_ARG )); // instance
400
406
umock_c_negative_tests_snapshot ();
401
407
@@ -407,7 +413,7 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
407
413
umock_c_negative_tests_fail_call (i );
408
414
409
415
///act
410
- result = dns_resolver_create ("fake.com" , NULL );
416
+ result = dns_resolver_create ("fake.com" , 53 , NULL );
411
417
412
418
///assert
413
419
ASSERT_IS_NULL (result );
@@ -423,7 +429,7 @@ BEGIN_TEST_SUITE(dns_resolver_ut)
423
429
///arrange
424
430
425
431
///act
426
- DNSRESOLVER_HANDLE result = dns_resolver_create (NULL , NULL );
432
+ DNSRESOLVER_HANDLE result = dns_resolver_create (NULL , 53 , NULL );
427
433
428
434
///assert
429
435
ASSERT_IS_NULL (result , "Unexpected success with NULL hostname" );
0 commit comments