python-nut2/tests/testclient.py at main · rshipp/python-nut2 · GitHub

History

253 lines (200 loc) · 8.59 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

import unittest

from mockserver import MockServer

import telnetlib

try:

from mock import Mock

except ImportError:

from unittest.mock import Mock

from nut2 import PyNUTClient, PyNUTError

class TestClient(unittest.TestCase):

def setUp(self):

self.client = PyNUTClient(connect=False, debug=True)

self.client._srv_handler = MockServer(broken=False)

self.broken_client = PyNUTClient(connect=False, debug=True)

self.broken_client._srv_handler = MockServer(broken=True)

self.not_ok_client = PyNUTClient(connect=False, debug=True)

self.not_ok_client._srv_handler = MockServer(ok=False,

broken=False)

self.valid = "test"

self.invalid = "does_not_exist"

self.valid_ups_name = "Test UPS 1"

self.valid_desc = self.valid_ups_name

self.valid_value = '100'

self.valid_command_desc = self.valid_desc

telnetlib.Telnet = Mock()

def test_init_with_args(self):

PyNUTClient(connect=False, login='test', password='test',

host='test', port=1)

def test_supports_context_manager(self):

try:

with PyNUTClient(connect=False) as client:

pass

except AttributeError:

assert(False)

def test_connect(self):

try:

PyNUTClient()

except Exception:

assert(False)

def test_connect_debug(self):

try:

PyNUTClient(debug=True)

except Exception:

assert(False)

def test_connect_broken(self):

telnetlib.Telnet = MockServer

client = PyNUTClient(login=self.valid, password=self.valid,

connect=False)

self.assertRaises(PyNUTError, client._connect)

def test_connect_credentials(self):

try:

PyNUTClient(login=self.valid, password=self.valid,

debug=True)

except TypeError:

pass

except PyNUTError:

pass

except Exception:

assert(False)

def test_connect_credentials_username_ok(self):

try:

telnetlib.Telnet = MockServer

PyNUTClient(login=self.valid, password=self.valid,

debug=True)

except TypeError:

pass

except PyNUTError:

pass

except Exception:

assert(False)

def test_get_ups_list(self):

ups_list = self.client.list_ups()

self.assertEquals(type(ups_list), dict)

self.assertEquals(len(ups_list), 2)

self.assertEquals(ups_list[self.valid], self.valid_ups_name)

def test_get_ups_list_broken(self):

self.assertRaises(PyNUTError, self.broken_client.list_ups)

def test_get_ups_vars_valid_ups(self):

vars = self.client.list_vars(self.valid)

self.assertEquals(type(vars), dict)

self.assertEquals(len(vars), 2)

self.assertEquals(vars['battery.charge'], '100')

def test_get_ups_vars_invalid_ups(self):

self.assertRaises(PyNUTError, self.client.list_vars, self.invalid)

def test_get_ups_vars_broken(self):

self.assertRaises(PyNUTError, self.broken_client.list_vars, self.valid)

def test_get_ups_commands_valid_ups(self):

commands = self.client.list_commands(self.valid)

self.assertEquals(type(commands), dict)

self.assertEquals(len(commands), 1)

self.assertEquals(commands[self.valid], self.valid_command_desc)

def test_get_ups_commands_invalid_ups(self):

self.assertRaises(PyNUTError, self.client.list_commands, self.invalid)

def test_get_ups_commands_broken(self):

self.assertRaises(PyNUTError, self.broken_client.list_commands,

self.valid)

def test_get_rw_vars_valid_ups(self):

vars = self.client.list_rw_vars(self.valid)

self.assertEquals(type(vars), dict)

self.assertEquals(vars[self.valid], self.valid)

def test_get_rw_vars_invalid_ups(self):

self.assertRaises(PyNUTError, self.client.list_rw_vars, self.invalid)

def test_set_rw_var_valid(self):

try:

self.client.set_var(self.valid, self.valid, self.valid)

except PyNUTError:

assert(False)

def test_set_rw_var_invalid(self):

self.assertRaises(PyNUTError, self.client.set_var, self.invalid,

self.invalid, self.invalid)

def test_set_rw_var_broken(self):

self.assertRaises(PyNUTError, self.broken_client.set_var, self.valid,

self.valid, self.valid)

def test_run_ups_command_valid(self):

try:

self.client.run_command(self.valid, self.valid)

except PyNUTError:

assert(False)

def test_run_ups_command_invalid(self):

self.assertRaises(PyNUTError, self.client.run_command, self.invalid,

self.invalid)

def test_run_ups_command_broken(self):

self.assertRaises(PyNUTError, self.broken_client.run_command, self.valid,

self.valid)

def test_fsd_valid_ups(self):

try:

self.client.fsd(self.valid)

except PyNUTError:

assert(False)

def test_fsd_invalid_ups(self):

self.assertRaises(PyNUTError, self.client.fsd, self.invalid)

def test_fsd_broken(self):

self.assertRaises(PyNUTError, self.broken_client.fsd, self.valid)

def test_fsd_not_ok(self):

self.assertRaises(PyNUTError, self.not_ok_client.fsd, self.valid)

def test_help(self):

self.assertEquals(self.client.help(), 'Commands: HELP VER GET LIST SET INSTCMD LOGIN LOGOUT USERNAME PASSWORD STARTTLS\n')

def test_ver(self):

self.assertEquals(self.client.ver(), 'Network UPS Tools upsd 2.7.1 - http://www.networkupstools.org/\n')

def test_list_clients_valid(self):

clients = self.client.list_clients(self.valid)

self.assertEquals(type(clients), dict)

self.assertEquals(clients[self.valid], [self.valid])

def test_list_clients_invalid(self):

self.assertRaises(PyNUTError, self.client.list_clients,

self.invalid)

def test_list_clients_none(self):

self.assertRaises(PyNUTError, self.client.list_clients)

def test_list_clients_broken(self):

self.assertRaises(PyNUTError, self.broken_client.list_clients,

self.valid)

def test_num_logins(self):

self.assertEquals(self.client.num_logins(self.valid), 1)

def test_num_logins_broken(self):

self.assertRaises(PyNUTError, self.broken_client.num_logins,

self.valid)

def test_description(self):

self.assertEquals(self.client.description(self.valid),

self.valid_desc)

def test_description_broken(self):

self.assertRaises(PyNUTError, self.broken_client.description,

self.valid)

def test_get(self):

self.assertEquals(self.client.get(self.valid, self.valid),

self.valid_value)

def test_get_var_alias_for_get(self):

self.assertEquals(self.client.get(self.valid, self.valid),

self.client.get_var(self.valid, self.valid))

def test_get_broken(self):

self.assertRaises(PyNUTError, self.broken_client.get,

self.valid, self.valid)

def test_var_description(self):

self.assertEquals(self.client.var_description(self.valid,

self.valid), self.valid_desc)

def test_var_description_broken(self):

self.assertRaises(PyNUTError, self.broken_client.var_description,

self.valid, self.valid)

def test_command_description(self):

self.assertEquals(self.client.command_description(self.valid,

self.valid), self.valid_desc)

def test_command_description_broken(self):

self.assertRaises(PyNUTError, self.broken_client.command_description,

self.valid, self.valid)

def test_list_enum(self):

self.assertEquals(self.client.list_enum(self.valid, self.valid),

[self.valid_desc])

def test_list_enum_broken(self):

self.assertRaises(PyNUTError, self.broken_client.list_enum,

self.valid, self.valid)

def test_list_range(self):

self.assertEquals(self.client.list_range(self.valid, self.valid),

[self.valid_desc])

def test_list_range_broken(self):

self.assertRaises(PyNUTError, self.broken_client.list_range,

self.valid, self.valid)

def test_var_type(self):

self.assertEquals(self.client.var_type(self.valid, self.valid),

"RW STRING:3")

def test_var_type_broken(self):

self.assertRaises(PyNUTError, self.broken_client.var_type,

self.valid, self.valid)

python-nut2/tests/testclient.py at main · rshipp/python-nut2 · GitHub

Expand file tree

Breadcrumbs

testclient.py

Latest commit

History

Breadcrumbs

testclient.py

File metadata and controls