@@ -0,0 +1,2417 @@
+package com.github.kongchen.swagger.docgen.mavenplugin;%0A%0Aimport com.google.common.collect.Sets;%0Aimport io.swagger.annotations.Extension;%0Aimport io.swagger.annotations.ExtensionProperty;%0Aimport io.swagger.annotations.SwaggerDefinition;%0Aimport io.swagger.models.Info;%0Aimport org.mockito.Mockito;%0Aimport org.mockito.MockitoAnnotations;%0Aimport org.mockito.Spy;%0Aimport org.testng.Assert;%0Aimport org.testng.annotations.BeforeMethod;%0Aimport org.testng.annotations.Test;%0A%0Aimport java.util.HashMap;%0Aimport java.util.Map;%0Aimport java.util.Set;%0A%0Apublic class ApiSourceTest %7B%0A%0A    @Spy%0A    protected ApiSource apiSource;%0A%0A    @BeforeMethod%0A    public void setUp() %7B%0A        MockitoAnnotations.initMocks(this);%0A    %7D%0A%0A    @Test%0A    public void testGetInfo0VendorExtensions() %7B%0A        Map%3CString, Object%3E expectedExtensions = new HashMap%3CString, Object%3E();%0A        expectedExtensions.put(%22x-logo%22, new HashMap%3CString, Object%3E()%7B%7B%0A            put(%22logo%22, %22logo url%22);%0A            put(%22description%22, %22This is our logo.%22);%0A        %7D%7D);%0A        expectedExtensions.put(%22x-website%22, new HashMap%3CString, Object%3E()%7B%7B%0A            put(%22website%22, %22website url%22);%0A            put(%22description%22, %22This is our website.%22);%0A        %7D%7D);%0A%0A%0A        Set%3CClass%3C?%3E%3E validClasses = Sets.newHashSet(ApiSourceTestClass.class);%0A%0A        Mockito.when(apiSource.getValidClasses(SwaggerDefinition.class)).thenReturn(validClasses);%0A        Info info = apiSource.getInfo();%0A%0A        Map%3CString, Object%3E vendorExtensions = info.getVendorExtensions();%0A        Assert.assertEquals(vendorExtensions.size(), 2);%0A        Assert.assertEquals(vendorExtensions, expectedExtensions);%0A    %7D%0A%0A    @SwaggerDefinition(info = @io.swagger.annotations.Info(%0A            title = %22ApiSourceTestClass%22,%0A            version = %221.0.0%22,%0A            extensions = %7B%0A                    @Extension(name = %22logo%22, properties = %7B%0A                            @ExtensionProperty(name = %22logo%22, value = %22logo url%22),%0A                            @ExtensionProperty(name = %22description%22, value = %22This is our logo.%22)%0A                    %7D),%0A                    @Extension(name = %22website%22, properties = %7B%0A                            @ExtensionProperty(name = %22website%22, value = %22website url%22),%0A                            @ExtensionProperty(name = %22description%22, value = %22This is our website.%22)%0A                    %7D)%0A            %7D%0A    ))%0A    private static class ApiSourceTestClass %7B%0A%0A    %7D%0A%7D%0A

Add Unit test