blob_file_writer_unittest.cc revision 3f39d5cc753905874d8d93bef94f857b8808f19e 
1//
2// Copyright (C) 2015 The Android Open Source Project
3//
4// Licensed under the Apache License, Version 2.0 (the "License");
5// you may not use this file except in compliance with the License.
6// You may obtain a copy of the License at
7//
8//      http://www.apache.org/licenses/LICENSE-2.0
9//
10// Unless required by applicable law or agreed to in writing, software
11// distributed under the License is distributed on an "AS IS" BASIS,
12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13// See the License for the specific language governing permissions and
14// limitations under the License.
15//
16
17#include "update_engine/payload_generator/blob_file_writer.h"
18
19#include <string>
20
21#include <gtest/gtest.h>
22
23#include "update_engine/test_utils.h"
24#include "update_engine/utils.h"
25
26using chromeos_update_engine::test_utils::FillWithData;
27using std::string;
28
29namespace chromeos_update_engine {
30
31class BlobFileWriterTest : public ::testing::Test {};
32
33TEST(BlobFileWriterTest, SimpleTest) {
34  string blob_path;
35  int blob_fd;
36  EXPECT_TRUE(utils::MakeTempFile("BlobFileWriterTest.XXXXXX",
37                                  &blob_path,
38                                  &blob_fd));
39  off_t blob_file_size = 0;
40  BlobFileWriter blob_file(blob_fd, &blob_file_size);
41
42  off_t blob_size = 1024;
43  brillo::Blob blob(blob_size);
44  FillWithData(&blob);
45  EXPECT_EQ(0, blob_file.StoreBlob(blob));
46  EXPECT_EQ(blob_size, blob_file.StoreBlob(blob));
47
48  brillo::Blob stored_blob(blob_size);
49  ssize_t bytes_read;
50  ASSERT_TRUE(utils::PReadAll(blob_fd,
51                              stored_blob.data(),
52                              blob_size,
53                              0,
54                              &bytes_read));
55  EXPECT_EQ(bytes_read, blob_size);
56  EXPECT_EQ(blob, stored_blob);
57}
58
59}  // namespace chromeos_update_engine
60