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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
// Copyright (c) 2011 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 "chrome/test/webdriver/commands/webdriver_command.h"
#include <string>
#include "base/memory/scoped_ptr.h"
#include "base/memory/singleton.h"
#include "base/string_util.h"
#include "base/stringprintf.h"
#include "base/values.h"
#include "chrome/test/webdriver/commands/response.h"
#include "chrome/test/webdriver/webdriver_error.h"
#include "chrome/test/webdriver/webdriver_logging.h"
#include "chrome/test/webdriver/webdriver_session.h"
#include "chrome/test/webdriver/webdriver_session_manager.h"
#include "chrome/test/webdriver/webdriver_util.h"
namespace webdriver {
WebDriverCommand::WebDriverCommand(
const std::vector<std::string>& path_segments,
const base::DictionaryValue* const parameters)
: Command(path_segments, parameters), session_(NULL) {
}
WebDriverCommand::~WebDriverCommand() {}
bool WebDriverCommand::Init(Response* const response) {
// There should be at least 3 path segments to match "/session/$id".
session_id_ = GetPathVariable(2);
if (session_id_.length() == 0) {
response->SetError(
new Error(kBadRequest, "No session ID specified"));
return false;
}
session_ = SessionManager::GetInstance()->GetSession(session_id_);
if (session_ == NULL) {
response->SetError(
new Error(kSessionNotFound, "Session not found: " + session_id_));
return false;
}
std::string message = base::StringPrintf(
"Command received (%s)", JoinString(path_segments_, '/').c_str());
if (parameters_.get())
message += " with params " + JsonStringifyForDisplay(parameters_.get());
session_->logger().Log(kFineLogLevel, message);
if (ShouldRunPreAndPostCommandHandlers()) {
Error* error = session_->BeforeExecuteCommand();
if (error) {
response->SetError(error);
return false;
}
}
response->SetField("sessionId", new base::StringValue(session_id_));
return true;
}
void WebDriverCommand::Finish(Response* const response) {
// The session may have been terminated as a result of the command.
if (!SessionManager::GetInstance()->Has(session_id_))
return;
if (ShouldRunPreAndPostCommandHandlers()) {
scoped_ptr<Error> error(session_->AfterExecuteCommand());
if (error.get()) {
session_->logger().Log(kWarningLogLevel,
"AfterExecuteCommand failed: " + error->details());
}
}
LogLevel level = kWarningLogLevel;
if (response->GetStatus() == kSuccess)
level = kFineLogLevel;
session_->logger().Log(
level, base::StringPrintf(
"Command finished (%s) with response %s",
JoinString(path_segments_, '/').c_str(),
JsonStringifyForDisplay(response->GetDictionary()).c_str()));
}
bool WebDriverCommand::ShouldRunPreAndPostCommandHandlers() {
return true;
}
} // namespace webdriver
|
