def register_create_keys_and_cert_arguments(session, argument_table, **kwargs):
"""Add outfile save arguments to create-keys-and-certificate
- ``--certificate-pem-outfile``
- ``--public-key-outfile``
- ``--private-key-outfile``
"""
after_event = 'after-call.iot.CreateKeysAndCertificate'
argument_table['certificate-pem-outfile'] = QueryOutFileArgument(
session=session,
name='certificate-pem-outfile',
query='certificatePem',
after_call_event=after_event,
perm=0o600)
argument_table['public-key-outfile'] = QueryOutFileArgument(
session=session,
name='public-key-outfile',
query='keyPair.PublicKey',
after_call_event=after_event,
perm=0o600)
argument_table['private-key-outfile'] = QueryOutFileArgument(
session=session,
name='private-key-outfile',
query='keyPair.PrivateKey',
after_call_event=after_event,
perm=0o600)
def test_saves_query_to_file_as_empty_string_when_none_result(self):
outfile = self.files.create_file("none-test", "")
session = mock.Mock()
arg = QueryOutFileArgument(session, "foo", "baz", "event", 0o600)
arg.add_to_params({}, outfile)
arg.save_query({"ResponseMetadata": {"HTTPStatusCode": 200}})
with open(outfile) as fp:
self.assertEquals("", fp.read())
def test_saves_query_to_file_as_empty_string_when_none_result(self):
outfile = self.files.create_file('none-test', '')
session = mock.Mock()
arg = QueryOutFileArgument(session, 'foo', 'baz', 'event')
arg.add_to_params({}, outfile)
arg.save_query({'ResponseMetadata': {'HTTPStatusCode': 200}})
with open(outfile) as fp:
self.assertEquals('', fp.read())
def test_permissions_on_created_file(self):
outfile = self.files.create_file("not-empty-test", "")
session = mock.Mock()
arg = QueryOutFileArgument(session, "foo", "baz", "event", 0o600)
arg.add_to_params({}, outfile)
arg.save_query({"ResponseMetadata": {"HTTPStatusCode": 200}, "baz": "abc123"})
with open(outfile) as fp:
fp.read()
self.assertEqual(os.stat(outfile).st_mode & 0xFFF, 0o600)
def test_saves_query_to_file(self):
outfile = self.files.create_file("not-empty-test", "")
session = mock.Mock()
arg = QueryOutFileArgument(session, "foo", "baz", "event", 0o600)
arg.add_to_params({}, outfile)
arg.save_query({"ResponseMetadata": {"HTTPStatusCode": 200}, "baz": "abc123"})
with open(outfile) as fp:
self.assertEquals("abc123", fp.read())
self.assertEquals(1, session.register.call_count)
session.register.assert_called_with("event", arg.save_query)
def test_saves_query_to_file(self):
outfile = self.files.create_file('not-empty-test', '')
session = mock.Mock()
arg = QueryOutFileArgument(session, 'foo', 'baz', 'event')
arg.add_to_params({}, outfile)
arg.save_query({'ResponseMetadata': {'HTTPStatusCode': 200},
'baz': 'abc123'})
with open(outfile) as fp:
self.assertEquals('abc123', fp.read())
self.assertEquals(1, session.register.call_count)
session.register.assert_called_with('event', arg.save_query)
def register_create_keys_from_csr_arguments(session, argument_table, **kwargs):
"""Add certificate-pem-outfile to create-certificate-from-csr"""
argument_table['certificate-pem-outfile'] = QueryOutFileArgument(
session=session,
name='certificate-pem-outfile',
query='certificatePem',
after_call_event='after-call.iot.CreateCertificateFromCsr')
def test_does_not_add_help_text_if_set(self):
session = mock.Mock()
arg = QueryOutFileArgument(session,
'foo',
'bar.baz',
'event',
help_text='abc')
self.assertEqual('abc', arg.documentation)
def test_saves_query_to_file_as_empty_string_when_none_result(self):
outfile = self.files.create_file('none-test', '')
session = mock.Mock()
arg = QueryOutFileArgument(session, 'foo', 'baz', 'event', 0o600)
arg.add_to_params({}, outfile)
arg.save_query({'ResponseMetadata': {'HTTPStatusCode': 200}})
with open(outfile) as fp:
self.assertEqual('', fp.read())
def test_permissions_on_created_file(self):
outfile = self.files.create_file('not-empty-test', '')
session = mock.Mock()
arg = QueryOutFileArgument(session, 'foo', 'baz', 'event', 0o600)
arg.add_to_params({}, outfile)
arg.save_query({
'ResponseMetadata': {
'HTTPStatusCode': 200
},
'baz': 'abc123'
})
with open(outfile) as fp:
fp.read()
self.assertEqual(os.stat(outfile).st_mode & 0xFFF, 0o600)
def test_saves_query_to_file(self):
outfile = self.files.create_file('not-empty-test', '')
session = mock.Mock()
arg = QueryOutFileArgument(session, 'foo', 'baz', 'event', 0o600)
arg.add_to_params({}, outfile)
arg.save_query({
'ResponseMetadata': {
'HTTPStatusCode': 200
},
'baz': 'abc123'
})
with open(outfile) as fp:
self.assertEqual('abc123', fp.read())
self.assertEqual(1, session.register.call_count)
session.register.assert_called_with('event', arg.save_query)
def test_proxies_to_super_ctor(self):
session = mock.Mock()
arg = QueryOutFileArgument(session, 'foo', 'bar.baz', 'event', 0o600)
self.assertEqual('foo', arg.name)
self.assertEqual('bar.baz', arg.query)
def test_does_not_save_when_not_set(self):
session = mock.Mock()
QueryOutFileArgument(session, 'foo', 'baz', 'event', 0o600)
self.assertEqual(0, session.register.call_count)
def test_adds_default_help_text(self):
session = mock.Mock()
arg = QueryOutFileArgument(session, 'foo', 'bar.baz', 'event', 0o600)
self.assertEqual(('Saves the command output contents of bar.baz '
'to the given filename'), arg.documentation)
