deferred_status_test.cc

#include "deferred_status.h"

#include "gmock.h"
#include "gunit.h"
#include "absl/time/time.h"
#include "grpcpp/support/status.h"

namespace {

using ::testing::status::StatusIs;

TEST(DeferredStatusTest, DeferredStatusIsReceived) {
  milotic::DeferredStatus status;
  status.Setter()(grpc::Status::OK);
  EXPECT_OK(status.AwaitStatus());
}

TEST(DeferredStatusTest, DeferredStatusTimesOut) {
  milotic::DeferredStatus status;
  EXPECT_THAT(status.AwaitStatus(absl::Milliseconds(10)),
              StatusIs(grpc::StatusCode::DEADLINE_EXCEEDED));
}
}  // namespace