#include "gtest/gtest.h" #include "common_defs.h" #include "url.h" #include #include using namespace std; namespace test_url { // The fixture for testing class Foo. class test_url : public ::testing::Test { protected: // You can remove any or all of the following functions if its body // is empty. test_url() { // You can do set-up work for each test here. } virtual ~test_url() { // You can do clean-up work that doesn't throw exceptions here. } // If the constructor and destructor are not enough for setting up // and cleaning up each test, you can define the following methods: virtual void SetUp() { // Code here will be called immediately after the constructor (right // before each test). } virtual void TearDown() { // Code here will be called immediately after each test (right // before the destructor). } // Objects declared here can be used by all tests in the test case for Foo. }; // Tests that Foo does Xyz. TEST_F(test_url, parse_url) { string url; PARSED_URL purl; //First is a simple basic test url = "http://www.example.com"; parse_url(url.c_str(), purl); EXPECT_EQ(purl.protocol, URL_PROTOCOL_HTTP); EXPECT_STREQ(purl.user, ""); EXPECT_STREQ(purl.passwd, ""); EXPECT_EQ(purl.port, 80); EXPECT_STREQ(purl.file, ""); EXPECT_STREQ(purl.host, "www.example.com"); //Test Getting https and file name url = "https://secure.example.com/test/file.txt"; parse_url(url.c_str(), purl); EXPECT_EQ(purl.protocol, URL_PROTOCOL_HTTPS); EXPECT_STREQ(purl.user, ""); EXPECT_STREQ(purl.passwd, ""); EXPECT_EQ(purl.port, 443); EXPECT_STREQ(purl.file, "test/file.txt"); EXPECT_STREQ(purl.host, "secure.example.com"); //Test Socks, different port and user being passed url = "socks://user@sock.example.com:5005/"; parse_url(url.c_str(), purl); EXPECT_EQ(purl.protocol, URL_PROTOCOL_SOCKS); EXPECT_STREQ(purl.user, "user"); EXPECT_STREQ(purl.passwd, ""); EXPECT_EQ(purl.port, 5005); EXPECT_STREQ(purl.file, ""); EXPECT_STREQ(purl.host, "sock.example.com"); //Test bad protocol passed, and password being set. port not set. url = "user:mypass@yikes.example.com/"; parse_url(url.c_str(), purl); EXPECT_EQ(purl.protocol, URL_PROTOCOL_UNKNOWN); EXPECT_STREQ(purl.user, "user"); EXPECT_STREQ(purl.passwd, "mypass"); EXPECT_EQ(purl.port, 80); EXPECT_STREQ(purl.file, ""); EXPECT_STREQ(purl.host, "yikes.example.com"); } TEST_F(test_url, is_https) { EXPECT_EQ(is_https("hello"), false); EXPECT_EQ(is_https("https://www.example.com"), true); EXPECT_EQ(is_https("http://www.example.com"), false); EXPECT_EQ(is_https("xhttps://www.example.com"), false); } TEST_F(test_url, escape_url) { string temp1 = "aeiou äëïöü áéíóú àèìòù"; string temp1_answer = "aeiou%20%C3%A4%C3%AB%C3%AF%C3%B6%C3%BC%20%C3%A1%C3%A9%C3%AD%C3%B3%C3%BA%20%C3%A0%C3%A8%C3%AC%C3%B2%C3%B9"; string temp2 = "Epäjärjestelmällistyttämättömyydellänsäkäänköhän"; string temp2_answer = "Ep%C3%A4j%C3%A4rjestelm%C3%A4llistytt%C3%A4m%C3%A4tt%C3%B6myydell%C3%A4ns%C3%A4k%C3%A4%C3%A4nk%C3%B6h%C3%A4n"; escape_url(temp1); EXPECT_EQ(temp1, temp1_answer); escape_url(temp2); EXPECT_EQ(temp2, temp2_answer); } TEST_F(test_url, unescape_url) { string temp1 = "aeiou%20%C3%A4%C3%AB%C3%AF%C3%B6%C3%BC%20%C3%A1%C3%A9%C3%AD%C3%B3%C3%BA%20%C3%A0%C3%A8%C3%AC%C3%B2%C3%B9"; string temp1_answer = "aeiou äëïöü áéíóú àèìòù"; string temp2 = "Ep%C3%A4j%C3%A4rjestelm%C3%A4llistytt%C3%A4m%C3%A4tt%C3%B6myydell%C3%A4ns%C3%A4k%C3%A4%C3%A4nk%C3%B6h%C3%A4n"; string temp2_answer = "Epäjärjestelmällistyttämättömyydellänsäkäänköhän"; unescape_url(temp1); EXPECT_EQ(temp1, temp1_answer); unescape_url(temp2); EXPECT_EQ(temp2, temp2_answer); } TEST_F(test_url, escape_url_readable) { char buf[1024]; char url[1024]; strncpy(url, "https://secure.example.com", sizeof (url)); escape_url_readable(url, buf); EXPECT_STREQ(buf, "secure.example.com"); //Convert the extra / and $ to _ strncpy(url, "https://money.example.com/Dollar$", sizeof (url)); escape_url_readable(url, buf); EXPECT_STREQ(buf, "money.example.com_Dollar_"); //Change all spaces and @ % to _ strncpy(url, "nothing@ should %", sizeof (url)); escape_url_readable(url, buf); EXPECT_STREQ(buf, "nothing__should__"); } TEST_F(test_url, canonicalize_master_url) { //Test to make sure a already good result comes back the same. string url = "http://secure.example.com/"; canonicalize_master_url(url); EXPECT_STREQ(url.c_str(), "http://secure.example.com/"); //Test that https works, also adds trailing / url = "https://secure.example.com"; canonicalize_master_url(url); EXPECT_STREQ(url.c_str(), "https://secure.example.com/"); //Test if someone forgot the leading http:// url = "www.example.com/"; canonicalize_master_url(url); EXPECT_STREQ(url.c_str(), "http://www.example.com/"); //Test removing extra slashes. And changing socks to https. url = "socks://sock.example.com////////hello"; canonicalize_master_url(url); EXPECT_STREQ(url.c_str(), "https://sock.example.com/hello/"); //Test if poorly written url url = "h://bad.example.com/"; canonicalize_master_url(url); EXPECT_STREQ(url.c_str(), "http://bad.example.com/"); } TEST_F(test_url, valid_master_url) { char url[1024]; //Check for a good unsecure url. strncpy(url, "http://www.example.com/", sizeof (url)); EXPECT_EQ(valid_master_url(url), true); //Check for a good secure url strncpy(url, "https://www.example.com/", sizeof (url)); EXPECT_EQ(valid_master_url(url), true); //Check for no http or https. strncpy(url, "hxxp://www.example.com/", sizeof (url)); EXPECT_EQ(valid_master_url(url), false); //Check if missing final slash. strncpy(url, "http://www.example.com", sizeof (url)); EXPECT_EQ(valid_master_url(url), false); //Check if it has no . in the name strncpy(url, "http://example/", sizeof (url)); EXPECT_EQ(valid_master_url(url), false); } TEST_F(test_url, escape_project_url) { char buf[1024]; char url[1024]; //testing a good url. strncpy(url, "https://secure.example.com", sizeof (url)); escape_project_url(url, buf); EXPECT_STREQ(buf, "secure.example.com"); //Testing url with bad character at the end removed. strncpy(url, "https://secure.example.com/Dollar$", sizeof (url)); escape_project_url(url, buf); EXPECT_STREQ(buf, "secure.example.com_Dollar"); } } // namespace