hal/dumpstate/1.1/tests/tests.cpp

/*
 * Copyright (C) 2020 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "gtest/gtest.h"

#include "DumpstateServer.h"
#include "ServiceDescriptor.h"
#include "ServiceSupplier.h"

#include <sstream>
#include <string>

static ServiceDescriptor MakePrinterService(const std::string& msg) {
    return ServiceDescriptor{msg, "/bin/echo -n \"" + msg + "\""};
}

class AccumulatorConsumer : public ServiceDescriptor::OutputConsumer {
  public:
    void Write(char* ptr, size_t len) override { ss.write(ptr, len); }

    std::string data() { return ss.str(); }

  private:
    std::stringstream ss;
};

TEST(DumpstateServer, RunCommand) {
    auto svc = MakePrinterService("hello world");
    AccumulatorConsumer ac;
    auto ok = svc.GetOutput(&ac);
    ASSERT_FALSE(ok.has_value());
    ASSERT_EQ("hello world", ac.data());
}