bsk.py

def test_bluesky_link_calculator():
    assert (
        bsk_calculate_link_size("http://google.com")
        == bsk_calculate_link_size("https://google.com")
        == bsk_calculate_link_size("https://google.com/")
        == bsk_calculate_link_size("https://google.com#")
        == 10
    )

    assert (
        bsk_calculate_link_size("https://google.com//")
        == bsk_calculate_link_size("https://google.com##")
        == 12
    )

    assert bsk_calculate_link_size("google.com") == 10

    assert bsk_calculate_link_size("google.com/") == 11

    assert bsk_calculate_link_size("google.com/#") == 12

    assert bsk_calculate_link_size("https://bitecode.dev/notes") == 18
    assert bsk_calculate_link_size("https://www.bitecode.dev/notes") == 22
    assert bsk_calculate_link_size("https://www.bitecode.dev/notes/") == 23
    assert bsk_calculate_link_size("https://www.bitecode.dev/notes/#") == 23
    assert bsk_calculate_link_size("https://www.bitecode.dev/notes#") == 22
    assert bsk_calculate_link_size("https://www.bitecode.dev/notes#tag") == 26
    assert bsk_calculate_link_size("https://www.bitecode.dev/notes?a=1") == 26
    assert bsk_calculate_link_size("https://www.bitecode.dev/notes?a=1#tag") == 30
    assert bsk_calculate_link_size("https://www.bitecode.dev/notes?a=1#") == 26

    assert (
        bsk_calculate_link_size("https://substack.com/@bitecode/note/c-60714440") == 28
    )
    assert (
        bsk_calculate_link_size(
            "https://docs.python.org/3/library/stdtypes.html#str.split"
        )
        == 31
    )

    long_url = "https://www.thelongestdomainnameintheworldandthensomeandthensomemoreandmore.com"
    long_domain = long_url.strip("https://")
    assert bsk_calculate_link_size(long_url) == len(long_domain)
    assert bsk_calculate_link_size(long_url + "/") == len(long_domain)
    assert bsk_calculate_link_size(long_url + "/test") == len(long_domain) + 5
    assert bsk_calculate_link_size(long_url + "/" + "a" * 20) == len(long_domain) + 16