2014-10-15 13:00:46 +04:00
/***
This file is part of systemd
Copyright 2013 Daniel Buch
systemd is free software ; you can redistribute it and / or modify it
under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation ; either version 2.1 of the License , or
( at your option ) any later version .
systemd is distributed in the hope that it will be useful , but
WITHOUT ANY WARRANTY ; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE . See the GNU
Lesser General Public License for more details .
You should have received a copy of the GNU Lesser General Public License
along with systemd ; If not , see < http : //www.gnu.org/licenses/>.
* * */
# include "strv.h"
# include "util.h"
# include "hashmap.h"
void test_hashmap_funcs ( void ) ;
static void test_hashmap_replace ( void ) {
Hashmap * m ;
char * val1 , * val2 , * val3 , * val4 , * val5 , * r ;
m = hashmap_new ( & string_hash_ops ) ;
val1 = strdup ( " val1 " ) ;
assert_se ( val1 ) ;
val2 = strdup ( " val2 " ) ;
assert_se ( val2 ) ;
val3 = strdup ( " val3 " ) ;
assert_se ( val3 ) ;
val4 = strdup ( " val4 " ) ;
assert_se ( val4 ) ;
val5 = strdup ( " val5 " ) ;
assert_se ( val5 ) ;
hashmap_put ( m , " key 1 " , val1 ) ;
hashmap_put ( m , " key 2 " , val2 ) ;
hashmap_put ( m , " key 3 " , val3 ) ;
hashmap_put ( m , " key 4 " , val4 ) ;
hashmap_replace ( m , " key 3 " , val1 ) ;
r = hashmap_get ( m , " key 3 " ) ;
assert_se ( streq ( r , " val1 " ) ) ;
hashmap_replace ( m , " key 5 " , val5 ) ;
r = hashmap_get ( m , " key 5 " ) ;
assert_se ( streq ( r , " val5 " ) ) ;
free ( val1 ) ;
free ( val2 ) ;
free ( val3 ) ;
free ( val4 ) ;
free ( val5 ) ;
hashmap_free ( m ) ;
}
static void test_hashmap_copy ( void ) {
Hashmap * m , * copy ;
char * val1 , * val2 , * val3 , * val4 , * r ;
val1 = strdup ( " val1 " ) ;
assert_se ( val1 ) ;
val2 = strdup ( " val2 " ) ;
assert_se ( val2 ) ;
val3 = strdup ( " val3 " ) ;
assert_se ( val3 ) ;
val4 = strdup ( " val4 " ) ;
assert_se ( val4 ) ;
m = hashmap_new ( & string_hash_ops ) ;
hashmap_put ( m , " key 1 " , val1 ) ;
hashmap_put ( m , " key 2 " , val2 ) ;
hashmap_put ( m , " key 3 " , val3 ) ;
hashmap_put ( m , " key 4 " , val4 ) ;
copy = hashmap_copy ( m ) ;
r = hashmap_get ( copy , " key 1 " ) ;
assert_se ( streq ( r , " val1 " ) ) ;
r = hashmap_get ( copy , " key 2 " ) ;
assert_se ( streq ( r , " val2 " ) ) ;
r = hashmap_get ( copy , " key 3 " ) ;
assert_se ( streq ( r , " val3 " ) ) ;
r = hashmap_get ( copy , " key 4 " ) ;
assert_se ( streq ( r , " val4 " ) ) ;
hashmap_free_free ( copy ) ;
hashmap_free ( m ) ;
}
static void test_hashmap_get_strv ( void ) {
Hashmap * m ;
char * * strv ;
char * val1 , * val2 , * val3 , * val4 ;
val1 = strdup ( " val1 " ) ;
assert_se ( val1 ) ;
val2 = strdup ( " val2 " ) ;
assert_se ( val2 ) ;
val3 = strdup ( " val3 " ) ;
assert_se ( val3 ) ;
val4 = strdup ( " val4 " ) ;
assert_se ( val4 ) ;
m = hashmap_new ( & string_hash_ops ) ;
hashmap_put ( m , " key 1 " , val1 ) ;
hashmap_put ( m , " key 2 " , val2 ) ;
hashmap_put ( m , " key 3 " , val3 ) ;
hashmap_put ( m , " key 4 " , val4 ) ;
strv = hashmap_get_strv ( m ) ;
# ifndef ORDERED
strv = strv_sort ( strv ) ;
# endif
assert_se ( streq ( strv [ 0 ] , " val1 " ) ) ;
assert_se ( streq ( strv [ 1 ] , " val2 " ) ) ;
assert_se ( streq ( strv [ 2 ] , " val3 " ) ) ;
assert_se ( streq ( strv [ 3 ] , " val4 " ) ) ;
strv_free ( strv ) ;
hashmap_free ( m ) ;
}
static void test_hashmap_move_one ( void ) {
Hashmap * m , * n ;
char * val1 , * val2 , * val3 , * val4 , * r ;
val1 = strdup ( " val1 " ) ;
assert_se ( val1 ) ;
val2 = strdup ( " val2 " ) ;
assert_se ( val2 ) ;
val3 = strdup ( " val3 " ) ;
assert_se ( val3 ) ;
val4 = strdup ( " val4 " ) ;
assert_se ( val4 ) ;
m = hashmap_new ( & string_hash_ops ) ;
n = hashmap_new ( & string_hash_ops ) ;
hashmap_put ( m , " key 1 " , val1 ) ;
hashmap_put ( m , " key 2 " , val2 ) ;
hashmap_put ( m , " key 3 " , val3 ) ;
hashmap_put ( m , " key 4 " , val4 ) ;
2014-10-15 13:06:08 +04:00
assert_se ( hashmap_move_one ( n , NULL , " key 3 " ) = = - ENOENT ) ;
assert_se ( hashmap_move_one ( n , m , " key 5 " ) = = - ENOENT ) ;
assert_se ( hashmap_move_one ( n , m , " key 3 " ) = = 0 ) ;
assert_se ( hashmap_move_one ( n , m , " key 4 " ) = = 0 ) ;
2014-10-15 13:00:46 +04:00
r = hashmap_get ( n , " key 3 " ) ;
assert_se ( r & & streq ( r , " val3 " ) ) ;
r = hashmap_get ( n , " key 4 " ) ;
assert_se ( r & & streq ( r , " val4 " ) ) ;
r = hashmap_get ( m , " key 3 " ) ;
assert_se ( ! r ) ;
2014-10-15 13:06:08 +04:00
assert_se ( hashmap_move_one ( n , m , " key 3 " ) = = - EEXIST ) ;
hashmap_free_free ( m ) ;
hashmap_free_free ( n ) ;
}
static void test_hashmap_move ( void ) {
Hashmap * m , * n ;
char * val1 , * val2 , * val3 , * val4 , * r ;
val1 = strdup ( " val1 " ) ;
assert_se ( val1 ) ;
val2 = strdup ( " val2 " ) ;
assert_se ( val2 ) ;
val3 = strdup ( " val3 " ) ;
assert_se ( val3 ) ;
val4 = strdup ( " val4 " ) ;
assert_se ( val4 ) ;
m = hashmap_new ( & string_hash_ops ) ;
n = hashmap_new ( & string_hash_ops ) ;
hashmap_put ( n , " key 1 " , strdup ( val1 ) ) ;
hashmap_put ( m , " key 1 " , val1 ) ;
hashmap_put ( m , " key 2 " , val2 ) ;
hashmap_put ( m , " key 3 " , val3 ) ;
hashmap_put ( m , " key 4 " , val4 ) ;
hashmap_move ( n , NULL ) ;
hashmap_move ( n , m ) ;
assert_se ( hashmap_size ( m ) = = 1 ) ;
r = hashmap_get ( m , " key 1 " ) ;
assert_se ( r & & streq ( r , " val1 " ) ) ;
r = hashmap_get ( n , " key 1 " ) ;
assert_se ( r & & streq ( r , " val1 " ) ) ;
r = hashmap_get ( n , " key 2 " ) ;
assert_se ( r & & streq ( r , " val2 " ) ) ;
r = hashmap_get ( n , " key 3 " ) ;
assert_se ( r & & streq ( r , " val3 " ) ) ;
r = hashmap_get ( n , " key 4 " ) ;
assert_se ( r & & streq ( r , " val4 " ) ) ;
2014-10-15 13:00:46 +04:00
hashmap_free_free ( m ) ;
hashmap_free_free ( n ) ;
}
static void test_hashmap_update ( void ) {
Hashmap * m ;
char * val1 , * val2 , * r ;
m = hashmap_new ( & string_hash_ops ) ;
val1 = strdup ( " old_value " ) ;
assert_se ( val1 ) ;
val2 = strdup ( " new_value " ) ;
assert_se ( val2 ) ;
hashmap_put ( m , " key 1 " , val1 ) ;
r = hashmap_get ( m , " key 1 " ) ;
assert_se ( streq ( r , " old_value " ) ) ;
2014-10-15 13:06:08 +04:00
assert_se ( hashmap_update ( m , " key 2 " , val2 ) = = - ENOENT ) ;
r = hashmap_get ( m , " key 1 " ) ;
assert_se ( streq ( r , " old_value " ) ) ;
assert_se ( hashmap_update ( m , " key 1 " , val2 ) = = 0 ) ;
2014-10-15 13:00:46 +04:00
r = hashmap_get ( m , " key 1 " ) ;
assert_se ( streq ( r , " new_value " ) ) ;
free ( val1 ) ;
free ( val2 ) ;
hashmap_free ( m ) ;
}
static void test_hashmap_put ( void ) {
2014-10-15 13:06:08 +04:00
Hashmap * m = NULL ;
2014-10-15 13:00:46 +04:00
int valid_hashmap_put ;
2014-10-15 13:06:08 +04:00
void * val1 = ( void * ) " val 1 " ;
2014-10-15 13:00:46 +04:00
2014-10-15 13:06:08 +04:00
hashmap_ensure_allocated ( & m , & string_hash_ops ) ;
assert_se ( m ) ;
2014-10-15 13:00:46 +04:00
2014-10-15 13:06:08 +04:00
valid_hashmap_put = hashmap_put ( m , " key 1 " , val1 ) ;
2014-10-15 13:00:46 +04:00
assert_se ( valid_hashmap_put = = 1 ) ;
2014-10-15 13:06:08 +04:00
assert_se ( hashmap_put ( m , " key 1 " , val1 ) = = 0 ) ;
assert_se ( hashmap_put ( m , " key 1 " , ( void * ) " val 2 " ) = = - EEXIST ) ;
2014-10-15 13:00:46 +04:00
hashmap_free ( m ) ;
}
2014-10-15 13:06:08 +04:00
static void test_hashmap_remove ( void ) {
_cleanup_hashmap_free_ Hashmap * m = NULL ;
char * r ;
r = hashmap_remove ( NULL , " key 1 " ) ;
assert_se ( r = = NULL ) ;
m = hashmap_new ( & string_hash_ops ) ;
assert_se ( m ) ;
r = hashmap_remove ( m , " no such key " ) ;
assert_se ( r = = NULL ) ;
hashmap_put ( m , " key 1 " , ( void * ) " val 1 " ) ;
hashmap_put ( m , " key 2 " , ( void * ) " val 2 " ) ;
r = hashmap_remove ( m , " key 1 " ) ;
assert_se ( streq ( r , " val 1 " ) ) ;
r = hashmap_get ( m , " key 2 " ) ;
assert_se ( streq ( r , " val 2 " ) ) ;
assert_se ( ! hashmap_get ( m , " key 1 " ) ) ;
}
static void test_hashmap_remove2 ( void ) {
_cleanup_hashmap_free_free_free_ Hashmap * m = NULL ;
char key1 [ ] = " key 1 " ;
char key2 [ ] = " key 2 " ;
char val1 [ ] = " val 1 " ;
char val2 [ ] = " val 2 " ;
void * r , * r2 ;
r = hashmap_remove2 ( NULL , " key 1 " , & r2 ) ;
assert_se ( r = = NULL ) ;
m = hashmap_new ( & string_hash_ops ) ;
assert_se ( m ) ;
r = hashmap_remove2 ( m , " no such key " , & r2 ) ;
assert_se ( r = = NULL ) ;
hashmap_put ( m , strdup ( key1 ) , strdup ( val1 ) ) ;
hashmap_put ( m , strdup ( key2 ) , strdup ( val2 ) ) ;
r = hashmap_remove2 ( m , key1 , & r2 ) ;
assert_se ( streq ( r , val1 ) ) ;
assert_se ( streq ( r2 , key1 ) ) ;
free ( r ) ;
free ( r2 ) ;
r = hashmap_get ( m , key2 ) ;
assert_se ( streq ( r , val2 ) ) ;
assert_se ( ! hashmap_get ( m , key1 ) ) ;
}
static void test_hashmap_remove_value ( void ) {
_cleanup_hashmap_free_ Hashmap * m = NULL ;
char * r ;
r = hashmap_remove_value ( NULL , " key 1 " , ( void * ) " val 1 " ) ;
assert_se ( r = = NULL ) ;
m = hashmap_new ( & string_hash_ops ) ;
assert_se ( m ) ;
r = hashmap_remove_value ( m , " key 1 " , ( void * ) " val 1 " ) ;
assert_se ( r = = NULL ) ;
hashmap_put ( m , " key 1 " , ( void * ) " val 1 " ) ;
hashmap_put ( m , " key 2 " , ( void * ) " val 2 " ) ;
r = hashmap_remove_value ( m , " key 1 " , ( void * ) " val 1 " ) ;
assert_se ( streq ( r , " val 1 " ) ) ;
r = hashmap_get ( m , " key 2 " ) ;
assert_se ( streq ( r , " val 2 " ) ) ;
assert_se ( ! hashmap_get ( m , " key 1 " ) ) ;
r = hashmap_remove_value ( m , " key 2 " , ( void * ) " val 1 " ) ;
assert_se ( r = = NULL ) ;
r = hashmap_get ( m , " key 2 " ) ;
assert_se ( streq ( r , " val 2 " ) ) ;
assert_se ( ! hashmap_get ( m , " key 1 " ) ) ;
}
2014-10-15 13:00:46 +04:00
static void test_hashmap_remove_and_put ( void ) {
_cleanup_hashmap_free_ Hashmap * m = NULL ;
int valid ;
char * r ;
m = hashmap_new ( & string_hash_ops ) ;
assert_se ( m ) ;
2014-10-15 13:06:08 +04:00
valid = hashmap_remove_and_put ( m , " invalid key " , " new key " , NULL ) ;
assert_se ( valid = = - ENOENT ) ;
2014-10-15 13:00:46 +04:00
valid = hashmap_put ( m , " key 1 " , ( void * ) ( const char * ) " val 1 " ) ;
assert_se ( valid = = 1 ) ;
2014-10-15 13:06:08 +04:00
valid = hashmap_remove_and_put ( NULL , " key 1 " , " key 2 " , ( void * ) ( const char * ) " val 2 " ) ;
assert_se ( valid = = - ENOENT ) ;
2014-10-15 13:00:46 +04:00
valid = hashmap_remove_and_put ( m , " key 1 " , " key 2 " , ( void * ) ( const char * ) " val 2 " ) ;
assert_se ( valid = = 0 ) ;
r = hashmap_get ( m , " key 2 " ) ;
assert_se ( streq ( r , " val 2 " ) ) ;
assert_se ( ! hashmap_get ( m , " key 1 " ) ) ;
valid = hashmap_put ( m , " key 3 " , ( void * ) ( const char * ) " val 3 " ) ;
assert_se ( valid = = 1 ) ;
valid = hashmap_remove_and_put ( m , " key 3 " , " key 2 " , ( void * ) ( const char * ) " val 2 " ) ;
2014-10-15 13:06:08 +04:00
assert_se ( valid = = - EEXIST ) ;
}
static void test_hashmap_remove_and_replace ( void ) {
_cleanup_hashmap_free_ Hashmap * m = NULL ;
int valid ;
void * key1 = UINT_TO_PTR ( 1 ) ;
void * key2 = UINT_TO_PTR ( 2 ) ;
void * key3 = UINT_TO_PTR ( 3 ) ;
void * r ;
m = hashmap_new ( & trivial_hash_ops ) ;
assert_se ( m ) ;
valid = hashmap_remove_and_replace ( m , key1 , key2 , NULL ) ;
assert_se ( valid = = - ENOENT ) ;
valid = hashmap_put ( m , key1 , key1 ) ;
assert_se ( valid = = 1 ) ;
valid = hashmap_remove_and_replace ( NULL , key1 , key2 , key2 ) ;
assert_se ( valid = = - ENOENT ) ;
valid = hashmap_remove_and_replace ( m , key1 , key2 , key2 ) ;
assert_se ( valid = = 0 ) ;
r = hashmap_get ( m , key2 ) ;
assert_se ( r = = key2 ) ;
assert_se ( ! hashmap_get ( m , key1 ) ) ;
valid = hashmap_put ( m , key3 , key3 ) ;
assert_se ( valid = = 1 ) ;
valid = hashmap_remove_and_replace ( m , key3 , key2 , key2 ) ;
assert_se ( valid = = 0 ) ;
r = hashmap_get ( m , key2 ) ;
assert_se ( r = = key2 ) ;
assert_se ( ! hashmap_get ( m , key3 ) ) ;
2014-10-15 13:00:46 +04:00
}
static void test_hashmap_ensure_allocated ( void ) {
Hashmap * m ;
int valid_hashmap ;
m = hashmap_new ( & string_hash_ops ) ;
valid_hashmap = hashmap_ensure_allocated ( & m , & string_hash_ops ) ;
assert_se ( valid_hashmap = = 0 ) ;
assert_se ( m ) ;
hashmap_free ( m ) ;
}
static void test_hashmap_foreach_key ( void ) {
Hashmap * m ;
Iterator i ;
bool key_found [ ] = { false , false , false , false } ;
const char * s ;
const char * key ;
static const char key_table [ ] =
" key 1 \0 "
" key 2 \0 "
" key 3 \0 "
" key 4 \0 " ;
m = hashmap_new ( & string_hash_ops ) ;
NULSTR_FOREACH ( key , key_table )
hashmap_put ( m , key , ( void * ) ( const char * ) " my dummy val " ) ;
HASHMAP_FOREACH_KEY ( s , key , m , i ) {
if ( ! key_found [ 0 ] & & streq ( key , " key 1 " ) )
key_found [ 0 ] = true ;
else if ( ! key_found [ 1 ] & & streq ( key , " key 2 " ) )
key_found [ 1 ] = true ;
else if ( ! key_found [ 2 ] & & streq ( key , " key 3 " ) )
key_found [ 2 ] = true ;
else if ( ! key_found [ 3 ] & & streq ( key , " fail " ) )
key_found [ 3 ] = true ;
}
assert_se ( m ) ;
assert_se ( key_found [ 0 ] & & key_found [ 1 ] & & key_found [ 2 ] & & ! key_found [ 3 ] ) ;
hashmap_free ( m ) ;
}
static void test_hashmap_foreach ( void ) {
Hashmap * m ;
Iterator i ;
bool value_found [ ] = { false , false , false , false } ;
char * val1 , * val2 , * val3 , * val4 , * s ;
2014-10-15 13:06:08 +04:00
unsigned count ;
2014-10-15 13:00:46 +04:00
val1 = strdup ( " my val1 " ) ;
assert_se ( val1 ) ;
val2 = strdup ( " my val2 " ) ;
assert_se ( val2 ) ;
val3 = strdup ( " my val3 " ) ;
assert_se ( val3 ) ;
val4 = strdup ( " my val4 " ) ;
assert_se ( val4 ) ;
2014-10-15 13:06:08 +04:00
m = NULL ;
count = 0 ;
HASHMAP_FOREACH ( s , m , i )
count + + ;
assert_se ( count = = 0 ) ;
2014-10-15 13:00:46 +04:00
m = hashmap_new ( & string_hash_ops ) ;
2014-10-15 13:06:08 +04:00
count = 0 ;
HASHMAP_FOREACH ( s , m , i )
count + + ;
assert_se ( count = = 0 ) ;
2014-10-15 13:00:46 +04:00
hashmap_put ( m , " Key 1 " , val1 ) ;
hashmap_put ( m , " Key 2 " , val2 ) ;
hashmap_put ( m , " Key 3 " , val3 ) ;
hashmap_put ( m , " Key 4 " , val4 ) ;
HASHMAP_FOREACH ( s , m , i ) {
if ( ! value_found [ 0 ] & & streq ( s , val1 ) )
value_found [ 0 ] = true ;
else if ( ! value_found [ 1 ] & & streq ( s , val2 ) )
value_found [ 1 ] = true ;
else if ( ! value_found [ 2 ] & & streq ( s , val3 ) )
value_found [ 2 ] = true ;
else if ( ! value_found [ 3 ] & & streq ( s , val4 ) )
value_found [ 3 ] = true ;
}
assert_se ( m ) ;
assert_se ( value_found [ 0 ] & & value_found [ 1 ] & & value_found [ 2 ] & & value_found [ 3 ] ) ;
hashmap_free_free ( m ) ;
}
static void test_hashmap_merge ( void ) {
Hashmap * m ;
Hashmap * n ;
char * val1 , * val2 , * val3 , * val4 , * r ;
val1 = strdup ( " my val1 " ) ;
assert_se ( val1 ) ;
val2 = strdup ( " my val2 " ) ;
assert_se ( val2 ) ;
val3 = strdup ( " my val3 " ) ;
assert_se ( val3 ) ;
val4 = strdup ( " my val4 " ) ;
assert_se ( val4 ) ;
n = hashmap_new ( & string_hash_ops ) ;
m = hashmap_new ( & string_hash_ops ) ;
hashmap_put ( m , " Key 1 " , val1 ) ;
hashmap_put ( m , " Key 2 " , val2 ) ;
hashmap_put ( n , " Key 3 " , val3 ) ;
hashmap_put ( n , " Key 4 " , val4 ) ;
assert_se ( hashmap_merge ( m , n ) = = 0 ) ;
r = hashmap_get ( m , " Key 3 " ) ;
assert_se ( r & & streq ( r , " my val3 " ) ) ;
r = hashmap_get ( m , " Key 4 " ) ;
assert_se ( r & & streq ( r , " my val4 " ) ) ;
assert_se ( n ) ;
assert_se ( m ) ;
hashmap_free ( n ) ;
hashmap_free_free ( m ) ;
}
static void test_hashmap_contains ( void ) {
Hashmap * m ;
char * val1 ;
val1 = strdup ( " my val " ) ;
assert_se ( val1 ) ;
m = hashmap_new ( & string_hash_ops ) ;
assert_se ( ! hashmap_contains ( m , " Key 1 " ) ) ;
hashmap_put ( m , " Key 1 " , val1 ) ;
assert_se ( hashmap_contains ( m , " Key 1 " ) ) ;
2014-10-15 13:06:08 +04:00
assert_se ( ! hashmap_contains ( m , " Key 2 " ) ) ;
assert_se ( ! hashmap_contains ( NULL , " Key 1 " ) ) ;
2014-10-15 13:00:46 +04:00
assert_se ( m ) ;
hashmap_free_free ( m ) ;
}
static void test_hashmap_isempty ( void ) {
Hashmap * m ;
char * val1 ;
val1 = strdup ( " my val " ) ;
assert_se ( val1 ) ;
m = hashmap_new ( & string_hash_ops ) ;
assert_se ( hashmap_isempty ( m ) ) ;
hashmap_put ( m , " Key 1 " , val1 ) ;
assert_se ( ! hashmap_isempty ( m ) ) ;
assert_se ( m ) ;
hashmap_free_free ( m ) ;
}
static void test_hashmap_size ( void ) {
Hashmap * m ;
char * val1 , * val2 , * val3 , * val4 ;
val1 = strdup ( " my val " ) ;
assert_se ( val1 ) ;
val2 = strdup ( " my val " ) ;
assert_se ( val2 ) ;
val3 = strdup ( " my val " ) ;
assert_se ( val3 ) ;
val4 = strdup ( " my val " ) ;
assert_se ( val4 ) ;
2014-10-15 13:06:08 +04:00
assert_se ( hashmap_size ( NULL ) = = 0 ) ;
assert_se ( hashmap_buckets ( NULL ) = = 0 ) ;
2014-10-15 13:00:46 +04:00
m = hashmap_new ( & string_hash_ops ) ;
hashmap_put ( m , " Key 1 " , val1 ) ;
hashmap_put ( m , " Key 2 " , val2 ) ;
hashmap_put ( m , " Key 3 " , val3 ) ;
hashmap_put ( m , " Key 4 " , val4 ) ;
assert_se ( m ) ;
assert_se ( hashmap_size ( m ) = = 4 ) ;
2014-10-15 13:06:08 +04:00
assert_se ( hashmap_buckets ( m ) > = 4 ) ;
2014-10-15 13:00:46 +04:00
hashmap_free_free ( m ) ;
}
static void test_hashmap_get ( void ) {
Hashmap * m ;
char * r ;
char * val ;
val = strdup ( " my val " ) ;
assert_se ( val ) ;
2014-10-15 13:06:08 +04:00
r = hashmap_get ( NULL , " Key 1 " ) ;
assert_se ( r = = NULL ) ;
2014-10-15 13:00:46 +04:00
m = hashmap_new ( & string_hash_ops ) ;
hashmap_put ( m , " Key 1 " , val ) ;
r = hashmap_get ( m , " Key 1 " ) ;
assert_se ( streq ( r , val ) ) ;
2014-10-15 13:06:08 +04:00
r = hashmap_get ( m , " no such key " ) ;
assert_se ( r = = NULL ) ;
2014-10-15 13:00:46 +04:00
assert_se ( m ) ;
hashmap_free_free ( m ) ;
}
2014-10-15 13:06:08 +04:00
static void test_hashmap_get2 ( void ) {
Hashmap * m ;
char * r ;
char * val ;
char key_orig [ ] = " Key 1 " ;
void * key_copy ;
val = strdup ( " my val " ) ;
assert_se ( val ) ;
key_copy = strdup ( key_orig ) ;
assert_se ( key_copy ) ;
r = hashmap_get2 ( NULL , key_orig , & key_copy ) ;
assert_se ( r = = NULL ) ;
m = hashmap_new ( & string_hash_ops ) ;
hashmap_put ( m , key_copy , val ) ;
key_copy = NULL ;
r = hashmap_get2 ( m , key_orig , & key_copy ) ;
assert_se ( streq ( r , val ) ) ;
assert_se ( key_orig ! = key_copy ) ;
assert_se ( streq ( key_orig , key_orig ) ) ;
r = hashmap_get2 ( m , " no such key " , NULL ) ;
assert_se ( r = = NULL ) ;
assert_se ( m ) ;
hashmap_free_free_free ( m ) ;
}
static unsigned long crippled_hashmap_func ( const void * p , const uint8_t hash_key [ HASH_KEY_SIZE ] ) {
return trivial_hash_func ( p , hash_key ) & 0xff ;
}
static const struct hash_ops crippled_hashmap_ops = {
. hash = crippled_hashmap_func ,
. compare = trivial_compare_func ,
} ;
2014-10-15 13:00:46 +04:00
static void test_hashmap_many ( void ) {
Hashmap * h ;
2014-10-15 13:06:08 +04:00
unsigned i , j ;
void * v , * k ;
static const struct {
const struct hash_ops * ops ;
unsigned n_entries ;
} tests [ ] = {
{ . ops = NULL , . n_entries = 1 < < 20 } ,
{ . ops = & crippled_hashmap_ops , . n_entries = 1 < < 11 } ,
} ;
2014-10-15 13:00:46 +04:00
2014-10-15 13:06:08 +04:00
for ( j = 0 ; j < ELEMENTSOF ( tests ) ; j + + ) {
assert_se ( h = hashmap_new ( tests [ j ] . ops ) ) ;
2014-10-15 13:00:46 +04:00
2014-10-15 13:06:08 +04:00
for ( i = 1 ; i < tests [ j ] . n_entries * 3 ; i + = 3 ) {
assert_se ( hashmap_put ( h , UINT_TO_PTR ( i ) , UINT_TO_PTR ( i ) ) > = 0 ) ;
assert_se ( PTR_TO_UINT ( hashmap_get ( h , UINT_TO_PTR ( i ) ) ) = = i ) ;
}
for ( i = 1 ; i < tests [ j ] . n_entries * 3 ; i + + )
assert_se ( hashmap_contains ( h , UINT_TO_PTR ( i ) ) = = ( i % 3 = = 1 ) ) ;
2014-10-15 13:00:46 +04:00
2014-10-15 13:06:08 +04:00
log_info ( " %u <= %u * 0.75 = %g " , hashmap_size ( h ) , hashmap_buckets ( h ) , hashmap_buckets ( h ) * 0.75 ) ;
2014-10-15 13:00:46 +04:00
2014-10-15 13:06:08 +04:00
assert_se ( hashmap_size ( h ) < = hashmap_buckets ( h ) * 0.75 ) ;
assert_se ( hashmap_size ( h ) = = tests [ j ] . n_entries ) ;
2014-10-15 13:00:46 +04:00
2014-10-15 13:06:08 +04:00
while ( ! hashmap_isempty ( h ) ) {
k = hashmap_first_key ( h ) ;
v = hashmap_remove ( h , k ) ;
assert_se ( v = = k ) ;
}
2014-10-15 13:00:46 +04:00
2014-10-15 13:06:08 +04:00
hashmap_free ( h ) ;
}
}
static void test_hashmap_first ( void ) {
_cleanup_hashmap_free_ Hashmap * m = NULL ;
m = hashmap_new ( & string_hash_ops ) ;
assert_se ( m ) ;
assert_se ( ! hashmap_first ( m ) ) ;
assert_se ( hashmap_put ( m , " key 1 " , ( void * ) " val 1 " ) = = 1 ) ;
assert_se ( streq ( hashmap_first ( m ) , " val 1 " ) ) ;
assert_se ( hashmap_put ( m , " key 2 " , ( void * ) " val 2 " ) = = 1 ) ;
# ifdef ORDERED
assert_se ( streq ( hashmap_first ( m ) , " val 1 " ) ) ;
assert_se ( hashmap_remove ( m , " key 1 " ) ) ;
assert_se ( streq ( hashmap_first ( m ) , " val 2 " ) ) ;
# endif
2014-10-15 13:00:46 +04:00
}
static void test_hashmap_first_key ( void ) {
_cleanup_hashmap_free_ Hashmap * m = NULL ;
m = hashmap_new ( & string_hash_ops ) ;
assert_se ( m ) ;
assert_se ( ! hashmap_first_key ( m ) ) ;
assert_se ( hashmap_put ( m , " key 1 " , NULL ) = = 1 ) ;
assert_se ( streq ( hashmap_first_key ( m ) , " key 1 " ) ) ;
assert_se ( hashmap_put ( m , " key 2 " , NULL ) = = 1 ) ;
# ifdef ORDERED
assert_se ( streq ( hashmap_first_key ( m ) , " key 1 " ) ) ;
assert_se ( hashmap_remove ( m , " key 1 " ) = = NULL ) ;
assert_se ( streq ( hashmap_first_key ( m ) , " key 2 " ) ) ;
# endif
}
static void test_hashmap_steal_first_key ( void ) {
_cleanup_hashmap_free_ Hashmap * m = NULL ;
m = hashmap_new ( & string_hash_ops ) ;
assert_se ( m ) ;
assert_se ( ! hashmap_steal_first_key ( m ) ) ;
assert_se ( hashmap_put ( m , " key 1 " , NULL ) = = 1 ) ;
assert_se ( streq ( hashmap_steal_first_key ( m ) , " key 1 " ) ) ;
assert_se ( hashmap_isempty ( m ) ) ;
}
static void test_hashmap_steal_first ( void ) {
_cleanup_hashmap_free_ Hashmap * m = NULL ;
int seen [ 3 ] = { } ;
char * val ;
m = hashmap_new ( & string_hash_ops ) ;
assert_se ( m ) ;
assert_se ( hashmap_put ( m , " key 1 " , ( void * ) " 1 " ) = = 1 ) ;
assert_se ( hashmap_put ( m , " key 2 " , ( void * ) " 22 " ) = = 1 ) ;
assert_se ( hashmap_put ( m , " key 3 " , ( void * ) " 333 " ) = = 1 ) ;
while ( ( val = hashmap_steal_first ( m ) ) )
seen [ strlen ( val ) - 1 ] + + ;
assert_se ( seen [ 0 ] = = 1 & & seen [ 1 ] = = 1 & & seen [ 2 ] = = 1 ) ;
assert_se ( hashmap_isempty ( m ) ) ;
}
static void test_hashmap_clear_free_free ( void ) {
_cleanup_hashmap_free_ Hashmap * m = NULL ;
m = hashmap_new ( & string_hash_ops ) ;
assert_se ( m ) ;
assert_se ( hashmap_put ( m , strdup ( " key 1 " ) , NULL ) = = 1 ) ;
assert_se ( hashmap_put ( m , strdup ( " key 2 " ) , NULL ) = = 1 ) ;
assert_se ( hashmap_put ( m , strdup ( " key 3 " ) , NULL ) = = 1 ) ;
hashmap_clear_free_free ( m ) ;
assert_se ( hashmap_isempty ( m ) ) ;
}
void test_hashmap_funcs ( void ) {
test_hashmap_copy ( ) ;
test_hashmap_get_strv ( ) ;
test_hashmap_move_one ( ) ;
2014-10-15 13:06:08 +04:00
test_hashmap_move ( ) ;
2014-10-15 13:00:46 +04:00
test_hashmap_replace ( ) ;
test_hashmap_update ( ) ;
test_hashmap_put ( ) ;
2014-10-15 13:06:08 +04:00
test_hashmap_remove ( ) ;
test_hashmap_remove2 ( ) ;
test_hashmap_remove_value ( ) ;
2014-10-15 13:00:46 +04:00
test_hashmap_remove_and_put ( ) ;
2014-10-15 13:06:08 +04:00
test_hashmap_remove_and_replace ( ) ;
2014-10-15 13:00:46 +04:00
test_hashmap_ensure_allocated ( ) ;
test_hashmap_foreach ( ) ;
test_hashmap_foreach_key ( ) ;
test_hashmap_contains ( ) ;
test_hashmap_merge ( ) ;
test_hashmap_isempty ( ) ;
test_hashmap_get ( ) ;
2014-10-15 13:06:08 +04:00
test_hashmap_get2 ( ) ;
2014-10-15 13:00:46 +04:00
test_hashmap_size ( ) ;
test_hashmap_many ( ) ;
2014-10-15 13:06:08 +04:00
test_hashmap_first ( ) ;
2014-10-15 13:00:46 +04:00
test_hashmap_first_key ( ) ;
test_hashmap_steal_first_key ( ) ;
test_hashmap_steal_first ( ) ;
test_hashmap_clear_free_free ( ) ;
}