Fix TestTSUIDQuery for MockBase with column family support