testHttpRange.cc

Go to the documentation of this file.

/*
 * Copyright (C) 1996-2023 The Squid Software Foundation and contributors
 *
 * Squid software is distributed under GPLv2+ license and includes
 * contributions from numerous individuals and organizations.
 * Please see the COPYING and CONTRIBUTORS files for details.
 */
 
#include "squid.h"
#include "compat/cppunit.h"
#include "HttpHeaderRange.h"
#include "unitTestMain.h"
 
class TestHttpRange : public CPPUNIT_NS::TestFixture
{
    CPPUNIT_TEST_SUITE(TestHttpRange);
    CPPUNIT_TEST(testRangeParser);
    CPPUNIT_TEST(testRangeIter);
    CPPUNIT_TEST(testRangeCanonization);
    CPPUNIT_TEST_SUITE_END();
 
protected:
    void testRangeParser();
    void testRangeParsing(char const *rangestring);
    void testRangeIter();
    void testRangeCanonization();
};
CPPUNIT_TEST_SUITE_REGISTRATION( TestHttpRange );
 
void
TestHttpRange::testRangeParser()
{
    testRangeParsing("bytes=0-3");
    testRangeParsing("bytes=-3");
    testRangeParsing("bytes=1-");
    testRangeParsing("bytes=0-3, 1-, -2");
}
 
void
TestHttpRange::testRangeParsing(char const *rangestring)
{
    String aString (rangestring);
    HttpHdrRange *range = HttpHdrRange::ParseCreate (&aString);
 
    if (!range)
        exit(EXIT_FAILURE);
 
    HttpHdrRange copy(*range);
 
    CPPUNIT_ASSERT_EQUAL(range->specs.size(), copy.specs.size());
 
    HttpHdrRange::iterator pos = range->begin();
 
    CPPUNIT_ASSERT(*pos);
 
    delete range;
}
 
static HttpHdrRange *
rangeFromString(char const *rangestring)
{
    String aString (rangestring);
    HttpHdrRange *range = HttpHdrRange::ParseCreate (&aString);
 
    if (!range)
        exit(EXIT_FAILURE);
 
    return range;
}
 
void
TestHttpRange::testRangeIter()
{
    HttpHdrRange *range=rangeFromString("bytes=0-3, 1-, -2");
    CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(3), range->specs.size());
    size_t counter = 0;
    HttpHdrRange::iterator i = range->begin();
 
    while (i != range->end()) {
        ++counter;
        ++i;
    }
 
    CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(3), counter);
    i = range->begin();
    CPPUNIT_ASSERT_EQUAL(static_cast<ptrdiff_t>(0), i - range->begin());
    ++i;
    CPPUNIT_ASSERT_EQUAL(static_cast<ptrdiff_t>(1), i - range->begin());
    CPPUNIT_ASSERT_EQUAL(static_cast<ptrdiff_t>(-2), i - range->end());
}
 
void
TestHttpRange::testRangeCanonization()
{
    HttpHdrRange *range=rangeFromString("bytes=0-3, 1-, -2");
    CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(3), range->specs.size());
 
    /* 0-3 needs a content length of 4 */
    /* This passes in the extant code - but should it? */
 
    if (!range->canonize(3))
        exit(EXIT_FAILURE);
 
    CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(3), range->specs.size());
 
    delete range;
 
    range=rangeFromString("bytes=0-3, 1-, -2");
 
    CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(3), range->specs.size());
 
    /* 0-3 needs a content length of 4 */
    if (!range->canonize(4))
        exit(EXIT_FAILURE);
 
    delete range;
 
    range=rangeFromString("bytes=3-6");
 
    CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), range->specs.size());
 
    /* 3-6 needs a content length of 4 or more */
    if (range->canonize(3))
        exit(EXIT_FAILURE);
 
    delete range;
 
    range=rangeFromString("bytes=3-6");
 
    CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), range->specs.size());
 
    /* 3-6 needs a content length of 4 or more */
    if (!range->canonize(4))
        exit(EXIT_FAILURE);
 
    delete range;
 
    range=rangeFromString("bytes=1-1,2-3");
 
    CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(2), range->specs.size());
 
    if (!range->canonize(4))
        exit(EXIT_FAILURE);
 
    CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(2), range->specs.size());
 
    delete range;
}
 
int
main(int argc, char *argv[])
{
    return TestProgram().run(argc, argv);
}