def METHOD_NAME(subscription, input_events):
    """Mocks pubsub input and returns the values based on the subscription name (i.e whether side input
       subscription or main input subscription)
    """
    if subscription == 'projects/test-project/subscriptions/sidesub':
        return TestStream().add_elements([
            beam.window.TimestampedValue(
                "path/to/sideinput/yyyymmddd".encode(),
                int(time.time()))
        ])
    return TestStream().add_elements(input_events)