def METHOD_NAME(sagemaker_session):
    sagemaker_session.sagemaker_client.add_association.return_value = {
        "AssociationArn": "association-arn",
    }
    obj = association.Association.create(
        source_arn="foo",
        destination_arn="bar",
        association_type="test-type",
        sagemaker_session=sagemaker_session,
    )
    tag = {"Key": "foo", "Value": "bar"}
    sagemaker_session.sagemaker_client.add_tags.return_value = {"Tags": tag}
    obj.set_tag(tag)
    sagemaker_session.sagemaker_client.add_tags.assert_called_with = {
        "ResourceArn": "association-arn",
        "Tags": [{"Key": "foo", "Value": "bar"}],
    }