// Copyright 2019 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package main

import (
	"bytes"
	"testing"
)

func TestRun(t *testing.T) {
	in := bytes.NewBuffer([]byte(`
test.com/A test.com/B@v1.2.3
test.com/B test.com/C@v4.5.6
`))
	out := bytes.Buffer{}

	if err := Run(in, &out); err != nil {
		t.Fatal(err)
	}

	want := `digraph gomodgraph {
	"test.com/A" -> "test.com/B@v1.2.3"
	"test.com/B" -> "test.com/C@v4.5.6"
}
`
	if out.String() != want {
		t.Fatalf("\ngot: %s\nwant: %s", out.String(), want)
	}
}