update test script for multi-key logic