Output returns responses from the remote credentials helper. It mocks those responses based in the input in the mock.
()
| 35 | // Output returns responses from the remote credentials helper. |
| 36 | // It mocks those responses based in the input in the mock. |
| 37 | func (m *mockCommand) Output() ([]byte, error) { |
| 38 | in, err := io.ReadAll(m.input) |
| 39 | if err != nil { |
| 40 | return nil, err |
| 41 | } |
| 42 | inS := string(in) |
| 43 | |
| 44 | switch m.arg { |
| 45 | case "erase": |
| 46 | switch inS { |
| 47 | case validServerAddress: |
| 48 | return nil, nil |
| 49 | default: |
| 50 | return []byte("program failed"), errCommandExited |
| 51 | } |
| 52 | case "get": |
| 53 | switch inS { |
| 54 | case validServerAddress: |
| 55 | return []byte(`{"Username": "foo", "Secret": "bar"}`), nil |
| 56 | case validServerAddress2: |
| 57 | return []byte(`{"Username": "<token>", "Secret": "abcd1234"}`), nil |
| 58 | case missingCredsAddress: |
| 59 | return []byte(credentials.NewErrCredentialsNotFound().Error()), errCommandExited |
| 60 | case invalidServerAddress: |
| 61 | return []byte("program failed"), errCommandExited |
| 62 | } |
| 63 | case "store": |
| 64 | var c credentials.Credentials |
| 65 | err := json.NewDecoder(strings.NewReader(inS)).Decode(&c) |
| 66 | if err != nil { |
| 67 | return []byte("program failed"), errCommandExited |
| 68 | } |
| 69 | switch c.ServerURL { |
| 70 | case validServerAddress: |
| 71 | return nil, nil |
| 72 | default: |
| 73 | return []byte("program failed"), errCommandExited |
| 74 | } |
| 75 | case "list": |
| 76 | return fmt.Appendf(nil, `{"%s": "%s", "%s": "%s"}`, validServerAddress, "foo", validServerAddress2, "<token>"), nil |
| 77 | } |
| 78 | |
| 79 | return fmt.Appendf(nil, "unknown argument %q with %q", m.arg, inS), errCommandExited |
| 80 | } |
| 81 | |
| 82 | // Input sets the input to send to a remote credentials helper. |
| 83 | func (m *mockCommand) Input(in io.Reader) { |