Add hive server mock for some tests