src/cmd/fix/context_test.go

// Copyright 2016 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

func init() {
	addTestCases(contextTests, ctxfix)
}

var contextTests = []testCase{
	{
		Name: "context.0",
		In: `package main

import "golang.org/x/net/context"

var _ = "golang.org/x/net/context"
`,
		Out: `package main

import "context"

var _ = "golang.org/x/net/context"
`,
	},
	{
		Name: "context.1",
		In: `package main

import ctx "golang.org/x/net/context"

var _ = ctx.Background()
`,
		Out: `package main

import ctx "context"

var _ = ctx.Background()
`,
	},
}