| // Copyright 2020 The Pigweed Authors |
| // |
| // 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 |
| // |
| // https://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. |
| #pragma once |
| |
| #include "pw_log/log.h" |
| #include "pw_unit_test/config.h" |
| #include "pw_unit_test/internal/rpc_event_handler.h" |
| #include "pw_unit_test_proto/unit_test.pwpb.h" |
| #include "pw_unit_test_proto/unit_test.raw_rpc.pb.h" |
| |
| namespace pw::unit_test { |
| |
| class UnitTestService final |
| : public pw_rpc::raw::UnitTest::Service<UnitTestService> { |
| public: |
| UnitTestService() : handler_(*this), verbose_(false) {} |
| |
| void Run(ConstByteSpan request, RawServerWriter& writer); |
| |
| private: |
| friend class internal::RpcEventHandler; |
| |
| // TODO(frolv): This function essentially performs what a pw_protobuf RPC |
| // method would do. Once that API is implemented, this service should be |
| // migrated to it. |
| template <typename WriteFunction> |
| void WriteEvent(WriteFunction event_writer) { |
| Event::MemoryEncoder event(encoding_buffer_); |
| event_writer(event); |
| if (event.status().ok()) { |
| writer_.Write(event) |
| .IgnoreError(); // TODO(pwbug/387): Handle Status properly |
| } |
| } |
| |
| void WriteTestRunStart(); |
| void WriteTestRunEnd(const RunTestsSummary& summary); |
| void WriteTestCaseStart(const TestCase& test_case); |
| void WriteTestCaseEnd(TestResult result); |
| void WriteTestCaseDisabled(const TestCase& test_case); |
| void WriteTestCaseExpectation(const TestExpectation& expectation); |
| |
| internal::RpcEventHandler handler_; |
| RawServerWriter writer_; |
| bool verbose_; |
| std::array<std::byte, config::kEventBufferSize> encoding_buffer_ = {}; |
| }; |
| |
| } // namespace pw::unit_test |