Add/fix tests for drone restriction/stat tracker