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', 0o600) arg.add_to_params({}, outfile) arg.save_query({'ResponseMetadata': {'HTTPStatusCode': 200}}) with open(outfile) as fp: self.assertEqual('', 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 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)