def list_command(argv): list_parser = argparse.ArgumentParser( description="List either seen or pending connections") list_parser.add_argument("list") args = list_parser.parse_args(argv) if args.list == "seen": for ssid in var_io.readSeenSSIDs(): print(ssid) if args.list == "pending": for con in var_io.readPendingConnections(): try: print(con["802-11-wireless"]["ssid"]) except KeyError: print("WARN: Found non wireless pending connection: %s" % con["connection"]["id"])
def list_seen(): for ssid in var_io.readSeenSSIDs(): print(ssid)
def test_good_seen_SSIDs(self): f = mock.mock_open(read_data='Foo \nBar \n') self.assertEqual(['Foo', 'Bar'], var_io.readSeenSSIDs(open=f))
def test_numerical_as_str_seen_SSIDs(self): f = mock.mock_open(read_data='1234\n0.1\n-5\n') self.assertIsInstance(var_io.readSeenSSIDs(open=f)[0], str) self.assertIsInstance(var_io.readSeenSSIDs(open=f)[1], str) self.assertIsInstance(var_io.readSeenSSIDs(open=f)[2], str)
def test_error_reading_seen_SSIDs(self): f = mock.Mock(side_effect=IOError('foo')) with self.assertRaises(IOError): var_io.readSeenSSIDs(open=f)
def test_not_existant_seen_SSIDs(self): f = mock.Mock(side_effect=FileNotFoundError('foo')) self.assertEqual([], var_io.readSeenSSIDs(open=f))
def test_duplicate_seen_SSIDs(self): f = mock.mock_open(read_data='Foo\nFoo\nFoo\n') self.assertEqual(['Foo', 'Foo', 'Foo'], var_io.readSeenSSIDs(open=f))
def test_no_newline_seen_SSIDs(self): f = mock.mock_open(read_data='Foo') self.assertEqual(['Foo'], var_io.readSeenSSIDs(open=f)) f = mock.mock_open(read_data='Foo\nBar') self.assertEqual(['Foo', 'Bar'], var_io.readSeenSSIDs(open=f))
def test_empty_seen_SSIDs(self): f = mock.mock_open(read_data='') self.assertEqual([], var_io.readSeenSSIDs(open=f))