Add integration tests for key commands