#NEWLINE# def test_get_memory_usage( self ) : #NEWLINE# #INDENT# flexmock ( utils ) . should_receive ( 'get_secret' ) . and_return ( 'fake secret' ) #NEWLINE# expected_keys = [ JSONTags . MEMORY , JSONTags . AVAILABLE , JSONTags . USED ] #NEWLINE# actual = json . loads ( SystemManager ( ) . get_memory_usage ( 'fake secret' ) ) #NEWLINE# actual_keys = [ actual . keys ( ) [ 0 ] ] #NEWLINE# for key in actual . values ( ) [ 0 ] . keys ( ) : #NEWLINE# #INDENT# actual_keys . append ( key ) #NEWLINE# #UNINDENT# self . assertSetEqual ( set ( expected_keys ) , set ( actual_keys ) )