summaryrefslogtreecommitdiffstats
path: root/chromium/components/policy/test_support/request_handler_for_unregister.cc
blob: ff8360d97dc2d722437cc1527f8d5239068bcb7d (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
// Copyright 2022 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 "components/policy/test_support/request_handler_for_unregister.h"

#include "components/policy/core/common/cloud/cloud_policy_constants.h"
#include "components/policy/proto/device_management_backend.pb.h"
#include "components/policy/test_support/client_storage.h"
#include "components/policy/test_support/test_server_helpers.h"
#include "net/http/http_status_code.h"
#include "net/test/embedded_test_server/http_request.h"
#include "net/test/embedded_test_server/http_response.h"

using net::test_server::HttpRequest;
using net::test_server::HttpResponse;

namespace em = enterprise_management;

namespace policy {

RequestHandlerForUnregister::RequestHandlerForUnregister(
    ClientStorage* client_storage,
    PolicyStorage* policy_storage)
    : EmbeddedPolicyTestServer::RequestHandler(client_storage, policy_storage) {
}

RequestHandlerForUnregister::~RequestHandlerForUnregister() = default;

std::string RequestHandlerForUnregister::RequestType() {
  return dm_protocol::kValueRequestUnregister;
}

std::unique_ptr<HttpResponse> RequestHandlerForUnregister::HandleRequest(
    const HttpRequest& request) {
  std::string request_device_token;
  if (!GetDeviceTokenFromRequest(request, &request_device_token) ||
      !client_storage()->DeleteClient(request_device_token)) {
    return CreateHttpResponse(net::HTTP_UNAUTHORIZED, "Invalid device token.");
  }

  em::DeviceManagementResponse device_management_response;
  device_management_response.mutable_unregister_response();
  return CreateHttpResponse(net::HTTP_OK,
                            device_management_response.SerializeAsString());
}

}  // namespace policy