'Validates that when a context query is constructed that the
keyspace of the returned model is toggled appropriately
@since 3.6
@jira_ticket PYTHON-598
@expected_result default keyspace should be used
@test_category query'

def test_context_manager(self):

    for ks in self.KEYSPACES:
        with ContextQuery(TestModel, keyspace=ks) as tm:
            self.assertEqual(tm.__keyspace__, ks)
    self.assertEqual(TestModel._get_keyspace(), 'ks1')
