def test_list_keys(): mocked_keys = [Mock(key='mykey1'), Mock(key='key2')] mocked_connection = Mock() # Start with patching connect_s3 with patch('boto.connect_s3', Mock(return_value=mocked_connection)): mocked_bucket = Mock() # Mock get_bucket() call mocked_connection.get_bucket = Mock(return_value=mocked_bucket) # Mock the list() call to return the keys you want mocked_bucket.list = Mock(return_value=mocked_keys) keys = list_keys() assert keys == mocked_keys
def test_list_keys(): expected_keys = ['key1', 'key2'] moto = mock_s3() # We enter "moto" mode using this moto.start() # Get the connection object conn = get_s3_conn() # Set up S3 as we expect it to be conn.create_bucket('bucket_name') for name in expected_keys: k = conn.get_bucket('bucket_name').new_key(name) k.set_contents_from_string('abcdedsd') # Now call the actual function keys = list_keys() assert expected_keys == [k.name for k in keys] # get out of moto mode moto.stop()