From cb1c046b9856cf928608de03ee78784529867af6 Mon Sep 17 00:00:00 2001 From: Bernie Hackett Date: Sun, 8 Mar 2015 14:41:52 -0700 Subject: [PATCH] PYTHON-815 - Fix server selection spec test runner. --- test/test_server_selection.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/test/test_server_selection.py b/test/test_server_selection.py index 133578c16..1cfc0b2d4 100644 --- a/test/test_server_selection.py +++ b/test/test_server_selection.py @@ -142,14 +142,11 @@ def create_test(scenario_def): if mode_string: mode_string = mode_string[:1].lower() + mode_string[1:] - - mode = read_preferences.ReadPreference[ - read_preferences.read_pref_mode_from_name(mode_string)] + mode = read_preferences.read_pref_mode_from_name(mode_string) + tag_sets = None if scenario_def['read_preference']['tag_sets'][0]: - instance = mode.__class__( - tag_sets=scenario_def['read_preference']['tag_sets']) - else: - instance = mode + tag_sets = scenario_def['read_preference']['tag_sets'] + instance = read_preferences.make_read_preference(mode, tag_sets) # Select servers. if not scenario_def['suitable_servers']: