30 lines
509 B
Go
30 lines
509 B
Go
package common
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestIsAzure(t *testing.T) {
|
|
f, err := ioutil.TempFile("", "TestIsAzure*")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.Remove(f.Name())
|
|
|
|
f.Seek(0, 0)
|
|
f.Truncate(0)
|
|
f.Write([]byte("not the azure assettag"))
|
|
|
|
assert.False(t, isAzureAssetTag(f.Name()), "asset tag is not Azure's")
|
|
|
|
f.Seek(0, 0)
|
|
f.Truncate(0)
|
|
f.Write(azureAssetTag)
|
|
|
|
assert.True(t, isAzureAssetTag(f.Name()), "asset tag is Azure's")
|
|
}
|