loggerv2_test.go 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. /*
  2. *
  3. * Copyright 2017 gRPC authors.
  4. *
  5. * Licensed under the Apache License, Version 2.0 (the "License");
  6. * you may not use this file except in compliance with the License.
  7. * You may obtain a copy of the License at
  8. *
  9. * http://www.apache.org/licenses/LICENSE-2.0
  10. *
  11. * Unless required by applicable law or agreed to in writing, software
  12. * distributed under the License is distributed on an "AS IS" BASIS,
  13. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. * See the License for the specific language governing permissions and
  15. * limitations under the License.
  16. *
  17. */
  18. package grpclog
  19. import (
  20. "bytes"
  21. "fmt"
  22. "regexp"
  23. "testing"
  24. )
  25. func TestLoggerV2Severity(t *testing.T) {
  26. buffers := []*bytes.Buffer{new(bytes.Buffer), new(bytes.Buffer), new(bytes.Buffer)}
  27. SetLoggerV2(NewLoggerV2(buffers[infoLog], buffers[warningLog], buffers[errorLog]))
  28. Info(severityName[infoLog])
  29. Warning(severityName[warningLog])
  30. Error(severityName[errorLog])
  31. for i := 0; i < fatalLog; i++ {
  32. buf := buffers[i]
  33. // The content of info buffer should be something like:
  34. // INFO: 2017/04/07 14:55:42 INFO
  35. // WARNING: 2017/04/07 14:55:42 WARNING
  36. // ERROR: 2017/04/07 14:55:42 ERROR
  37. for j := i; j < fatalLog; j++ {
  38. b, err := buf.ReadBytes('\n')
  39. if err != nil {
  40. t.Fatal(err)
  41. }
  42. if err := checkLogForSeverity(j, b); err != nil {
  43. t.Fatal(err)
  44. }
  45. }
  46. }
  47. }
  48. // check if b is in the format of:
  49. //
  50. // 2017/04/07 14:55:42 WARNING: WARNING
  51. func checkLogForSeverity(s int, b []byte) error {
  52. expected := regexp.MustCompile(fmt.Sprintf(`^[0-9]{4}/[0-9]{2}/[0-9]{2} [0-9]{2}:[0-9]{2}:[0-9]{2} %s: %s\n$`, severityName[s], severityName[s]))
  53. if m := expected.Match(b); !m {
  54. return fmt.Errorf("got: %v, want string in format of: %v", string(b), severityName[s]+": 2016/10/05 17:09:26 "+severityName[s])
  55. }
  56. return nil
  57. }