summaryrefslogtreecommitdiffstats
path: root/crypto/openssl_bio_string_unittest.cc
blob: 9dfa0e70f73325a7a6d24d840c2636210b68021b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "crypto/openssl_bio_string.h"

#include <openssl/bio.h>

#include "crypto/scoped_openssl_types.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace crypto {

TEST(OpenSSLBIOString, TestWrite) {
  std::string s;
  const std::string expected1("a one\nb 2\n");
  const std::string expected2("c d e f");
  const std::string expected3("g h i");
  {
    ScopedBIO bio(BIO_new_string(&s));
    ASSERT_TRUE(bio.get());

    EXPECT_EQ(static_cast<int>(expected1.size()),
              BIO_printf(bio.get(), "a %s\nb %i\n", "one", 2));
    EXPECT_EQ(expected1, s);

    EXPECT_EQ(1, BIO_flush(bio.get()));
    EXPECT_EQ(expected1, s);

    EXPECT_EQ(static_cast<int>(expected2.size()),
              BIO_write(bio.get(), expected2.data(), expected2.size()));
    EXPECT_EQ(expected1 + expected2, s);

    EXPECT_EQ(static_cast<int>(expected3.size()),
              BIO_puts(bio.get(), expected3.c_str()));
    EXPECT_EQ(expected1 + expected2 + expected3, s);
  }
  EXPECT_EQ(expected1 + expected2 + expected3, s);
}

TEST(OpenSSLBIOString, TestReset) {
  std::string s;
  const std::string expected1("a b c\n");
  const std::string expected2("d e f g\n");
  {
    ScopedBIO bio(BIO_new_string(&s));
    ASSERT_TRUE(bio.get());

    EXPECT_EQ(static_cast<int>(expected1.size()),
              BIO_write(bio.get(), expected1.data(), expected1.size()));
    EXPECT_EQ(expected1, s);

    EXPECT_EQ(1, BIO_reset(bio.get()));
    EXPECT_EQ(std::string(), s);

    EXPECT_EQ(static_cast<int>(expected2.size()),
              BIO_write(bio.get(), expected2.data(), expected2.size()));
    EXPECT_EQ(expected2, s);
  }
  EXPECT_EQ(expected2, s);
}

}  // namespace crypto