Python QueryOutFileArgument.save_query示例

示例#1

文件： test_arguments.py 项目： krux/aws-cli

 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())

示例#2

 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())

示例#3

文件： test_arguments.py 项目： mtdowling/aws-cli

 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())

示例#4

文件： test_arguments.py 项目： krux/aws-cli

 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)

示例#5

文件： test_arguments.py 项目： krux/aws-cli

 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)

示例#6

文件： test_arguments.py 项目： mtdowling/aws-cli

 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)

示例#7

 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)

示例#8

 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)