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/scoped_openssl_types.h"
10#include "testing/gtest/include/gtest/gtest.h"
11
12namespace crypto {
13
14TEST(OpenSSLBIOString, TestWrite) {
15  std::string s;
16  const std::string expected1("a one\nb 2\n");
17  const std::string expected2("c d e f");
18  const std::string expected3("g h i");
19  {
20    ScopedBIO bio(BIO_new_string(&s));
21    ASSERT_TRUE(bio.get());
22
23    EXPECT_EQ(static_cast<int>(expected1.size()),
24              BIO_printf(bio.get(), "a %s\nb %i\n", "one", 2));
25    EXPECT_EQ(expected1, s);
26
27    EXPECT_EQ(1, BIO_flush(bio.get()));
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
34    EXPECT_EQ(static_cast<int>(expected3.size()),
35              BIO_puts(bio.get(), expected3.c_str()));
36    EXPECT_EQ(expected1 + expected2 + expected3, s);
37  }
38  EXPECT_EQ(expected1 + expected2 + expected3, s);
39}
40
41TEST(OpenSSLBIOString, TestReset) {
42  std::string s;
43  const std::string expected1("a b c\n");
44  const std::string expected2("d e f g\n");
45  {
46    ScopedBIO bio(BIO_new_string(&s));
47    ASSERT_TRUE(bio.get());
48
49    EXPECT_EQ(static_cast<int>(expected1.size()),
50              BIO_write(bio.get(), expected1.data(), expected1.size()));
51    EXPECT_EQ(expected1, s);
52
53    EXPECT_EQ(1, BIO_reset(bio.get()));
54    EXPECT_EQ(std::string(), s);
55
56    EXPECT_EQ(static_cast<int>(expected2.size()),
57              BIO_write(bio.get(), expected2.data(), expected2.size()));
58    EXPECT_EQ(expected2, s);
59  }
60  EXPECT_EQ(expected2, s);
61}
62
63}  // namespace crypto
64