func getTestRouter(t *testing.T) *testRouter { ctrl := gomock.NewController(t) discovery := discovery.NewMockClient(ctrl) r := &testRouter{ router: &router{ Component: &component.Component{ Discovery: discovery, Ctx: GetLogger(t, "TestRouter"), }, gateways: map[string]*gateway.Gateway{}, }, ctrl: ctrl, discovery: discovery, } r.InitStatus() return r }
func TestGetAndVerifyContext(t *testing.T) { r := rand.New(rand.NewSource(time.Now().UnixNano())) tmpDir := fmt.Sprintf("%s/%d", os.TempDir(), r.Int63()) os.Mkdir(tmpDir, 755) defer os.Remove(tmpDir) a := assertions.New(t) c := new(Component) c.Identity = new(discovery.Announcement) c.Config.KeyDir = tmpDir security.GenerateKeypair(tmpDir) c.initKeyPair() { ctx := c.GetContext("") _, err := c.ValidateNetworkContext(ctx) a.So(err, assertions.ShouldNotBeNil) } c.Identity.Id = "test-context" { ctx := c.GetContext("") _, err := c.ValidateNetworkContext(ctx) a.So(err, assertions.ShouldNotBeNil) } c.Identity.ServiceName = "test-service" ctrl := gomock.NewController(t) discoveryClient := discovery.NewMockClient(ctrl) c.Discovery = discoveryClient discoveryClient.EXPECT().Get("test-service", "test-context").Return(c.Identity, nil) ctx := c.GetContext("") _, err := c.ValidateNetworkContext(ctx) a.So(err, assertions.ShouldBeNil) }
func getTestBroker(t *testing.T) *testBroker { ctrl := gomock.NewController(t) discovery := pb_discovery.NewMockClient(ctrl) ns := pb_networkserver.NewMockNetworkServerClient(ctrl) b := &testBroker{ broker: &broker{ Component: &component.Component{ Discovery: discovery, Ctx: GetLogger(t, "TestBroker"), }, handlers: make(map[string]*handler), activationDeduplicator: NewDeduplicator(10 * time.Millisecond), uplinkDeduplicator: NewDeduplicator(10 * time.Millisecond), ns: ns, }, ns: ns, ctrl: ctrl, discovery: discovery, } b.InitStatus() return b }