adding two missing client fakes (#31)

Author: mpholcomb

accounting_client_fake.go

@@ -0,0 +1,41 @@
+package modzy
+
+import (
+	"context"
+)
+
+// AccountingClientFake is meant to help in mocking the AccountingClient interface easily for unit testing.
+type AccountingClientFake struct {
+	GetEntitlementsFunc     func(ctx context.Context) (*GetEntitlementsOutput, error)
+	HasEntitlementFunc      func(ctx context.Context, entitlement string) (bool, error)
+	GetLicenseFunc          func(ctx context.Context) (*GetLicenseOutput, error)
+	ListAccountingUsersFunc func(ctx context.Context, input *ListAccountingUsersInput) (*ListAccountingUsersOutput, error)
+	ListProjectsFunc        func(ctx context.Context, input *ListProjectsInput) (*ListProjectsOutput, error)
+	GetProjectDetailsFunc   func(ctx context.Context, input *GetProjectDetailsInput) (*GetProjectDetailsOutput, error)
+}
+
+var _ AccountingClient = &AccountingClientFake{}
+
+func (c *AccountingClientFake) GetEntitlements(ctx context.Context) (*GetEntitlementsOutput, error) {
+	return c.GetEntitlementsFunc(ctx)
+}
+
+func (c *AccountingClientFake) HasEntitlement(ctx context.Context, entitlement string) (bool, error) {
+	return c.HasEntitlementFunc(ctx, entitlement)
+}
+
+func (c *AccountingClientFake) GetLicense(ctx context.Context) (*GetLicenseOutput, error) {
+	return c.GetLicenseFunc(ctx)
+}
+
+func (c *AccountingClientFake) ListAccountingUsers(ctx context.Context, input *ListAccountingUsersInput) (*ListAccountingUsersOutput, error) {
+	return c.ListAccountingUsersFunc(ctx, input)
+}
+
+func (c *AccountingClientFake) ListProjects(ctx context.Context, input *ListProjectsInput) (*ListProjectsOutput, error) {
+	return c.ListProjectsFunc(ctx, input)
+}
+
+func (c *AccountingClientFake) GetProjectDetails(ctx context.Context, input *GetProjectDetailsInput) (*GetProjectDetailsOutput, error) {
+	return c.GetProjectDetailsFunc(ctx, input)
+}

accounting_client_fake_test.go

@@ -0,0 +1,79 @@
+package modzy
+
+import (
+	"context"
+	"testing"
+)
+
+func TestAccountingClientFake(t *testing.T) {
+	expectedCtx := context.WithValue(context.TODO(), testContextKey("a"), "b")
+
+	calls := 0
+	fake := &AccountingClientFake{
+		GetEntitlementsFunc: func(ctx context.Context) (*GetEntitlementsOutput, error) {
+			calls++
+			if ctx != expectedCtx {
+				t.Errorf("not expected ctx")
+			}
+			return nil, nil
+		},
+		HasEntitlementFunc: func(ctx context.Context, entitlement string) (bool, error) {
+			calls++
+			if ctx != expectedCtx {
+				t.Errorf("not expected ctx")
+			}
+			if entitlement != "entitlement" {
+				t.Errorf("input was not passed through")
+			}
+			return false, nil
+		},
+		GetLicenseFunc: func(ctx context.Context) (*GetLicenseOutput, error) {
+			calls++
+			if ctx != expectedCtx {
+				t.Errorf("not expected ctx")
+			}
+			return nil, nil
+		},
+		ListAccountingUsersFunc: func(ctx context.Context, input *ListAccountingUsersInput) (*ListAccountingUsersOutput, error) {
+			calls++
+			if ctx != expectedCtx {
+				t.Errorf("not expected ctx")
+			}
+			if input == nil {
+				t.Errorf("input was not passed through")
+			}
+			return nil, nil
+		},
+		ListProjectsFunc: func(ctx context.Context, input *ListProjectsInput) (*ListProjectsOutput, error) {
+			calls++
+			if ctx != expectedCtx {
+				t.Errorf("not expected ctx")
+			}
+			if input == nil {
+				t.Errorf("input was not passed through")
+			}
+			return nil, nil
+		},
+		GetProjectDetailsFunc: func(ctx context.Context, input *GetProjectDetailsInput) (*GetProjectDetailsOutput, error) {
+			calls++
+			if ctx != expectedCtx {
+				t.Errorf("not expected ctx")
+			}
+			if input == nil {
+				t.Errorf("input was not passed through")
+			}
+			return nil, nil
+		},
+	}
+
+	fake.GetEntitlements(expectedCtx)
+	fake.HasEntitlement(expectedCtx, "entitlement")
+	fake.GetLicense(expectedCtx)
+	fake.ListAccountingUsers(expectedCtx, &ListAccountingUsersInput{})
+	fake.ListProjects(expectedCtx, &ListProjectsInput{})
+	fake.GetProjectDetails(expectedCtx, &GetProjectDetailsInput{})
+
+	if calls != 6 {
+		t.Errorf("Did not call all of the funcs: %d", calls)
+	}
+}

resources_client_fake.go

@@ -0,0 +1,16 @@
+package modzy
+
+import (
+	"context"
+)
+
+// ResourcesClientFake is meant to help in mocking the ResourcesClient interface easily for unit testing.
+type ResourcesClientFake struct {
+	GetProcessingModelsFunc func(ctx context.Context) (*GetProcessingModelsOutput, error)
+}
+
+var _ ResourcesClient = &ResourcesClientFake{}
+
+func (c *ResourcesClientFake) GetProcessingModels(ctx context.Context) (*GetProcessingModelsOutput, error) {
+	return c.GetProcessingModelsFunc(ctx)
+}

resources_client_fake_test.go

@@ -0,0 +1,27 @@
+package modzy
+
+import (
+	"context"
+	"testing"
+)
+
+func TestResourcesClientFake(t *testing.T) {
+	expectedCtx := context.WithValue(context.TODO(), testContextKey("a"), "b")
+
+	calls := 0
+	fake := &ResourcesClientFake{
+		GetProcessingModelsFunc: func(ctx context.Context) (*GetProcessingModelsOutput, error) {
+			calls++
+			if ctx != expectedCtx {
+				t.Errorf("not expected ctx")
+			}
+			return nil, nil
+		},
+	}
+
+	fake.GetProcessingModels(expectedCtx)
+
+	if calls != 1 {
+		t.Errorf("Did not call all of the funcs: %d", calls)
+	}
+}