File: client_test.go

package info (click to toggle)
golang-github-denverdino-aliyungo 0.0~git20180921.13fa8aa-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye
  • size: 1,824 kB
  • sloc: xml: 1,359; makefile: 3
file content (58 lines) | stat: -rw-r--r-- 1,493 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package sls

import (
	"os"
	"testing"

	"github.com/denverdino/aliyungo/common"
)

var (
	TestAccessKeyId     = os.Getenv("AccessKeyId")
	TestAccessKeySecret = os.Getenv("AccessKeySecret")
	TestSecurityToken   = os.Getenv("SecurityToken")
	TestRegionID        = common.Region(os.Getenv("RegionId"))
	TestMachineGroup    = os.Getenv("MachineGroup")
	Region              = common.Hangzhou
	TestProjectName     = os.Getenv("ProjectName")
	TestLogstoreName    = "test-logstore"
)

func DefaultProject(t *testing.T) *Project {
	client := NewClient(Region, false, TestAccessKeyId, TestAccessKeySecret)
	err := client.CreateProject(TestProjectName, "description")
	if err != nil {
		if e, ok := err.(*Error); ok && e.Code != "ProjectAlreadyExist" {
			t.Fatalf("create project fail: %s", err.Error())
		}
	}
	p, err := client.Project(TestProjectName)
	if err != nil {
		t.Fatalf("get project fail: %s", err.Error())
	}
	//Create default logstore

	logstore := &Logstore{
		TTL:   2,
		Shard: 3,
		Name:  TestLogstoreName,
	}
	err = p.CreateLogstore(logstore)
	if err != nil {
		if e, ok := err.(*Error); ok && e.Code != "LogStoreAlreadyExist" {
			t.Fatalf("create logstore fail: %s", err.Error())
		}
	}

	return p
}

var testDebugClient *Client

func NewTestClientForDebug() *Client {
	if testDebugClient == nil {
		testDebugClient = NewClientForAssumeRole(TestRegionID, false, TestAccessKeyId, TestAccessKeySecret, TestSecurityToken)
		testDebugClient.SetDebug(true)
	}
	return testDebugClient
}