1ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko// Copyright (c) 2014 The Chromium OS Authors. All rights reserved. 2ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko// Use of this source code is governed by a BSD-style license that can be 3ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko// found in the LICENSE file. 4ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko 59ed0cab99f18acb3570a35e9408f24355f6b8324Alex Vakulenko#include <brillo/mime_utils.h> 6ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko 7ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko#include <gtest/gtest.h> 8ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko 99ed0cab99f18acb3570a35e9408f24355f6b8324Alex Vakulenkonamespace brillo { 10ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko 11ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex VakulenkoTEST(MimeUtils, Combine) { 12ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko std::string mime_string = mime::Combine(mime::types::kText, "xml"); 13ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ(mime::text::kXml, mime_string); 1405d29044d14a60775ed6c51c75a414eb0cb50347Alex Vakulenko EXPECT_EQ( 1505d29044d14a60775ed6c51c75a414eb0cb50347Alex Vakulenko "application/json; charset=utf-8", 1605d29044d14a60775ed6c51c75a414eb0cb50347Alex Vakulenko mime::Combine(mime::types::kApplication, "json", {{"charset", "utf-8"}})); 17ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko} 18ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko 19ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex VakulenkoTEST(MimeUtils, Split) { 20ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko std::string s1, s2; 21ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_TRUE(mime::Split(mime::image::kJpeg, &s1, &s2)); 22ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ(mime::types::kImage, s1); 23ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("jpeg", s2); 24ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko 25ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko mime::Parameters parameters; 2605d29044d14a60775ed6c51c75a414eb0cb50347Alex Vakulenko EXPECT_TRUE( 2705d29044d14a60775ed6c51c75a414eb0cb50347Alex Vakulenko mime::Split("application/json;charset=utf-8", &s1, &s2, ¶meters)); 28ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ(mime::types::kApplication, s1); 29ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("json", s2); 30ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ(mime::application::kJson, mime::Combine(s1, s2)); 31ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ(1, parameters.size()); 32ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ(mime::parameters::kCharset, parameters.front().first); 33ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("utf-8", parameters.front().second); 34ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("application/json; charset=utf-8", 35ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko mime::Combine(s1, s2, parameters)); 36ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko} 37ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko 38ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex VakulenkoTEST(MimeUtils, ExtractParts) { 39ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko mime::Parameters parameters; 40ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko 41ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ(mime::types::kText, mime::GetType(mime::text::kPlain)); 42ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("plain", mime::GetSubtype(mime::text::kPlain)); 43ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko 44ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko parameters = mime::GetParameters("text/plain; charset=iso-8859-1;foo=bar"); 45ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ(2, parameters.size()); 46ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ(mime::parameters::kCharset, parameters[0].first); 47ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("iso-8859-1", parameters[0].second); 48ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("foo", parameters[1].first); 49ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("bar", parameters[1].second); 50ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko} 51ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko 52ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex VakulenkoTEST(MimeUtils, AppendRemoveParams) { 5305d29044d14a60775ed6c51c75a414eb0cb50347Alex Vakulenko std::string mime_string = mime::AppendParameter( 5405d29044d14a60775ed6c51c75a414eb0cb50347Alex Vakulenko mime::text::kXml, mime::parameters::kCharset, "utf-8"); 55ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("text/xml; charset=utf-8", mime_string); 56ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko mime_string = mime::AppendParameter(mime_string, "foo", "bar"); 57ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("text/xml; charset=utf-8; foo=bar", mime_string); 5805d29044d14a60775ed6c51c75a414eb0cb50347Alex Vakulenko EXPECT_EQ("utf-8", 5905d29044d14a60775ed6c51c75a414eb0cb50347Alex Vakulenko mime::GetParameterValue(mime_string, mime::parameters::kCharset)); 60ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("bar", mime::GetParameterValue(mime_string, "foo")); 61ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ("", mime::GetParameterValue(mime_string, "baz")); 62ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko mime_string = mime::RemoveParameters(mime_string); 63ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko EXPECT_EQ(mime::text::kXml, mime_string); 64ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko} 65ec5dabd1371006296e2cd4f7400aff85ebc980dfAlex Vakulenko 669ed0cab99f18acb3570a35e9408f24355f6b8324Alex Vakulenko} // namespace brillo 67