Add HttpsMonitor

infra_libs/ts_mon/test/config_test.py

 # Copyright 2015 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.
 
 import argparse
 import json
 import os
 import requests
 import tempfile
 import unittest
@@ skipping 47 matching lines @@
 
     self.assertIsInstance(interface.state.target, targets.DeviceTarget)
     self.assertEquals(interface.state.target.hostname, 'slave1-a1')
     self.assertEquals(interface.state.target.region, 'reg')
     self.assertEquals(args.ts_mon_flush, 'auto')
     self.assertIsNotNone(interface.state.flush_thread)
     self.assertTrue(standard_metrics.up.get())
 
   @mock.patch('requests.get', autospec=True)
   @mock.patch('socket.getfqdn', autospec=True)
+  @mock.patch('infra_libs.ts_mon.common.monitors.HttpsMonitor.'
+              '_load_credentials', autospec=True)
+  def test_https_monitor_args(self, _load_creds, fake_fqdn, fake_get):
+    print [_load_creds, fake_fqdn, fake_get]
+    fake_fqdn.return_value = 'slave1-a1.reg.tld'
+    fake_get.return_value.side_effect = requests.exceptions.ConnectionError
+    p = argparse.ArgumentParser()
+    config.add_argparse_options(p)
+    args = p.parse_args([
+        '--ts-mon-credentials', '/path/to/creds.p8.json',
+        '--ts-mon-endpoint', 'https://test/random:insert'])
+
+    config.process_argparse_options(args)
+
+    self.assertIsInstance(interface.state.global_monitor,
+                          monitors.HttpsMonitor)
+
+    self.assertIsInstance(interface.state.target, targets.DeviceTarget)
+    self.assertEquals(interface.state.target.hostname, 'slave1-a1')
+    self.assertEquals(interface.state.target.region, 'reg')
+    self.assertEquals(args.ts_mon_flush, 'auto')
+    self.assertIsNotNone(interface.state.flush_thread)
+    self.assertTrue(standard_metrics.up.get())
+
+  @mock.patch('requests.get', autospec=True)
+  @mock.patch('socket.getfqdn', autospec=True)
   def test_default_target_uppercase_fqdn(self, fake_fqdn, fake_get):
     fake_fqdn.return_value = 'SLAVE1-A1.REG.TLD'
     fake_get.return_value.side_effect = requests.exceptions.ConnectionError
     p = argparse.ArgumentParser()
     config.add_argparse_options(p)
     args = p.parse_args([
         '--ts-mon-credentials', '/path/to/creds.p8.json',
         '--ts-mon-endpoint', 'unsupported://www.googleapis.com/some/api'])
     config.process_argparse_options(args)
     self.assertIsInstance(interface.state.target, targets.DeviceTarget)
@@ skipping 280 matching lines @@
     with infra_libs.temporary_directory() as temp_dir:
       filename = os.path.join(temp_dir, 'config')
       with open(filename, 'w') as fh:
         fh.write('not a json file')
 
       with self.assertRaises(ValueError):
         config.load_machine_config(filename)
 
   def test_load_machine_config_not_exists(self):
     self.assertEquals({}, config.load_machine_config('does not exist'))