Properly mock the keyspace.

The keyspace has two components that need to be mocked independently:
keyspace and get_keyspace.