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)