crypto/openssl_bio_string_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "crypto/openssl_bio_string.h"
   6
   7 #include <openssl/bio.h>
   8
   9 #include "crypto/openssl_util.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 TEST(OpenSSLBIOString, TestWrite) {
  13   std::string s;
  14   const std::string expected1("a one\nb 2\n");
  15   const std::string expected2("c d e f");
  16   const std::string expected3("g h i");
  17   {
  18     crypto::ScopedOpenSSL<BIO, BIO_free_all> bio(crypto::BIO_new_string(&s));
  19     ASSERT_TRUE(bio.get());
  20
  21     EXPECT_EQ(static_cast<int>(expected1.size()),
  22               BIO_printf(bio.get(), "a %s\nb %i\n", "one", 2));
  23     EXPECT_EQ(expected1, s);
  24     EXPECT_EQ(static_cast<int>(expected1.size()), BIO_tell(bio.get()));
  25
  26     EXPECT_EQ(1, BIO_flush(bio.get()));
  27     EXPECT_EQ(-1, BIO_seek(bio.get(), 0));
  28     EXPECT_EQ(expected1, s);
  29
  30     EXPECT_EQ(static_cast<int>(expected2.size()),
  31               BIO_write(bio.get(), expected2.data(), expected2.size()));
  32     EXPECT_EQ(expected1 + expected2, s);
  33     EXPECT_EQ(static_cast<int>(expected1.size() + expected2.size()),
  34               BIO_tell(bio.get()));
  35
  36     EXPECT_EQ(static_cast<int>(expected3.size()),
  37               BIO_puts(bio.get(), expected3.c_str()));
  38     EXPECT_EQ(expected1 + expected2 + expected3, s);
  39     EXPECT_EQ(static_cast<int>(expected1.size() + expected2.size() +
  40                                expected3.size()),
  41               BIO_tell(bio.get()));
  42   }
  43   EXPECT_EQ(expected1 + expected2 + expected3, s);
  44 }
  45
  46 TEST(OpenSSLBIOString, TestReset) {
  47   std::string s;
  48   const std::string expected1("a b c\n");
  49   const std::string expected2("d e f g\n");
  50   {
  51     crypto::ScopedOpenSSL<BIO, BIO_free_all> bio(crypto::BIO_new_string(&s));
  52     ASSERT_TRUE(bio.get());
  53
  54     EXPECT_EQ(static_cast<int>(expected1.size()),
  55               BIO_write(bio.get(), expected1.data(), expected1.size()));
  56     EXPECT_EQ(expected1, s);
  57
  58     EXPECT_EQ(1, BIO_reset(bio.get()));
  59     EXPECT_EQ(std::string(), s);
  60
  61     EXPECT_EQ(static_cast<int>(expected2.size()),
  62               BIO_write(bio.get(), expected2.data(), expected2.size()));
  63     EXPECT_EQ(expected2, s);
  64   }
  65   EXPECT_EQ(expected2, s);
  66 }