main_test.go

package main

import (
	"bytes"
	"testing"
)

func Test_processLine(t *testing.T) {
	tests := []struct {
		name string
		text string
		out  *bytes.Buffer
		want string
	}{
		{
			name: "empty",
			text: "",
			out:  &bytes.Buffer{},
			want: `
`,
		},
		{
			name: "whitespace",
			text: "  ",
			out:  &bytes.Buffer{},
			want: "  " + `
`,
		},
		{
			name: "new resource",
			text: " # module.example ",
			out:  &bytes.Buffer{},
			want: `# module.example
`,
		},
		{
			name: "no change",
			text: "    id  = \"some string\"",
			out:  &bytes.Buffer{},
			want: `    id  = "some string"
`,
		},
		{
			name: "added line",
			text: "   + id  = \"some string\"",
			out:  &bytes.Buffer{},
			want: `+    id  = "some string"
`,
		},
		{
			name: "removed line",
			text: "     - item = 0 -> null",
			out:  &bytes.Buffer{},
			want: `-      item = 0 -> null
`,
		},
		{
			name: "changed line",
			text: "  ~ \"new/version\"    = \"latest\" -> \"1.0.1\"",
			out:  &bytes.Buffer{},
			want: `~   "new/version"    = "latest" -> "1.0.1"
`,
		},
		{
			name: "complex changed line",
			text: "  ~ \"new/version = some -> thing\"    = \"latest\" -> \"1.0.1\"",
			out:  &bytes.Buffer{},
			want: `~   "new/version = some -> thing"    = "latest" -> "1.0.1"
`,
		},
		{
			name: "cannot edit in place",
			text: "     - item = 0 -> null",
			out:  &bytes.Buffer{},
			want: `-      item = 0 -> null
`,
		},
		{
			name: "pre-existing color",
			text: "   + id  = \"some string\"",
			out: &bytes.Buffer{},
			want: `+    id  = "some string"
`,
		},
		{
			name: "filter refreshing state",
			text: "module.apps.kubernetes_config_map.config-map: Refreshing state... [id=id1]",
			out:  &bytes.Buffer{},
			want: "", //no new line
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			processLine(tt.text, tt.out)
			got := tt.out.String()
			if got != tt.want {
				t.Errorf("output is now aligned. \nGot:  %v, \nwant: %v", got, tt.want)
				// t.Errorf("length  \nGot:  %v, \nwant: %v", len(got), len(tt.want))
			}
		})
	}
}

func Test_cleanRawInput(t *testing.T) {
	type args struct {
	}
	tests := []struct {
		name string
		raw  string
		want string
	}{
		{
			name: "only whitespace",
			raw:  "     ",
			want: "     ",
		},
		{
			name: "no color with space",
			raw:  "    id  = \"some string\"   ",
			want: "    id  = \"some string\"   ",
		},
		{
			name: "partial colored",
			raw: "   + id  = \"some string\"",
			want: "   + id  = \"some string\"",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := cleanRawInput(tt.raw); got != tt.want {
				t.Errorf("cleanRawInput(%s) \ngot:  %v, \nwant: %v", tt.name, got, tt.want)
			}
		})
	}
}

func Test_iterateInput(t *testing.T) {
	tests := []struct {
		name    string
		input   string
		out     *bytes.Buffer
		wantOut string
	}{
		{
			name:    "empty",
			input:   "",
			out:     &bytes.Buffer{},
			wantOut: ``,
		},
		{
			name: "example",
			input: `
  # module.apps.kubernetes_deployment.deployment is tainted, so must be replaced
-/+ resource "kubernetes_deployment" "deployment" {
      ~ id               = "some-app" -> (known after apply)
        wait_for_rollout = true

              ~ spec {
                  - active_deadline_seconds          = 0 -> null
`,
			out: &bytes.Buffer{},
			wantOut: `
# module.apps.kubernetes_deployment.deployment is tainted, so must be replaced
-/+ resource "kubernetes_deployment" "deployment" {
~       id               = "some-app" -> (known after apply)
        wait_for_rollout = true

~               spec {
-                   active_deadline_seconds          = 0 -> null
`,
		},
		{
			name: "multiline example",
			input: `
 # local_file.foo will be created
+ resource "local_file" "foo" {
    + content              = <<~EOT
        can:
          enter:
          - yaml
          - or
          + anything
        EOT
    + directory_permission = "0777"
`,
			out: &bytes.Buffer{},
			wantOut: `
# local_file.foo will be created
+ resource "local_file" "foo" {
+     content              = <<~EOT
        can:
          enter:
          - yaml
          - or
          + anything
        EOT
+     directory_permission = "0777"
`,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			var stdin bytes.Buffer
			stdin.Write([]byte(tt.input))
			iterateInput(&stdin, tt.out)
			got := tt.out.String()
			if got != tt.wantOut {
				t.Errorf("output for %s is now aligned. \nGot:  %v, \nwant: %v", tt.name, got, tt.wantOut)
				t.Errorf("length  \nGot:  %v, \nwant: %v", len(got), len(tt.wantOut))
			}
		})
	}
}